Update main.py

This commit is contained in:
qiaofeng1227 2023-04-21 11:18:41 +08:00 committed by GitHub
parent a3466be24f
commit ee2653f7d7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -10,8 +10,8 @@ from fastapi.openapi.docs import (
get_swagger_ui_oauth2_redirect_html, get_swagger_ui_oauth2_redirect_html,
) )
myLogger.info_logger("Starting server") myLogger.info_logger("Start server...")
app = FastAPI(docs_url="/docs", redoc_url="/redoc",openapi_url="/openapi.json") app = FastAPI(docs_url=None, redoc_url=None)
def get_app(): def get_app():
origins = [ origins = [
@ -30,7 +30,7 @@ def get_app():
app.include_router(api_router_v1.get_api()) app.include_router(api_router_v1.get_api())
return app return app
@app.get("/docs", include_in_schema=True) @app.get("/docs", include_in_schema=False)
async def custom_swagger_ui_html(): async def custom_swagger_ui_html():
return get_swagger_ui_html( return get_swagger_ui_html(
openapi_url=app.openapi_url, openapi_url=app.openapi_url,
@ -40,11 +40,11 @@ async def custom_swagger_ui_html():
swagger_css_url="/static/swagger-ui/swagger-ui.css", swagger_css_url="/static/swagger-ui/swagger-ui.css",
) )
@app.get(app.swagger_ui_oauth2_redirect_url, include_in_schema=True) @app.get(app.swagger_ui_oauth2_redirect_url, include_in_schema=False)
async def swagger_ui_redirect(): async def swagger_ui_redirect():
return get_swagger_ui_oauth2_redirect_html() return get_swagger_ui_oauth2_redirect_html()
@app.get("/redoc", include_in_schema=True) @app.get("/redoc", include_in_schema=False)
async def redoc_html(): async def redoc_html():
return get_redoc_html( return get_redoc_html(
openapi_url=app.openapi_url, openapi_url=app.openapi_url,