API - refactor user decorators

fittrackee/users/auth.py

@@ -19,9 +19,9 @@ from werkzeug.exceptions import RequestEntityTooLarge
 from werkzeug.utils import secure_filename
 
 from ..workouts.utils_files import get_absolute_file_path
+from .decorators import authenticate
 from .models import User
 from .utils import (
-    authenticate,
     check_passwords,
     display_readable_file_size,
     get_readable_duration,

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

fittrackee/users/users.py

@@ -15,8 +15,8 @@ from flask import Blueprint, request, send_file
 from sqlalchemy import exc
 
 from ..workouts.utils_files import get_absolute_file_path
+from .decorators import authenticate, authenticate_as_admin
 from .models import User, Workout
-from .utils import authenticate, authenticate_as_admin
 
 users_blueprint = Blueprint('users', __name__)
 

fittrackee/users/utils.py

@@ -1,7 +1,6 @@
 import re
 from datetime import timedelta
-from functools import wraps
-from typing import Any, Callable, Optional, Tuple, Union
+from typing import Optional, Tuple, Union
 
 import humanize
 from fittrackee.responses import (
@@ -11,7 +10,7 @@ from fittrackee.responses import (
     PayloadTooLargeErrorResponse,
     UnauthorizedErrorResponse,
 )
-from flask import Request, current_app, request
+from flask import Request, current_app
 
 from .models import User
 
@@ -130,34 +129,6 @@ def verify_user(
     return None, resp
 
 
-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
-
-
 def can_view_workout(
     auth_user_id: int, workout_user_id: int
 ) -> Optional[HttpResponse]: