rDrama
/
rDrama
9
10
Fork 7
Code Issues Pull Requests Packages Releases Wiki Activity

move cookie domain setting to before_request

pull/50/head
justcool393 2022-12-06 14:24:18 -06:00
parent 8038b5ca05
commit f8cfe7cf82
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
files/routes/allroutes.py
View File

@ -14,6 +14,9 @@ def session_init():
@app.before_request
def before_request():
g.desires_auth = False
if not IS_LOCALHOST:
app.config["COOKIE_DOMAIN"] = f".{request.host}"
app.config["SESSION_COOKIE_DOMAIN"] = app.config["COOKIE_DOMAIN"]
if SITE == 'marsey.world' and request.path != '/kofi':
abort(404)
@ -85,7 +88,7 @@ def _fix_frozen_sessions(response:Response) -> None:
domain = app.config["SESSION_COOKIE_DOMAIN"]
if IS_LOCALHOST or not '.' in domain: return # "dotless" domains in general aren't really supportable
bad_domain = f'.{domain}'
bad_domain = f'{domain}'
cookie_header = request.headers.get("Cookie")
response.delete_cookie(app.config["SESSION_COOKIE_NAME"], domain=bad_domain, httponly=True, secure=True)
if not cookie_header or not f'domain={bad_domain}' in cookie_header: return