nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-03-20 03:57:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix db

Browse Source
This commit is contained in:
jxxghp
2025-08-19 11:40:34 +08:00
parent aeb297efcf
commit 15833f94cf
16 changed files with 22 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/db/__init__.py
View File

@@ -9,13 +9,16 @@ from sqlalchemy.orm import Session, as_declarative, declared_attr, scoped_sessio
from app.core.config import settings
def get_id_column():
def get_id_column(table_name: str = None):
"""
根据数据库类型返回合适的ID列定义
"""
if settings.DB_TYPE.lower() == "postgresql":
# PostgreSQL使用SERIAL类型
return Column(Integer, primary_key=True, index=True)
# PostgreSQL使用显式序列，确保序列正确创建
if table_name:
return Column(Integer, Sequence(f'{table_name}_id_seq'), primary_key=True, index=True)
else:
return Column(Integer, Sequence('id_seq'), primary_key=True, index=True)
else:
# SQLite使用Sequence
return Column(Integer, Sequence('id'), primary_key=True, index=True)