added user images partitioning
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user