catalog/packages/biz/sdc_ns_package.py

   1 # Copyright 2017 ZTE Corporation.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #         http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 import logging
  16 import os
  17 import sys
  18 import traceback
  19
  20 from catalog.packages.biz.ns_descriptor import NsDescriptor
  21 from catalog.pub.config.config import CATALOG_ROOT_PATH, CATALOG_URL_PATH
  22 from catalog.pub.config.config import REG_TO_MSB_REG_PARAM
  23 from catalog.pub.database.models import NSPackageModel, VnfPackageModel
  24 from catalog.pub.exceptions import CatalogException
  25 from catalog.pub.msapi import sdc
  26 from catalog.pub.utils import fileutil
  27 from catalog.pub.utils import toscaparser
  28
  29 logger = logging.getLogger(__name__)
  30
  31 STATUS_SUCCESS, STATUS_FAILED = "success", "failed"
  32
  33 METADATA = "metadata"
  34
  35
  36 def fmt_ns_pkg_rsp(status, desc, error_code="500"):
  37     return [0, {"status": status, "statusDescription": desc, "errorCode": error_code}]
  38
  39
  40 def ns_on_distribute(csar_id):
  41     ret = None
  42     try:
  43         ret = NsPackage().on_distribute(csar_id)
  44     except CatalogException as e:
  45         NsPackage().delete_csar(csar_id)
  46         return fmt_ns_pkg_rsp(STATUS_FAILED, e.args[0])
  47     except:
  48         logger.error(traceback.format_exc())
  49         NsPackage().delete_csar(csar_id)
  50         return fmt_ns_pkg_rsp(STATUS_FAILED, str(sys.exc_info()))
  51     if ret[0]:
  52         return fmt_ns_pkg_rsp(STATUS_FAILED, ret[1])
  53     return fmt_ns_pkg_rsp(STATUS_SUCCESS, ret[1], "")
  54
  55
  56 def ns_delete_csar(csar_id):
  57     ret = None
  58     try:
  59         ret = NsPackage().delete_csar(csar_id)
  60     except CatalogException as e:
  61         return fmt_ns_pkg_rsp(STATUS_FAILED, e.args[0])
  62     except:
  63         logger.error(traceback.format_exc())
  64         return fmt_ns_pkg_rsp(STATUS_FAILED, str(sys.exc_info()))
  65     return fmt_ns_pkg_rsp(STATUS_SUCCESS, ret[1], "")
  66
  67
  68 def ns_get_csars():
  69     ret = None
  70     try:
  71         ret = NsPackage().get_csars()
  72     except CatalogException as e:
  73         return [1, e.args[0]]
  74     except:
  75         logger.error(traceback.format_exc())
  76         return [1, str(sys.exc_info())]
  77     return ret
  78
  79
  80 def ns_get_csar(csar_id):
  81     ret = None
  82     try:
  83         ret = NsPackage().get_csar(csar_id)
  84     except CatalogException as e:
  85         return [1, e.args[0]]
  86     except Exception as e:
  87         logger.error(e.args[0])
  88         logger.error(traceback.format_exc())
  89         return [1, str(sys.exc_info())]
  90     return ret
  91
  92
  93 def parse_nsd(csar_id, inputs):
  94     ret = None
  95     try:
  96         ns_pkg = NSPackageModel.objects.filter(nsPackageId=csar_id)
  97         if not ns_pkg:
  98             raise CatalogException("NS CSAR(%s) does not exist." % csar_id)
  99         csar_path = ns_pkg[0].localFilePath
 100         ret = {"model": toscaparser.parse_nsd(csar_path, inputs)}
 101     except CatalogException as e:
 102         return [1, e.args[0]]
 103     except Exception as e:
 104         logger.error(e.args[0])
 105         logger.error(traceback.format_exc())
 106         return [1, str(sys.exc_info())]
 107     return [0, ret]
 108
 109
 110 class NsPackage(object):
 111     """
 112     Actions for sdc ns package.
 113     """
 114
 115     def __init__(self):
 116         pass
 117
 118     def on_distribute(self, csar_id):
 119         if NSPackageModel.objects.filter(nsPackageId=csar_id):
 120             return [1, "NS CSAR(%s) already exists." % csar_id]
 121
 122         ns = sdc.get_asset(sdc.ASSETTYPE_SERVICES, csar_id)
 123         # check if the related resources exist
 124         resources = ns.get('resources', None)
 125         if resources:
 126             for resource in resources:
 127                 if resource['resoucreType'].upper == 'VF' and not VnfPackageModel.objects.filter(
 128                         vnfPackageId=resource['resourceUUID']):
 129                     logger.error("VF [%s] is not distributed.", resource['resourceUUID'])
 130                     raise CatalogException("VF (%s) is not distributed." % resource['resourceUUID'])
 131                 # if resource['resoucreType'] == 'PNF' and not PnfPackageModel.objects.filter(
 132                 #         pnfPackageId=resource['resourceUUID']):
 133                 #     logger.error("PNF [%s] is not distributed.", resource['resourceUUID'])
 134                 #     raise CatalogException("PNF (%s) is not distributed." % resource['resourceUUID'])
 135
 136         # download csar package
 137         local_path = os.path.join(CATALOG_ROOT_PATH, csar_id)
 138         csar_name = "%s.csar" % ns.get("name", csar_id)
 139         local_file_name = sdc.download_artifacts(ns["toscaModelURL"], local_path, csar_name)
 140         if local_file_name.endswith(".csar") or local_file_name.endswith(".zip"):
 141             artifact_vnf_file = fileutil.unzip_file(local_file_name, local_path, "Artifacts/Deployment/OTHER/ns.csar")
 142             if os.path.exists(artifact_vnf_file):
 143                 local_file_name = artifact_vnf_file
 144
 145         data = {
 146             'userDefinedData': {}
 147         }
 148         nsd = NsDescriptor()
 149         nsd.create(data, csar_id)
 150         nsd.parse_nsd_and_save(csar_id, local_file_name)
 151         return [0, "CSAR(%s) distributed successfully." % csar_id]
 152
 153     def delete_csar(self, csar_id):
 154         nsd = NsDescriptor()
 155         nsd.delete_single(csar_id)
 156         return [0, "Delete CSAR(%s) successfully." % csar_id]
 157
 158     def get_csars(self):
 159         csars = []
 160         nss = NSPackageModel.objects.filter()
 161         for ns in nss:
 162             ret = self.get_csar(ns.nsPackageId)
 163             csars.append(ret[1])
 164         return [0, csars]
 165
 166     def get_csar(self, csar_id):
 167         package_info = {}
 168         csars = NSPackageModel.objects.filter(nsPackageId=csar_id)
 169         if csars:
 170             package_info["nsdId"] = csars[0].nsdId
 171             package_info["nsPackageId"] = csars[0].nsPackageId
 172             package_info["nsdProvider"] = csars[0].nsdDesginer
 173             package_info["nsdVersion"] = csars[0].nsdVersion
 174             package_info["csarName"] = csars[0].nsPackageUri
 175             package_info["nsdModel"] = csars[0].nsdModel
 176             package_info["nsdInvariantId"] = csars[0].invariantId
 177             package_info["downloadUrl"] = "http://%s:%s/%s/%s/%s" % (
 178                 REG_TO_MSB_REG_PARAM[0]["nodes"][0]["ip"],
 179                 REG_TO_MSB_REG_PARAM[0]["nodes"][0]["port"],
 180                 CATALOG_URL_PATH,
 181                 csar_id,
 182                 csars[0].nsPackageUri)
 183         else:
 184             raise CatalogException("Ns package[%s] not Found." % csar_id)
 185
 186         return [0, {"csarId": csar_id, "packageInfo": package_info}]