27 lines
670 B
Python
27 lines
670 B
Python
"""
|
|
数据库会话管理
|
|
"""
|
|
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
|
|
|