minor-fixes

services/auth.py

@@ -1,22 +1,14 @@
-from functools import wraps
-import aiohttp
-
-from orm.author import Author
-from services.db import local_session
-from settings import AUTH_URL
-
-
 from functools import wraps
 import aiohttp
 from aiohttp.web import HTTPUnauthorized
-
+from orm.author import Author
+from services.db import local_session
 from settings import AUTH_URL
 
 
 async def check_auth(req) -> (bool, int | None):
     token = req.headers.get("Authorization")
     if token:
-        # Logging the authentication token
         print(f"[services.auth] checking auth token: {token}")
         query_name = "validate_jwt_token"
         operation = "ValidateToken"
@@ -38,15 +30,12 @@ async def check_auth(req) -> (bool, int | None):
         }
         print(f"[services.auth] Graphql: {gql}")
         try:
-            # Asynchronous HTTP request to the authentication server
             async with aiohttp.ClientSession() as session:
                 async with session.post(AUTH_URL, json=gql, headers=headers) as response:
-                    # Logging the GraphQL response
                     response_text = await response.text()
                     print(f"[services.auth] GraphQL Response: {response_text}")
 
                     if response.status == 200:
-                        # Parsing JSON response
                         data = await response.json()
                         errors = data.get("errors")
                         if errors:
@@ -55,18 +44,14 @@ async def check_auth(req) -> (bool, int | None):
                             user_id = data.get("data", {}).get(query_name, {}).get("claims", {}).get("sub")
 
                             if user_id:
-                                # Logging the retrieved user ID
                                 print(f"[services.auth] User ID retrieved: {user_id}")
                                 return True, user_id
                             else:
-                                # Logging when no user ID is found in the response
                                 print("[services.auth] No user ID found in the response")
                     else:
-                        # Logging when the request to the authentication server fails
                         print(f"[services.auth] Request failed with status: {response.status}")
 
         except Exception as e:
-            # Handling and logging exceptions during authentication check
             print(f"[services.auth] {e}")
 
     return False, None
@@ -75,16 +60,13 @@ async def check_auth(req) -> (bool, int | None):
 def login_required(f):
     @wraps(f)
     async def decorated_function(*args, **kwargs):
-        print(args)
         info = args[1]
         context = info.context
         req = context.get("request")
         is_authenticated, user_id = await check_auth(req)
         if not is_authenticated:
-            # Raising HTTPUnauthorized exception if the user is not authenticated
             raise HTTPUnauthorized(text="Please, login first")
         else:
-            # Добавляем author_id и user_id в контекст
             with local_session() as session:
                 author = session.query(Author).filter(Author.user == user_id).first()
                 if author:
@@ -92,7 +74,6 @@ def login_required(f):
                 if user_id:
                     context["user_id"] = user_id
 
-        # Если пользователь аутентифицирован, выполняем резолвер
         return await f(*args, **kwargs)
 
     return decorated_function