数据库位置迁移

This commit is contained in:
2026-01-04 09:23:18 +00:00
parent 04d05cd1fe
commit 55997cfe29
2 changed files with 9 additions and 5 deletions

View File

@@ -4,7 +4,13 @@ from models import Base
import os
# 数据库配置
DATABASE_URL = os.getenv('DATABASE_URL', 'sqlite:///worklist.db')
# 确保 data 目录存在
DATA_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'data')
os.makedirs(DATA_DIR, exist_ok=True)
# 数据库文件路径
DB_PATH = os.path.join(DATA_DIR, 'worklist.db')
DATABASE_URL = os.getenv('DATABASE_URL', f'sqlite:///{DB_PATH}')
# 创建引擎
engine = create_engine(