klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2024-11-24 10:04:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix missing HTTP_REFERER

Browse Source
This commit is contained in:
thomas girod 2024-08-05 15:53:41 +02:00
parent e37ce4172e
commit f6fbad8403
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
counter/utils.py
View File

@ -23,13 +23,12 @@ def is_logged_in_counter(request: HttpRequest) -> bool:
- The current session has a counter token associated with it. - The current session has a counter token associated with it.
- A counter with this token exists. - A counter with this token exists.
""" """
referer = urlparse(request.META["HTTP_REFERER"]).path referer_ok = (
path_ok = ( "HTTP_REFERER" in request.META
request.resolver_match.app_name == "counter" and resolve(urlparse(request.META["HTTP_REFERER"]).path).app_name == "counter"
or resolve(referer).app_name == "counter"
) )
return ( return (
path_ok (referer_ok or request.resolver_match.app_name == "counter")
and "counter_token" in request.session and "counter_token" in request.session
and request.session["counter_token"] and request.session["counter_token"]
and Counter.objects.filter(token=request.session["counter_token"]).exists() and Counter.objects.filter(token=request.session["counter_token"]).exists()