API - refactor user decorators

fittrackee/users/decorators.py

@@ -0,0 +1,35 @@
+from functools import wraps
+from typing import Any, Callable, Union
+
+from fittrackee.responses import HttpResponse
+from flask import request
+
+from .utils import verify_user
+
+
+def authenticate(f: Callable) -> Callable:
+    @wraps(f)
+    def decorated_function(
+        *args: Any, **kwargs: Any
+    ) -> Union[Callable, HttpResponse]:
+        verify_admin = False
+        response_object, resp = verify_user(request, verify_admin)
+        if response_object:
+            return response_object
+        return f(resp, *args, **kwargs)
+
+    return decorated_function
+
+
+def authenticate_as_admin(f: Callable) -> Callable:
+    @wraps(f)
+    def decorated_function(
+        *args: Any, **kwargs: Any
+    ) -> Union[Callable, HttpResponse]:
+        verify_admin = True
+        response_object, resp = verify_user(request, verify_admin)
+        if response_object:
+            return response_object
+        return f(resp, *args, **kwargs)
+
+    return decorated_function