refactor: 更新数据库和模型
- 更新数据库连接和会话管理 - 更新数据模型以支持用户认证 - 添加RefreshToken模型用于刷新令牌管理
This commit is contained in:
@@ -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})
|
||||
|
||||
Reference in New Issue
Block a user