refactor: 更新数据库和模型

- 更新数据库连接和会话管理
- 更新数据模型以支持用户认证
- 添加RefreshToken模型用于刷新令牌管理
This commit is contained in:
徐在坤
2026-01-18 15:57:47 +08:00
parent 9132447db7
commit 3a4c9f0838
3 changed files with 40 additions and 4 deletions

View File

@@ -9,8 +9,24 @@ from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sess
from .models import Base
# 数据库文件路径
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./life_echo.db")
ASYNC_DATABASE_URL = DATABASE_URL.replace("sqlite://", "sqlite+aiosqlite://")
# 从环境变量获取,如果格式不正确则使用默认值
raw_database_url = os.getenv("DATABASE_URL", "sqlite:///./life_echo.db")
# 处理数据库 URL
# 如果已经是异步格式,需要提取同步格式用于同步引擎
if raw_database_url.startswith("sqlite+aiosqlite://"):
# 提取文件路径(移除协议部分)
file_path = raw_database_url.replace("sqlite+aiosqlite://", "")
DATABASE_URL = f"sqlite://{file_path}"
ASYNC_DATABASE_URL = raw_database_url
elif raw_database_url.startswith("sqlite://"):
DATABASE_URL = raw_database_url
ASYNC_DATABASE_URL = raw_database_url.replace("sqlite://", "sqlite+aiosqlite://")
else:
# 如果格式不正确,使用默认值并打印警告
print(f"警告: DATABASE_URL 格式不正确 ({raw_database_url}),使用默认值")
DATABASE_URL = "sqlite:///./life_echo.db"
ASYNC_DATABASE_URL = "sqlite+aiosqlite:///./life_echo.db"
# 创建同步引擎(用于迁移等)
engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})