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

moving database files

Browse Source
This commit is contained in:
Ryan Kegel
2025-12-10 22:21:00 -05:00
parent 600e993dfd
commit 93c88ade01
1 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
db/db.py
View File

@@ -4,7 +4,16 @@ import threading
from tinydb import TinyDB
DB_ENV = os.environ.get('DB_ENV', 'prod')
base_dir = os.path.dirname(__file__)
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')
os.makedirs(base_dir, exist_ok=True)
class LockedTable:
@@ -66,19 +75,11 @@ class LockedTable:
# Setup DB files next to this module
if DB_ENV == 'test':
child_path = os.path.join(base_dir, 'test_children.json')
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')
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)