Enhance: Enable O2 DMS by exposing k8s API endpoint

[pti/o2.git] / o2dms / api / dms_route.py

diff --git a/o2dms/api/dms_route.py b/o2dms/api/dms_route.py
index 480d965..c92bdc1 100644 (file)
--- a/o2dms/api/dms_route.py
+++ b/o2dms/api/dms_route.py
@@ -13,7 +13,9 @@
 #  limitations under the License.
 
 # from flask import jsonify
-from flask_restx import Resource
+from os.path import exists
+from flask import send_file
+from flask_restx import Resource, reqparse
 
 from o2dms.api.dms_dto import DmsDTO
 from o2dms.api import dms_lcm_view
@@ -31,6 +33,8 @@ def configure_api_route():
 # ----------  DeploymentManagers ---------- #
 @api_dms_lcm_v1.route("/<deploymentManagerID>")
 @api_dms_lcm_v1.param('deploymentManagerID', 'ID of the deployment manager')
+@api_dms_lcm_v1.param('profile', 'DMS profile',
+                      location='args')
 @api_dms_lcm_v1.response(404, 'Deployment manager not found')
 class DmsGetRouter(Resource):
 
@@ -43,9 +47,32 @@ class DmsGetRouter(Resource):
             deploymentManagerID
         ))
         bus = MessageBus.get_instance()
+
+        parser = reqparse.RequestParser()
+        parser.add_argument('profile', location='args')
+        args = parser.parse_args()
+
         result = dms_lcm_view.deployment_manager_one(
-            deploymentManagerID, bus.uow)
+            deploymentManagerID, bus.uow, args.profile)
         if result is not None:
             return result
         api_dms_lcm_v1.abort(404, "Deployment manager {} doesn't exist".format(
             deploymentManagerID))
+
+
+@api_dms_lcm_v1.route("/<deploymentManagerID>/download/<filename>")
+@api_dms_lcm_v1.param('deploymentManagerID',
+                      'ID of the deployment manager')
+@api_dms_lcm_v1.param('filename',
+                      'profile filename')
+@api_dms_lcm_v1.response(404, 'profile not found')
+class DeploymentManagerGetFileRouter(Resource):
+    def get(self, deploymentManagerID, filename):
+        path = "/tmp/kubeconfig_" + filename
+
+        if exists(path):
+            return send_file(path, as_attachment=True)
+        api_dms_lcm_v1.abort(
+            404,
+            "Deployment manager {}'s Kube config file doesn't exist".
+            format(deploymentManagerID))