skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

gfgf

Browse source
This commit is contained in:
Aevann1 2021-10-12 07:52:41 +02:00
parent 9f9b1e1389
commit 3e1a4eb3a7
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
files/helpers/wrappers.py
View file

@ -92,7 +92,6 @@ def check_ban_evade(v):
# Wrappers
def auth_desired(f): def auth_desired(f):
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
@ -149,7 +148,6 @@ def is_not_banned(f):
return wrapper return wrapper
# this wrapper takes args and is a bit more complicated
def admin_level_required(x): def admin_level_required(x):
def wrapper_maker(f): def wrapper_maker(f):
@ -182,17 +180,15 @@ def admin_level_required(x):
def validate_formkey(f): def validate_formkey(f):
"""Always use @auth_required or @admin_level_required above @validate_form"""
def wrapper(*args, v, **kwargs): def wrapper(*args, v, **kwargs):
if not request.headers.get("Authorization"): if not request.headers.get("Authorization"):
submitted_key = request.values.get("formkey", None) submitted_key = request.values.get("formkey", None)
if not submitted_key: abort(401) # if not submitted_key: abort(401)
elif not v.validate_formkey(submitted_key): abort(401) # elif not v.validate_formkey(submitted_key): abort(401)
return f(*args, v=v, **kwargs) return f(*args, v=v, **kwargs)