From 386c9c1a88cf9fe1501fa3f494aed8810cb3826e Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Wed, 15 Sep 2021 07:51:05 +0200 Subject: [PATCH] fd --- files/__main__.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/files/__main__.py b/files/__main__.py index 426167f30..b839ad9dd 100644 --- a/files/__main__.py +++ b/files/__main__.py @@ -114,7 +114,6 @@ class RetryingQuery(_Query): def first(self): return super().first() -db_session=scoped_session(sessionmaker(bind=_engine, query_cls=RetryingQuery)) Base = declarative_base() @@ -139,7 +138,7 @@ def before_request(): if app.config["BOT_DISABLE"] and request.headers.get("X-User-Type")=="Bot": abort(503) - g.db = db_session() + g.db = scoped_session(sessionmaker(bind=_engine, query_cls=RetryingQuery)) g.timestamp = int(time.time()) @@ -172,7 +171,7 @@ def before_request(): @app.after_request def after_request(response): - if g.db: + if hasattr(g, 'db') and g.db: g.db.commit() g.db.close()