ryan/chore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added user images partitioning

Browse Source
This commit is contained in:
Ryan Kegel
2025-12-12 16:17:23 -05:00
parent 9b90ca12fb
commit f8d709f292
22 changed files with 91 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
db/db.py
View File

@@ -1,18 +1,10 @@
# python
import os
from config.paths import get_database_dir
import threading
from tinydb import TinyDB
DB_ENV = os.environ.get('DB_ENV', 'prod')
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if DB_ENV == 'prod':
base_dir = os.path.join(project_root, 'data/db')
else:
base_dir = os.path.join(project_root, 'test_data/db')
base_dir = get_database_dir()
os.makedirs(base_dir, exist_ok=True)
@@ -95,7 +87,7 @@ reward_db = LockedTable(_reward_db)
image_db = LockedTable(_image_db)
pending_reward_db = LockedTable(_pending_rewards_db)
if DB_ENV == 'test':
if os.environ.get('DB_ENV', 'prod') == 'test':
child_db.truncate()
task_db.truncate()
reward_db.truncate()

42
db/default.py
View File

@@ -1,14 +1,14 @@
# python
# File: db/debug.py
import random
from models.child import Child
from models.task import Task
from models.reward import Reward
from models.image import Image
from tinydb import Query
from db.db import child_db, task_db, reward_db, image_db
from api.image_api import IMAGE_TYPE_ICON, IMAGE_TYPE_PROFILE
from db.db import task_db, reward_db, image_db
from models.image import Image
from models.reward import Reward
from models.task import Task
def populate_default_data():
# Create tasks
@@ -86,19 +86,33 @@ def initializeImages():
"""Initialize the image database with default images if empty."""
if len(image_db.all()) == 0:
image_defs = [
('computer-game', IMAGE_TYPE_ICON, '.png', True),
('ice-cream', IMAGE_TYPE_ICON, '.png', True),
('meal', IMAGE_TYPE_ICON, '.png', True),
('playground', IMAGE_TYPE_ICON, '.png', True),
('tablet', IMAGE_TYPE_ICON, '.png', True),
('boy01', IMAGE_TYPE_PROFILE, '.png', True),
('girl01', IMAGE_TYPE_PROFILE, '.png', True),
('girl02', IMAGE_TYPE_PROFILE, '.png', True),
('boy02', IMAGE_TYPE_PROFILE, '.png', True),
('boy03', IMAGE_TYPE_PROFILE, '.png', True),
('girl03', IMAGE_TYPE_PROFILE, '.png', True),
('boy04', IMAGE_TYPE_PROFILE, '.png', True),
('broom', IMAGE_TYPE_ICON, '.png', True),
('computer-game', IMAGE_TYPE_ICON, '.png', True),
('fighting', IMAGE_TYPE_ICON, '.png', True),
('games-with-dad', IMAGE_TYPE_ICON, '.png', True),
('girl01', IMAGE_TYPE_PROFILE, '.png', True),
('girl02', IMAGE_TYPE_PROFILE, '.png', True),
('girl03', IMAGE_TYPE_PROFILE, '.png', True),
('girl04', IMAGE_TYPE_PROFILE, '.png', True),
('good', IMAGE_TYPE_ICON, '.png', True),
('homework', IMAGE_TYPE_ICON, '.png', True),
('ice-cream', IMAGE_TYPE_ICON, '.png', True),
('ignore', IMAGE_TYPE_ICON, '.png', True),
('lying', IMAGE_TYPE_ICON, '.png', True),
('make-the-bed', IMAGE_TYPE_ICON, '.png', True),
('meal', IMAGE_TYPE_ICON, '.png', True),
('money', IMAGE_TYPE_ICON, '.png', True),
('playground', IMAGE_TYPE_ICON, '.png', True),
('tablet', IMAGE_TYPE_ICON, '.png', True),
('toilet', IMAGE_TYPE_ICON, '.png', True),
('trash-can', IMAGE_TYPE_ICON, '.png', True),
('tv', IMAGE_TYPE_ICON, '.png', True),
('vacuum', IMAGE_TYPE_ICON, '.png', True),
('yelling', IMAGE_TYPE_ICON, '.png', True),
]
for _id, _type, ext, perm in image_defs:
img = Image(type=_type, extension=ext, permanent=perm)