refactor: format

Signed-off-by: Adam Tao <tcx4c70@gmail.com>
This commit is contained in:
Adam Tao
2025-07-19 19:20:53 +08:00
parent 635cb8e3ff
commit 7bd7559bfe
3 changed files with 5 additions and 3 deletions

View File

@@ -115,6 +115,7 @@ elif "sqlite" in SQLALCHEMY_DATABASE_URL:
engine = create_engine(
SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
)
def on_connect(dbapi_connection, connection_record):
cursor = dbapi_connection.cursor()
if DATABASE_ENABLE_SQLITE_WAL:
@@ -122,6 +123,7 @@ elif "sqlite" in SQLALCHEMY_DATABASE_URL:
else:
cursor.execute("PRAGMA journal_mode=DELETE")
cursor.close()
event.listen(engine, "connect", on_connect)
else:
if isinstance(DATABASE_POOL_SIZE, int):