linted+fmt

services/auth.py

@@ -1,16 +1,16 @@
 from functools import wraps
 from typing import Tuple
 
+from sqlalchemy import exc
 from starlette.requests import Request
 
+from auth.internal import verify_internal_auth
+from auth.orm import Author, Role
 from cache.cache import get_cached_author_by_id
 from resolvers.stat import get_with_stat
-from utils.logger import root_logger as logger
-from auth.internal import verify_internal_auth
-from sqlalchemy import exc
 from services.db import local_session
-from auth.orm import Author, Role
 from settings import SESSION_TOKEN_HEADER
+from utils.logger import root_logger as logger
 
 # Список разрешенных заголовков
 ALLOWED_HEADERS = ["Authorization", "Content-Type"]
@@ -31,21 +31,21 @@ async def check_auth(req: Request) -> Tuple[str, list[str], bool]:
     - is_admin: bool - Флаг наличия у пользователя административных прав
     """
     logger.debug(f"[check_auth] Проверка авторизации...")
-    
+
     # Получаем заголовок авторизации
     token = None
-    
+
     # Проверяем заголовок с учетом регистра
     headers_dict = dict(req.headers.items())
     logger.debug(f"[check_auth] Все заголовки: {headers_dict}")
-    
+
     # Ищем заголовок Authorization независимо от регистра
     for header_name, header_value in headers_dict.items():
         if header_name.lower() == SESSION_TOKEN_HEADER.lower():
             token = header_value
             logger.debug(f"[check_auth] Найден заголовок {header_name}: {token[:10]}...")
             break
-    
+
     if not token:
         logger.debug(f"[check_auth] Токен не найден в заголовках")
         return "", [], False
@@ -57,8 +57,10 @@ async def check_auth(req: Request) -> Tuple[str, list[str], bool]:
     # Проверяем авторизацию внутренним механизмом
     logger.debug("[check_auth] Вызов verify_internal_auth...")
     user_id, user_roles, is_admin = await verify_internal_auth(token)
-    logger.debug(f"[check_auth] Результат verify_internal_auth: user_id={user_id}, roles={user_roles}, is_admin={is_admin}")
-    
+    logger.debug(
+        f"[check_auth] Результат verify_internal_auth: user_id={user_id}, roles={user_roles}, is_admin={is_admin}"
+    )
+
     # Если в ролях нет админа, но есть ID - проверяем в БД
     if user_id and not is_admin:
         try:
@@ -71,16 +73,19 @@ async def check_auth(req: Request) -> Tuple[str, list[str], bool]:
                 else:
                     # Проверяем наличие админских прав через БД
                     from auth.orm import AuthorRole
-                    admin_role = session.query(AuthorRole).filter(
-                        AuthorRole.author == user_id_int,
-                        AuthorRole.role.in_(["admin", "super"])
-                    ).first()
+
+                    admin_role = (
+                        session.query(AuthorRole)
+                        .filter(AuthorRole.author == user_id_int, AuthorRole.role.in_(["admin", "super"]))
+                        .first()
+                    )
                     is_admin = admin_role is not None
         except Exception as e:
             logger.error(f"Ошибка при проверке прав администратора: {e}")
-    
+
     return user_id, user_roles, is_admin
 
+
 async def add_user_role(user_id: str, roles: list[str] = None):
     """
     Добавление ролей пользователю в локальной БД.
@@ -131,32 +136,32 @@ def login_required(f):
 
         info = args[1]
         req = info.context.get("request")
-        
+
         logger.debug(f"[login_required] Проверка авторизации для запроса: {req.method} {req.url.path}")
         logger.debug(f"[login_required] Заголовки: {req.headers}")
-        
+
         user_id, user_roles, is_admin = await check_auth(req)
-        
+
         if not user_id:
             logger.debug(f"[login_required] Пользователь не авторизован, {dict(req)}, {info}")
             raise GraphQLError("Требуется авторизация")
-            
+
         # Проверяем наличие роли reader
-        if 'reader' not in user_roles:
+        if "reader" not in user_roles:
             logger.error(f"Пользователь {user_id} не имеет роли 'reader'")
             raise GraphQLError("У вас нет необходимых прав для доступа")
-            
+
         logger.info(f"Авторизован пользователь {user_id} с ролями: {user_roles}")
         info.context["roles"] = user_roles
-        
+
         # Проверяем права администратора
         info.context["is_admin"] = is_admin
-        
+
         author = await get_cached_author_by_id(user_id, get_with_stat)
         if not author:
             logger.error(f"Профиль автора не найден для пользователя {user_id}")
         info.context["author"] = author
-            
+
         return await f(*args, **kwargs)
 
     return decorated_function
@@ -177,7 +182,7 @@ def login_accepted(f):
         if user_id and user_roles:
             logger.info(f"login_accepted: Пользователь авторизован: {user_id} с ролями {user_roles}")
             info.context["roles"] = user_roles
-            
+
             # Проверяем права администратора
             info.context["is_admin"] = is_admin