forked from rDrama/rDrama
1
0
Fork 0

don't commit if error happens

master
Aevann1 2022-06-27 07:24:11 +02:00
parent 5186b9a3f7
commit fd75733831
1 changed files with 6 additions and 1 deletions

View File

@ -95,13 +95,18 @@ def before_request():
def after_request(response):
response.headers.add("Strict-Transport-Security", "max-age=31536000")
response.headers.add("X-Frame-Options", "deny")
if response.status_code < 400:
g.db.commit()
g.db.close()
del g.db
return response
@app.teardown_appcontext
def teardown_request(error):
if hasattr(g, 'db') and g.db:
g.db.commit()
g.db.rollback()
g.db.close()
del g.db
stdout.flush()
if app.config["SERVER_NAME"] == 'localhost':