web.py

import uvicorn

from pathlib import Path
from fastapi import FastAPI, APIRouter
from starlette.responses import FileResponse


class WebAppWrapper:

    def __init__(self, res_path: Path, port: int = 5000):
        self.router = APIRouter()
        self.res_path = res_path
        self.port = port
        self.app = None
        self.srv = None

    def run(self) -> uvicorn.Server:
        self.app = FastAPI(title='TGBF2')
        self.app.include_router(self.router)

        @self.app.exception_handler(404)
        async def ex(req, exc): return FileResponse(self.res_path / '404.html')

        @self.app.get('/', include_in_schema=False)
        async def root(): return FileResponse(self.res_path / 'root.html')

        self.srv = uvicorn.Server(
            uvicorn.Config(self.app, host='0.0.0.0', port=self.port)
        )

        return self.srv

    def add_endpoint(self, path, endpoint):
        if self.app:
            self.app.add_api_route(path, endpoint)
        else:
            self.router.add_api_route(path, endpoint)

    def remove_endpoint(self, path):
        for route in self.router.routes:
            if route.path == path:
                if self.app:
                    self.app.routes.remove(route)
                else:
                    self.router.routes.remove(route)