28 lines
951 B
Python
28 lines
951 B
Python
# python
|
|
# file: config/paths.py
|
|
import os
|
|
|
|
# Constant directory names
|
|
DATA_DIR_NAME = 'data'
|
|
TEST_DATA_DIR_NAME = 'test_data'
|
|
|
|
# Project root (two levels up from this file)
|
|
PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
def get_database_dir(db_env: str | None = None) -> str:
|
|
"""
|
|
Return the absolute base directory path for the given DB env.
|
|
db_env: 'prod' uses `data/db`, anything else uses `test_data/db`.
|
|
"""
|
|
env = (db_env or os.environ.get('DB_ENV', 'prod')).lower()
|
|
base_name = DATA_DIR_NAME if env == 'prod' else TEST_DATA_DIR_NAME
|
|
return os.path.join(PROJECT_ROOT, base_name, 'db')
|
|
|
|
def get_user_image_dir(username: str | None) -> str:
|
|
"""
|
|
Return the absolute directory path for storing images for a specific user.
|
|
"""
|
|
if username:
|
|
return os.path.join(PROJECT_ROOT, DATA_DIR_NAME, 'images', username)
|
|
return os.path.join(PROJECT_ROOT, 'resources', 'images')
|