Refactor: Make resources do "init_app" that will do the router inclusion (so that middleware, sub-apps, etc. can be added)

Author: nb-programmer

sample_fastapi/app/app_factory.py

@@ -27,5 +27,5 @@ async def app_lifespan_handler(app: FastAPI):
 
 def init_app():
     app = FastAPI(lifespan=app_lifespan_handler)
-    resources.init_routes(app)
+    resources.init_app(app)
     return app

sample_fastapi/app/resources/__init__.py

@@ -1,4 +1,4 @@
 from .lifespan import lifespan
-from .routes import init_routes
+from .routes import init_app
 
-__all__ = ["lifespan", "init_routes"]
+__all__ = ["lifespan", "init_app"]

sample_fastapi/app/resources/calculator/__init__.py

@@ -1,3 +1,3 @@
-from .routes import init_routes
+from .routes import init_app, init_routes
 
-__all__ = ["init_routes"]
+__all__ = ["init_app", "init_routes"]

sample_fastapi/app/resources/calculator/routes.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, FastAPI
 
 from .views import conv_dec2hex, conv_hex2dec
 
@@ -10,3 +10,7 @@ def init_routes():
     router.add_api_route("/convert/hex2dec", conv_hex2dec, methods={"POST"})
 
     return router
+
+
+def init_app(app: FastAPI):
+    app.include_router(init_routes())

sample_fastapi/app/resources/hello/__init__.py

@@ -1,3 +1,3 @@
-from .routes import init_routes
+from .routes import init_app, init_routes
 
-__all__ = ["init_routes"]
+__all__ = ["init_app", "init_routes"]

sample_fastapi/app/resources/hello/routes.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, FastAPI
 
 from .views import respond_hello, respond_name_greet
 
@@ -10,3 +10,7 @@ def init_routes():
     router.add_api_route("/greet", respond_name_greet, methods={"GET"})
 
     return router
+
+
+def init_app(app: FastAPI):
+    app.include_router(init_routes())

sample_fastapi/app/resources/routes.py

@@ -1,7 +1,8 @@
 from fastapi import FastAPI
 
 
-def init_routes(app: FastAPI):
+def init_app(app: FastAPI):
+    """Initialize routes, middleware, sub-apps, etc. to the given Application"""
     from . import calculator, hello
-    app.include_router(hello.init_routes())
-    app.include_router(calculator.init_routes())
+    hello.init_app(app)
+    calculator.init_app(app)