standardize g.db behavior

remotes/1693176582716663532/tmp_refs/heads/watchparty
Aevann1 2022-11-15 17:39:00 +02:00
parent f1288ba991
commit 532ebd3ac8
4 changed files with 11 additions and 4 deletions

View File

@ -51,6 +51,7 @@ def cron(every_5m, every_1h, every_1d, every_1mo):
g.db.commit()
g.db.close()
del g.db
stdout.flush()
def _sub_inactive_purge_task():

View File

@ -43,9 +43,10 @@ def after_request(response):
if CLOUDFLARE_AVAILABLE and CLOUDFLARE_COOKIE_VALUE and getattr(g, 'desires_auth', False):
logged_in = bool(getattr(g, 'v', None))
response.set_cookie("lo", CLOUDFLARE_COOKIE_VALUE if logged_in else '', max_age=60*60*24*365 if logged_in else 1)
g.db.commit()
g.db.close()
del g.db
if getattr(g, 'db', None):
g.db.commit()
g.db.close()
del g.db
return response
@app.teardown_appcontext

View File

@ -48,7 +48,10 @@ def error_401(e):
@app.errorhandler(500)
def error_500(e):
g.db.rollback()
if getattr(g, 'db', None):
g.db.rollback()
g.db.close()
del g.db
return error(e)

View File

@ -32,6 +32,8 @@ def authorize(v):
g.db.add(new_auth)
except sqlalchemy.exc.IntegrityError:
g.db.rollback()
g.db.close()
del g.db
old_auth = g.db.query(ClientAuth).filter_by(oauth_client = application.id, user_id = v.id).one()
access_token = old_auth.access_token