Mawoka
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from fastapi import FastAPI, Request
|
|
import sentry_sdk
|
|
from socketio import ASGIApp
|
|
|
|
from classquiz.db import database
|
|
from classquiz.config import settings
|
|
from classquiz.routers import users, quiz, utils, stats, storage
|
|
from classquiz.socket_server import sio
|
|
from sentry_sdk.integrations.redis import RedisIntegration
|
|
|
|
settings = settings()
|
|
if settings.sentry_dsn:
|
|
sentry_sdk.init(dsn=settings.sentry_dsn, integrations=[RedisIntegration()])
|
|
app = FastAPI(redoc_url="", docs_url="/api/docs")
|
|
app.state.database = database
|
|
|
|
|
|
@app.middleware("http")
|
|
async def sentry_exception(request: Request, call_next):
|
|
try:
|
|
response = await call_next(request)
|
|
return response
|
|
except Exception as e:
|
|
with sentry_sdk.push_scope() as scope:
|
|
scope.set_context("request", request)
|
|
sentry_sdk.capture_exception(e)
|
|
raise e
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def startup() -> None:
|
|
database_ = app.state.database
|
|
if not database_.is_connected:
|
|
await database_.connect()
|
|
|
|
|
|
@app.on_event("shutdown")
|
|
async def shutdown() -> None:
|
|
database_ = app.state.database
|
|
if database_.is_connected:
|
|
await database_.disconnect()
|
|
|
|
|
|
|
|
|
|
app.include_router(users.router, tags=["users"], prefix="/api/v1/users")
|
|
app.include_router(quiz.router, tags=["quiz"], prefix="/api/v1/quiz")
|
|
app.include_router(utils.router, tags=["utils"], prefix="/api/v1/utils")
|
|
app.include_router(stats.router, tags=["stats"], prefix="/api/v1/stats")
|
|
app.include_router(storage.router, tags=["storage"], prefix="/api/v1/storage")
|
|
app.mount("/", ASGIApp(sio))
|