static Swagger files

Author: Kilo59

app/main.py

@@ -6,10 +6,11 @@
 import pydantic
 import sentry_sdk
 import uvicorn
-from fastapi import FastAPI, Request, Response
+from fastapi import FastAPI, Request, Response, openapi
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.gzip import GZipMiddleware
 from fastapi.responses import JSONResponse
+from fastapi.staticfiles import StaticFiles
 from scout_apm.async_.starlette import ScoutMiddleware
 from sentry_sdk.integrations.asgi import SentryAsgiMiddleware
 
@@ -35,8 +36,8 @@
         " Project page: https://github.com/ExpDev07/coronavirus-tracker-api."
     ),
     version="2.0.3",
-    docs_url="/",
-    redoc_url="/docs",
+    docs_url=None,
+    redoc_url=None,
     on_startup=[setup_client_session],
     on_shutdown=[teardown_client_session],
 )
@@ -108,6 +109,31 @@ async def handle_validation_error(
 # Include routers.
 APP.include_router(V1, prefix="", tags=["v1"])
 APP.include_router(V2, prefix="/v2", tags=["v2"])
+APP.mount("/static", StaticFiles(directory="static"), name="static")
+
+# ##############
+# Swagger/Redocs
+# ##############
+
+
+@APP.get("/", include_in_schema=False)
+async def custom_swagger_ui_html():
+    """Serve Swagger UI."""
+    return openapi.docs.get_swagger_ui_html(
+        openapi_url=APP.openapi_url,
+        title=f"{APP.title} - Swagger UI",
+        oauth2_redirect_url=APP.swagger_ui_oauth2_redirect_url,
+        swagger_js_url="/static/swagger-ui-bundle.js",
+        swagger_css_url="/static/swagger-ui.css",
+    )
+
+
+@APP.get("/docs", include_in_schema=False)
+async def redoc_html():
+    """Serve ReDoc UI."""
+    return openapi.docs.get_redoc_html(
+        openapi_url=APP.openapi_url, title=f"{APP.title} - ReDoc", redoc_js_url="/static/redoc.standalone.js",
+    )
 
 
 # Running of app.