From 63ddaaa70dc7e2ed6c8853420c2c55d6854f0a62 Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Sun, 15 Mar 2026 14:40:32 +0100 Subject: [PATCH] chore: add stats DB cleanup and gitignore Co-Authored-By: Claude Opus 4.6 --- .gitignore | 1 + proxy.py | 7 +++++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 434ee8a..46c23a9 100644 --- a/.gitignore +++ b/.gitignore @@ -1 +1,2 @@ .worktrees +civitai_stats.db diff --git a/proxy.py b/proxy.py index fde6640..f5dfad8 100644 --- a/proxy.py +++ b/proxy.py @@ -625,11 +625,14 @@ async def lm_remote_proxy_middleware(request: web.Request, handler): async def _cleanup_proxy_session(app) -> None: - """Shutdown hook to close the shared proxy session.""" - global _proxy_session + """Shutdown hook to close the shared proxy session and stats DB.""" + global _proxy_session, _stats_db if _proxy_session and not _proxy_session.closed: await _proxy_session.close() _proxy_session = None + if _stats_db: + _stats_db.close() + _stats_db = None def register_proxy(app) -> None: