Mawoka
110 lines
4.0 KiB
Python
110 lines
4.0 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
import asyncio
|
|
|
|
import sentry_sdk
|
|
from fastapi import FastAPI, Request
|
|
from sentry_sdk.integrations.redis import RedisIntegration
|
|
from socketio import ASGIApp
|
|
from starlette.middleware.sessions import SessionMiddleware
|
|
|
|
from classquiz.config import settings
|
|
from classquiz.db import database
|
|
from datetime import timedelta
|
|
|
|
from classquiz.oauth import rememberme_middleware
|
|
from classquiz.routers import (
|
|
users,
|
|
quiz,
|
|
utils,
|
|
stats,
|
|
storage,
|
|
search,
|
|
testing_routes,
|
|
editor,
|
|
live,
|
|
eximport,
|
|
login,
|
|
sitemap,
|
|
remote,
|
|
community,
|
|
avatar,
|
|
results,
|
|
admin,
|
|
)
|
|
from classquiz.socket_server import sio
|
|
from classquiz.helpers import meilisearch_init, telemetry_ping, bg_tasks
|
|
from scheduler.asyncio import Scheduler
|
|
|
|
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
|
|
|
|
|
|
async def background_tasks():
|
|
schedule = Scheduler()
|
|
schedule.cyclic(timedelta(hours=6), bg_tasks.clean_editor_images_up)
|
|
while True:
|
|
await asyncio.sleep(1)
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def startup() -> None:
|
|
database_ = app.state.database
|
|
if not database_.is_connected:
|
|
await database_.connect()
|
|
await meilisearch_init()
|
|
await telemetry_ping()
|
|
asyncio.create_task(background_tasks())
|
|
|
|
|
|
@app.on_event("shutdown")
|
|
async def shutdown() -> None:
|
|
database_ = app.state.database
|
|
if database_.is_connected:
|
|
await database_.disconnect()
|
|
|
|
|
|
@app.middleware("http")
|
|
async def auth_middleware_wrapper(request: Request, call_next):
|
|
return await rememberme_middleware(request, call_next)
|
|
|
|
|
|
app.include_router(results.router, tags=["results"], prefix="/api/v1/results", include_in_schema=True)
|
|
app.include_router(remote.router, tags=["remote"], prefix="/api/v1/remote", include_in_schema=True)
|
|
app.include_router(login.router, tags=["auth"], prefix="/api/v1/login", include_in_schema=True)
|
|
|
|
app.add_middleware(SessionMiddleware, secret_key=settings.secret_key)
|
|
app.include_router(users.router, tags=["users"], prefix="/api/v1/users", include_in_schema=True)
|
|
app.include_router(quiz.router, tags=["quiz"], prefix="/api/v1/quiz", include_in_schema=True)
|
|
app.include_router(utils.router, tags=["utils"], prefix="/api/v1/utils", include_in_schema=True)
|
|
app.include_router(stats.router, tags=["stats"], prefix="/api/v1/stats", include_in_schema=True)
|
|
app.include_router(storage.router, tags=["storage"], prefix="/api/v1/storage", include_in_schema=True)
|
|
app.include_router(search.router, tags=["search"], prefix="/api/v1/search", include_in_schema=True),
|
|
app.include_router(live.router, tags=["live"], prefix="/api/v1/live", include_in_schema=True)
|
|
app.include_router(
|
|
testing_routes.router, tags=["internal", "testing"], prefix="/api/v1/internal/testing", include_in_schema=False
|
|
)
|
|
app.include_router(editor.router, tags=["editor"], prefix="/api/v1/editor", include_in_schema=True)
|
|
app.include_router(eximport.router, tags=["export", "import"], prefix="/api/v1/eximport", include_in_schema=True)
|
|
app.include_router(sitemap.router, tags=["sitemap"], prefix="/api/v1/sitemap", include_in_schema=True)
|
|
app.include_router(community.router, tags=["community"], prefix="/api/v1/community", include_in_schema=True)
|
|
app.include_router(avatar.router, tags=["avatar"], prefix="/api/v1/avatar", include_in_schema=True)
|
|
app.include_router(admin.router, tags=["admin"], prefix="/api/v1/admin", include_in_schema=True)
|
|
app.mount("/", ASGIApp(sio))
|