Register V1 router

ChauffeurPrive · Jul 27, 2020 · d39cebf · d39cebf
1 parent 00753a4
commit d39cebf
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 16 deletions.
diff --git a/nestor_api/api/api.py b/nestor_api/api/api.py
diff --git a/nestor_api/api/api_routes/api.py b/nestor_api/api/api_routes/api.py
@@ -0,0 +1,12 @@
+"""Register all routes under the /api prefixes. The route /heartbeat is not included."""
+from flask import Blueprint
+
+from nestor_api.api.api_routes.v1 import create_api_v1
+
+
+def get_apis() -> list:
+    """Return the list of apis to register."""
+
+    return [
+        create_api_v1(),
+    ]
diff --git a/nestor_api/api/flask_app.py b/nestor_api/api/flask_app.py
@@ -1,7 +1,7 @@
 """Return an initialized Flask application with the API."""
 from flask import Flask
 
-from nestor_api.api.api import create_api
+from nestor_api.api.api_routes.api import get_apis
 from nestor_api.api.public_routes import heartbeat
 
 
@@ -12,7 +12,7 @@ def create_app() -> Flask:
     # Note: heartbeat is exposed without authentication
     app.register_blueprint(heartbeat.blueprint)
 
-    api = create_api()
-    app.register_blueprint(api)
+    for api in get_apis():
+        app.register_blueprint(api)
 
     return app