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: