Configurable FastAPI endpoints

api/api-pod.yaml

@@ -10,6 +10,8 @@ data:
   RABBITMQ_BROKER: pyamqp://skipper:[email protected]//
   RABBITMQ_HOST: rabbitmq-0.rabbitmq.rabbits.svc.cluster.local
   RABBITMQ_PORT: '5672'
+  BOSTON_ENABLED: 'y'
+  MOBILENET_ENABLED: 'y'
 
 ---
 
@@ -78,6 +80,16 @@ spec:
                 configMapKeyRef:
                   name: api-config
                   key: RABBITMQ_PORT
+            - name: BOSTON_ENABLED
+              valueFrom:
+                configMapKeyRef:
+                  name: api-config
+                  key: BOSTON_ENABLED
+            - name: MOBILENET_ENABLED
+              valueFrom:
+                configMapKeyRef:
+                  name: api-config
+                  key: MOBILENET_ENABLED
             - name: RABBITMQ_USER
               valueFrom:
                 secretKeyRef:

api/endpoint.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from api.routers import skipper, boston, mobilenet
+import os
 
 app = FastAPI(openapi_url="/api/v1/skipper/tasks/openapi.json",
               docs_url="/api/v1/skipper/tasks/docs")
@@ -13,6 +14,11 @@
     allow_credentials=True,
 )
 
+boston_enabled = os.getenv('BOSTON_ENABLED', 'y')
+mobilenet_enabled = os.getenv('MOBILENET_ENABLED', 'y')
+
 app.include_router(skipper.router, prefix='/api/v1/skipper/tasks')
-app.include_router(boston.router, prefix='/api/v1/skipper/tasks')
-app.include_router(mobilenet.router, prefix='/api/v1/skipper/tasks')
+if boston_enabled == 'y':
+    app.include_router(boston.router, prefix='/api/v1/skipper/tasks')
+if mobilenet_enabled == 'y':
+    app.include_router(mobilenet.router, prefix='/api/v1/skipper/tasks')

docker-compose.yml

@@ -65,6 +65,8 @@ services:
       - RABBITMQ_PORT=5672
       - RABBITMQ_USER=skipper
       - RABBITMQ_PASSWORD=welcome1
+      - BOSTON_ENABLED=y
+      - MOBILENET_ENABLED=y
     networks:
       - network1
     depends_on: