bdim/Canto
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor: Simplify SQLite checks by introducing a variable for database type

Browse Source
This commit is contained in:
bdim404
2026-03-11 18:04:16 +08:00
parent d9082b12a8
commit f9a0e2bcc4
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
qwen3-tts-backend/db/database.py
View File

@@ -1,14 +1,18 @@
from sqlalchemy import create_engine, event from sqlalchemy import create_engine, event
from sqlalchemy.orm import sessionmaker, declarative_base from sqlalchemy.orm import sessionmaker, declarative_base
from sqlalchemy.pool import NullPool
from config import settings from config import settings
_is_sqlite = "sqlite" in settings.DATABASE_URL
engine = create_engine( engine = create_engine(
settings.DATABASE_URL, settings.DATABASE_URL,
connect_args={"check_same_thread": False} if "sqlite" in settings.DATABASE_URL else {} connect_args={"check_same_thread": False} if _is_sqlite else {},
poolclass=NullPool if _is_sqlite else None,
) )
if "sqlite" in settings.DATABASE_URL: if _is_sqlite:
@event.listens_for(engine, "connect") @event.listens_for(engine, "connect")
def _set_wal(dbapi_conn, _): def _set_wal(dbapi_conn, _):
dbapi_conn.execute("PRAGMA journal_mode=WAL") dbapi_conn.execute("PRAGMA journal_mode=WAL")