57 lines
2.7 KiB
Python
57 lines
2.7 KiB
Python
from ariadne.asgi.handlers import GraphQLHTTPHandler
|
||
from starlette.requests import Request
|
||
from starlette.responses import JSONResponse
|
||
|
||
from auth.middleware import auth_middleware
|
||
from utils.logger import root_logger as logger
|
||
|
||
|
||
class EnhancedGraphQLHTTPHandler(GraphQLHTTPHandler):
|
||
"""
|
||
Улучшенный GraphQL HTTP обработчик с поддержкой cookie и авторизации.
|
||
|
||
Расширяет стандартный GraphQLHTTPHandler для:
|
||
1. Создания расширенного контекста запроса с авторизационными данными
|
||
2. Корректной обработки ответов с cookie и headers
|
||
3. Интеграции с AuthMiddleware
|
||
"""
|
||
|
||
async def get_context_for_request(self, request: Request, data: dict) -> dict:
|
||
"""
|
||
Расширяем контекст для GraphQL запросов.
|
||
|
||
Добавляет к стандартному контексту:
|
||
- Объект response для установки cookie
|
||
- Интеграцию с AuthMiddleware
|
||
- Расширения для управления авторизацией
|
||
|
||
Args:
|
||
request: Starlette Request объект
|
||
data: данные запроса
|
||
|
||
Returns:
|
||
dict: контекст с дополнительными данными для авторизации и cookie
|
||
"""
|
||
# Получаем стандартный контекст от базового класса
|
||
context = await super().get_context_for_request(request, data)
|
||
|
||
# Создаем объект ответа для установки cookie
|
||
response = JSONResponse({})
|
||
context["response"] = response
|
||
|
||
# Интегрируем с AuthMiddleware
|
||
auth_middleware.set_context(context)
|
||
context["extensions"] = auth_middleware
|
||
|
||
# Добавляем данные авторизации только если они доступны
|
||
# Проверяем наличие данных авторизации в scope
|
||
if hasattr(request, "scope") and isinstance(request.scope, dict) and "auth" in request.scope:
|
||
auth_cred = request.scope.get("auth")
|
||
context["auth"] = auth_cred
|
||
# Безопасно логируем информацию о типе объекта auth
|
||
logger.debug(f"[graphql] Добавлены данные авторизации в контекст из scope: {type(auth_cred).__name__}")
|
||
|
||
logger.debug("[graphql] Подготовлен расширенный контекст для запроса")
|
||
|
||
return context
|