From f9a0e2bcc47cc15971fe4e7b1dfce9fde1c68e15 Mon Sep 17 00:00:00 2001 From: bdim404 Date: Wed, 11 Mar 2026 18:04:16 +0800 Subject: [PATCH] refactor: Simplify SQLite checks by introducing a variable for database type --- qwen3-tts-backend/db/database.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/qwen3-tts-backend/db/database.py b/qwen3-tts-backend/db/database.py index 254e1e0..f81b2eb 100644 --- a/qwen3-tts-backend/db/database.py +++ b/qwen3-tts-backend/db/database.py @@ -1,14 +1,18 @@ from sqlalchemy import create_engine, event from sqlalchemy.orm import sessionmaker, declarative_base +from sqlalchemy.pool import NullPool from config import settings +_is_sqlite = "sqlite" in settings.DATABASE_URL + engine = create_engine( 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") def _set_wal(dbapi_conn, _): dbapi_conn.execute("PRAGMA journal_mode=WAL")