RIC-769: Committing individual files rather than tar archive

[ric-plt/appmgr.git] / xapp_orchestrater / dev / xapp_onboarder / xapp_onboarder / api / charts.py

diff --git a/xapp_orchestrater/dev/xapp_onboarder/xapp_onboarder/api/charts.py b/xapp_orchestrater/dev/xapp_onboarder/xapp_onboarder/api/charts.py
new file mode 100644 (file)
index 0000000..223a5be
--- /dev/null
+++ b/xapp_orchestrater/dev/xapp_onboarder/xapp_onboarder/api/charts.py
@@ -0,0 +1,81 @@
+################################################################################
+#   Copyright (c) 2020 AT&T Intellectual Property.                             #
+#                                                                              #
+#   Licensed under the Apache License, Version 2.0 (the "License");            #
+#   you may not use this file except in compliance with the License.           #
+#   You may obtain a copy of the License at                                    #
+#                                                                              #
+#       http://www.apache.org/licenses/LICENSE-2.0                             #
+#                                                                              #
+#   Unless required by applicable law or agreed to in writing, software        #
+#   distributed under the License is distributed on an "AS IS" BASIS,          #
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   #
+#   See the License for the specific language governing permissions and        #
+#   limitations under the License.                                             #
+################################################################################
+
+import logging
+import io
+import tarfile
+from xapp_onboarder.repo_manager.repo_manager import repo_manager, RepoManagerError
+from xapp_onboarder.api.models.response_models import error_message_model, response
+
+log = logging.getLogger(__name__)
+
+
+def get_charts_list(xapp_chart_name=None):
+
+    if not repo_manager.is_repo_ready():
+        response_message = response(model=error_message_model, status_code=500,
+                                    error_source="xapp_onboarder",
+                                    error_message="Cannot connect to local helm repo.",
+                                    status="Service not ready.")
+        return response_message.get_return()
+
+    try:
+        content = repo_manager.get_xapp_list(xapp_chart_name=xapp_chart_name)
+    except RepoManagerError as err:
+        log.error(str(err))
+        response_message = response(model=error_message_model, status_code=err.status_code,
+                                    error_source="charts_fetcher",
+                                    error_message=str(err),
+                                    status="Fetching helm chart list failed")
+        return response_message.get_return()
+    return content, 200
+
+
+def download_chart_package(xapp_chart_name, version):
+
+    if not repo_manager.is_repo_ready():
+        response_message = response(model=error_message_model, status_code=500,
+                                    error_source="xapp_onboarder",
+                                    error_message="Cannot connect to local helm repo.",
+                                    status="Service not ready.")
+        return response_message.get_return()
+    try:
+        content = repo_manager.download_xapp_chart(xapp_chart_name=xapp_chart_name, version=version)
+    except RepoManagerError as err:
+        log.error(str(err))
+        response_message = response(model=error_message_model, status_code=err.status_code,
+                                    error_source="charts_fetcher",
+                                    error_message=str(err),
+                                    status="Downloading helm chart package failed")
+        return response_message.get_return()
+    return content, 200
+
+
+def download_values_yaml(xapp_chart_name, version):
+
+    content, status = download_chart_package(xapp_chart_name=xapp_chart_name, version=version)
+
+    if status != 200:
+        return content, status
+
+    file_stream = io.BytesIO(content)
+
+    with tarfile.open(fileobj=file_stream) as tar:
+        values_yaml_file = tar.extractfile(xapp_chart_name + '/values.yaml')
+        return_response = values_yaml_file.read()
+
+    return return_response, 200
+