初始版本

backend/app/db/session.py

@@ -0,0 +1,26 @@
+"""
+数据库会话管理
+"""
+from sqlalchemy.engine import Engine
+from sqlalchemy import create_engine
+from sqlmodel import Session
+from typing import Generator
+from app.core.config import settings
+
+# 创建数据库引擎
+engine: Engine = create_engine(
+    settings.DATABASE_URL,
+    connect_args={"check_same_thread": False},  # SQLite 需要此参数
+    echo=True  # 开发环境显示SQL语句，生产环境设为False
+)
+
+
+def get_db() -> Generator[Session, None, None]:
+    """获取数据库会话（用于依赖注入）"""
+    with Session(engine) as session:
+        yield session
+
+
+# 为了向后兼容，保留 SessionLocal 别名
+SessionLocal = Session
+