move swagger controller to main

Author: dpgiakatos

controllers/root_controller.py

@@ -1,6 +1,5 @@
 from fastapi import APIRouter
 from fastapi.responses import RedirectResponse
-from fastapi.openapi.docs import get_swagger_ui_html
 from dotenv import load_dotenv
 import os
 
@@ -18,15 +17,3 @@ class RootController:
     @router.get("/", include_in_schema=False)
     def get_root():
         return RedirectResponse(url="/docs" if PROXY_PATH is None else f"/{PROXY_PATH}/docs")
-    
-    @staticmethod
-    @router.get("/docs", include_in_schema=False)
-    @router.get("/docs/", include_in_schema=False)
-    def get_swagger_ui_html():
-        return get_swagger_ui_html(
-            openapi_url=router.openapi_url,
-            title=router.title,
-            swagger_js_url="https://unpkg.com/swagger-ui-dist@5/swagger-ui-bundle.js",
-            swagger_css_url="https://unpkg.com/swagger-ui-dist@5/swagger-ui.css",
-        )
-

main.py

@@ -2,6 +2,7 @@
 import pkgutil
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.openapi.docs import get_swagger_ui_html
 from controllers import __path__ as controllers_path
 from dotenv import load_dotenv
 import os
@@ -33,6 +34,7 @@
     title="IHR API",
     description=description,
     version="v1.11",
+    docs_url=None,
     redoc_url=None,
     redirect_slashes=False,
 )
@@ -43,6 +45,16 @@
     if hasattr(module, "router"):
         app.include_router(module.router)
 
+@app.get("/docs", include_in_schema=False)
+@app.get("/docs/", include_in_schema=False)
+def swagger_ui_html():
+    return get_swagger_ui_html(
+        openapi_url=app.openapi_url,
+        title=app.title,
+        swagger_js_url="https://unpkg.com/swagger-ui-dist@5/swagger-ui-bundle.js",
+        swagger_css_url="https://unpkg.com/swagger-ui-dist@5/swagger-ui.css",
+    )
+
 origins = [
     "http://localhost:5173",
     "http://www.ihr.live",