nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-04-24 18:52:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

添加对 PostgreSQL 的支持

Browse Source
This commit is contained in:
jxxghp
2025-08-18 11:19:17 +08:00
parent e8058c8813
commit b6cf54d57f
23 changed files with 304 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
database/env.py
View File

@@ -40,13 +40,25 @@ def run_migrations_offline() -> None:
"""
url = config.get_main_option("sqlalchemy.url")
context.configure(
url=url,
target_metadata=target_metadata,
literal_binds=True,
dialect_opts={"paramstyle": "named"},
render_as_batch=True
)
# 根据数据库类型配置不同的参数
if url and "postgresql" in url:
# PostgreSQL配置
context.configure(
url=url,
target_metadata=target_metadata,
literal_binds=True,
dialect_opts={"paramstyle": "named"},
)
else:
# SQLite配置
context.configure(
url=url,
target_metadata=target_metadata,
literal_binds=True,
dialect_opts={"paramstyle": "named"},
render_as_batch=True
)
with context.begin_transaction():
context.run_migrations()
@@ -66,9 +78,22 @@ def run_migrations_online() -> None:
)
with connectable.connect() as connection:
context.configure(
connection=connection, target_metadata=target_metadata
)
url = config.get_main_option("sqlalchemy.url")
# 根据数据库类型配置不同的参数
if url and "postgresql" in url:
# PostgreSQL配置
context.configure(
connection=connection,
target_metadata=target_metadata
)
else:
# SQLite配置
context.configure(
connection=connection,
target_metadata=target_metadata,
render_as_batch=True
)
with context.begin_transaction():
context.run_migrations()