xapp_onboarder/xapp_onboarder/api/onboard.py

   1 ################################################################################
   2 #   Copyright (c) 2020 AT&T Intellectual Property.                             #
   3 #                                                                              #
   4 #   Licensed under the Apache License, Version 2.0 (the "License");            #
   5 #   you may not use this file except in compliance with the License.           #
   6 #   You may obtain a copy of the License at                                    #
   7 #                                                                              #
   8 #       http://www.apache.org/licenses/LICENSE-2.0                             #
   9 #                                                                              #
  10 #   Unless required by applicable law or agreed to in writing, software        #
  11 #   distributed under the License is distributed on an "AS IS" BASIS,          #
  12 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   #
  13 #   See the License for the specific language governing permissions and        #
  14 #   limitations under the License.                                             #
  15 ################################################################################
  16
  17 import logging
  18 import json
  19 from jsonschema import ValidationError, SchemaError
  20 from jsonschema import validate, Draft7Validator
  21 from xapp_onboarder.helm_controller.xApp_builder import xApp, xAppError
  22 from xapp_onboarder.server import settings
  23 from xapp_onboarder.repo_manager.repo_manager import requests_retry_session, repo_manager
  24 from xapp_onboarder.api.models.response_models import error_message_model, response, status_message_model
  25
  26
  27 log = logging.getLogger(__name__)
  28
  29
  30 def onboard(config_file, schema_file):
  31     if not repo_manager.is_repo_ready():
  32         response_message = response(model=error_message_model, status_code=500,
  33                                     error_source="xapp_onboarder",
  34                                     error_message="Cannot connect to local helm repo.",
  35                                     status="Service not ready.")
  36         return response_message.get_return()
  37
  38     try:
  39         Draft7Validator.check_schema(schema_file)
  40         validate(config_file, schema_file)
  41     except ValidationError as err:
  42         log.debug(err.message)
  43         response_message = response(model=error_message_model, status_code=400,
  44                                     error_source="config-file.json",
  45                                     error_message=err.message,
  46                                     status="Input payload validation failed")
  47         return response_message.get_return()
  48     except SchemaError as err:
  49         log.debug(err.message)
  50         response_message = response(model=error_message_model, status_code=400,
  51                                     error_source="schema.json",
  52                                     error_message=err.message,
  53                                     status="Input payload validation failed")
  54         return response_message.get_return()
  55
  56     try:
  57         xapp = xApp(config_file, schema_file)
  58         xapp.package_chart()
  59         xapp.distribute_chart()
  60     except xAppError as err:
  61         log.error(str(err))
  62         response_message = response(model=error_message_model, status_code=err.status_code,
  63                                     error_source="xApp_builder",
  64                                     error_message=str(err),
  65                                     status="xApp onboarding failed")
  66         return response_message.get_return()
  67     return response(model=status_message_model, status_code=201, status="Created").get_return()
  68
  69
  70 def download_config_and_schema_and_onboard(config_file_url, schema_url):
  71     if not repo_manager.is_repo_ready():
  72         response_message = response(model=error_message_model, status_code=500,
  73                                     error_source="xapp_onboarder",
  74                                     error_message="Cannot connect to local helm repo.",
  75                                     status="Service not ready.")
  76         return response_message.get_return()
  77
  78     session = requests_retry_session()
  79     try:
  80         response_content = session.get(config_file_url, timeout=settings.HTTP_TIME_OUT)
  81     except Exception as err:
  82         log.error(err.message)
  83         response_message = response(model=error_message_model, status_code=500,
  84                                     error_source="config-file.json",
  85                                     error_message=err.message,
  86                                     status="Downloading config-file.json failed")
  87         return response_message.get_return()
  88     else:
  89         if response_content.status_code != 200:
  90             error_message = "Wrong response code: {}, {}".format(response_content.status_code, response_content.content.decode("utf-8"))
  91             log.error(error_message)
  92             response_message = response(model=error_message_model, status_code=500,
  93                                         error_source="config-file.json",
  94                                         error_message=error_message,
  95                                         status="Downloading config-file.json failed")
  96             return response_message.get_return()
  97         config_file = json.loads(response_content.content)
  98
  99     try:
 100         response_content = session.get(schema_url, timeout=settings.HTTP_TIME_OUT)
 101     except Exception as err:
 102         log.error(err.message)
 103         response_message = response(model=error_message_model, status_code=500,
 104                                     error_source="schema.json",
 105                                     error_message=err.message,
 106                                     status="Downloading schema.json failed")
 107         return response_message.get_return()
 108     else:
 109         if response_content.status_code != 200:
 110             error_message = "Wrong response code. {}, {}".format(response_content.status_code, response_content.content.decode("utf-8"))
 111             log.error(error_message)
 112             response_message = response(model=error_message_model, status_code=500,
 113                                         error_source="schema.json",
 114                                         error_message=error_message,
 115                                         status="Downloading schema.json failed")
 116             return response_message.get_return()
 117         schema_file = json.loads(response_content.content)
 118
 119     return onboard(config_file, schema_file)