Alexander/FitTrackee
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

API - minor refactoring

Browse Source
This commit is contained in:
Sam 2021-12-01 19:44:57 +01:00
parent c4d19089e4
commit 1697f6051f
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
fittrackee/users/decorators.py
View File

@ -8,15 +8,22 @@ from fittrackee.responses import HttpResponse
from .utils import verify_user from .utils import verify_user
def verify_auth_user(
f: Callable, verify_admin: bool, *args: Any, **kwargs: Any
) -> Union[Callable, HttpResponse]:
response_object, user = verify_user(request, verify_admin=verify_admin)
if response_object:
return response_object
return f(user, *args, **kwargs)
def authenticate(f: Callable) -> Callable: def authenticate(f: Callable) -> Callable:
@wraps(f) @wraps(f)
def decorated_function( def decorated_function(
*args: Any, **kwargs: Any *args: Any, **kwargs: Any
) -> Union[Callable, HttpResponse]: ) -> Union[Callable, HttpResponse]:
response_object, user = verify_user(request, verify_admin=False) verify_admin = False
if response_object: return verify_auth_user(f, verify_admin, *args, **kwargs)
return response_object
return f(user, *args, **kwargs)
return decorated_function return decorated_function
@ -26,9 +33,7 @@ def authenticate_as_admin(f: Callable) -> Callable:
def decorated_function( def decorated_function(
*args: Any, **kwargs: Any *args: Any, **kwargs: Any
) -> Union[Callable, HttpResponse]: ) -> Union[Callable, HttpResponse]:
response_object, user = verify_user(request, verify_admin=True) verify_admin = True
if response_object: return verify_auth_user(f, verify_admin, *args, **kwargs)
return response_object
return f(user, *args, **kwargs)
return decorated_function return decorated_function