116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
import asyncio
|
||
import os
|
||
import sys
|
||
from importlib import import_module
|
||
from os.path import exists
|
||
|
||
from ariadne import load_schema_from_path, make_executable_schema
|
||
from ariadne.asgi import GraphQL
|
||
from starlette.applications import Starlette
|
||
from starlette.middleware import Middleware
|
||
from starlette.middleware.cors import CORSMiddleware
|
||
from starlette.requests import Request
|
||
from starlette.responses import JSONResponse, Response
|
||
from starlette.routing import Route
|
||
|
||
from cache.precache import precache_data
|
||
from cache.revalidator import revalidation_manager
|
||
from services.exception import ExceptionHandlerMiddleware
|
||
from services.redis import redis
|
||
from services.schema import create_all_tables, resolvers
|
||
from services.search import search_service
|
||
from services.viewed import ViewedStorage
|
||
from services.webhook import WebhookEndpoint, create_webhook_endpoint
|
||
from settings import DEV_SERVER_PID_FILE_NAME, MODE
|
||
|
||
import_module("resolvers")
|
||
schema = make_executable_schema(load_schema_from_path("schema/"), resolvers)
|
||
|
||
|
||
async def start():
|
||
if MODE == "development":
|
||
if not exists(DEV_SERVER_PID_FILE_NAME):
|
||
# pid file management
|
||
with open(DEV_SERVER_PID_FILE_NAME, "w", encoding="utf-8") as f:
|
||
f.write(str(os.getpid()))
|
||
print(f"[main] process started in {MODE} mode")
|
||
|
||
|
||
async def lifespan(_app):
|
||
try:
|
||
create_all_tables()
|
||
await asyncio.gather(
|
||
redis.connect(),
|
||
precache_data(),
|
||
ViewedStorage.init(),
|
||
create_webhook_endpoint(),
|
||
search_service.info(),
|
||
start(),
|
||
revalidation_manager.start(),
|
||
)
|
||
yield
|
||
finally:
|
||
tasks = [redis.disconnect(), ViewedStorage.stop(), revalidation_manager.stop()]
|
||
await asyncio.gather(*tasks, return_exceptions=True)
|
||
|
||
|
||
# Создаем экземпляр GraphQL
|
||
graphql_app = GraphQL(schema, debug=True)
|
||
|
||
|
||
# Оборачиваем GraphQL-обработчик для лучшей обработки ошибок
|
||
async def graphql_handler(request: Request):
|
||
if request.method not in ["GET", "POST"]:
|
||
return JSONResponse({"error": "Method Not Allowed"}, status_code=405)
|
||
|
||
try:
|
||
result = await graphql_app.handle_request(request)
|
||
if isinstance(result, Response):
|
||
return result
|
||
return JSONResponse(result)
|
||
except asyncio.CancelledError:
|
||
return JSONResponse({"error": "Request cancelled"}, status_code=499)
|
||
except Exception as e:
|
||
print(f"GraphQL error: {str(e)}")
|
||
return JSONResponse({"error": str(e)}, status_code=500)
|
||
|
||
middleware = [
|
||
# Начинаем с обработки ошибок
|
||
Middleware(ExceptionHandlerMiddleware),
|
||
# CORS должен быть перед другими middleware для корректной обработки preflight-запросов
|
||
Middleware(
|
||
CORSMiddleware,
|
||
allow_origins=[
|
||
"https://localhost:3000",
|
||
"https://testing.discours.io",
|
||
"https://testing3.discours.io",
|
||
"https://discours.io",
|
||
"https://new.discours.io"
|
||
],
|
||
allow_methods=["GET", "POST", "OPTIONS"], # Явно указываем OPTIONS
|
||
allow_headers=["*"],
|
||
allow_credentials=True,
|
||
),
|
||
]
|
||
|
||
# Обновляем маршрут в Starlette
|
||
app = Starlette(
|
||
routes=[
|
||
Route("/", graphql_handler, methods=["GET", "POST"]),
|
||
Route("/new-author", WebhookEndpoint),
|
||
],
|
||
middleware=middleware,
|
||
lifespan=lifespan,
|
||
debug=True,
|
||
)
|
||
|
||
app.add_middleware(ExceptionHandlerMiddleware)
|
||
if "dev" in sys.argv:
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["https://localhost:3000"],
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|