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

round 4

Browse Source
This commit is contained in:
Ryan Kegel
2025-12-05 17:40:57 -05:00
parent 6423d1c1a2
commit fa9fabcd9f
43 changed files with 1506 additions and 529 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
db/db.py
View File

@@ -70,26 +70,32 @@ if DB_ENV == 'test':
task_path = os.path.join(base_dir, 'test_tasks.json')
reward_path = os.path.join(base_dir, 'test_rewards.json')
image_path = os.path.join(base_dir, 'test_images.json')
pending_reward_path = os.path.join(base_dir, 'test_pending_rewards.json')
else:
child_path = os.path.join(base_dir, 'children.json')
task_path = os.path.join(base_dir, 'tasks.json')
reward_path = os.path.join(base_dir, 'rewards.json')
image_path = os.path.join(base_dir, 'images.json')
pending_reward_path = os.path.join(base_dir, 'pending_rewards.json')
# Use separate TinyDB instances/files for each collection
_child_db = TinyDB(child_path, indent=2)
_task_db = TinyDB(task_path, indent=2)
_reward_db = TinyDB(reward_path, indent=2)
_image_db = TinyDB(image_path, indent=2)
_pending_rewards_db = TinyDB(pending_reward_path, indent=2)
# Expose table objects wrapped with locking
child_db = LockedTable(_child_db)
task_db = LockedTable(_task_db)
reward_db = LockedTable(_reward_db)
image_db = LockedTable(_image_db)
pending_reward_db = LockedTable(_pending_rewards_db)
if DB_ENV == 'test':
child_db.truncate()
task_db.truncate()
reward_db.truncate()
image_db.truncate()
image_db.truncate()
pending_reward_db.truncate()