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     """
  42     Get NS pckage from SDC
  43     :param csar_id:
  44     :return:
  45     """
  46     ret = None
  47     try:
  48         ret = NsPackage().on_distribute(csar_id)
  49     except CatalogException as e:
  50         NsPackage().delete_csar(csar_id)
  51         return fmt_ns_pkg_rsp(STATUS_FAILED, e.args[0])
  52     except:
  53         logger.error(traceback.format_exc())
  54         NsPackage().delete_csar(csar_id)
  55         return fmt_ns_pkg_rsp(STATUS_FAILED, str(sys.exc_info()))
  56     if ret[0]:
  57         return fmt_ns_pkg_rsp(STATUS_FAILED, ret[1])
  58     return fmt_ns_pkg_rsp(STATUS_SUCCESS, ret[1], "")
  59
  60
  61 def ns_delete_csar(csar_id):
  62     """
  63     Delete NS package
  64     :param csar_id:
  65     :return:
  66     """
  67     ret = None
  68     try:
  69         ret = NsPackage().delete_csar(csar_id)
  70     except CatalogException as e:
  71         return fmt_ns_pkg_rsp(STATUS_FAILED, e.args[0])
  72     except:
  73         logger.error(traceback.format_exc())
  74         return fmt_ns_pkg_rsp(STATUS_FAILED, str(sys.exc_info()))
  75     return fmt_ns_pkg_rsp(STATUS_SUCCESS, ret[1], "")
  76
  77
  78 def ns_get_csars():
  79     """
  80     Get NS packages
  81     :return:
  82     """
  83     ret = None
  84     try:
  85         ret = NsPackage().get_csars()
  86     except CatalogException as e:
  87         return [1, e.args[0]]
  88     except:
  89         logger.error(traceback.format_exc())
  90         return [1, str(sys.exc_info())]
  91     return ret
  92
  93
  94 def ns_get_csar(csar_id):
  95     """
  96     Get NS package by id
  97     :param csar_id:
  98     :return:
  99     """
 100     ret = None
 101     try:
 102         ret = NsPackage().get_csar(csar_id)
 103     except CatalogException as e:
 104         return [1, e.args[0]]
 105     except Exception as e:
 106         logger.error(e.args[0])
 107         logger.error(traceback.format_exc())
 108         return [1, str(sys.exc_info())]
 109     return ret
 110
 111
 112 def parse_nsd(csar_id, inputs):
 113     """
 114     Parse NSD
 115     :param csar_id:
 116     :param inputs:
 117     :return:
 118     """
 119     ret = None
 120     try:
 121         ns_pkg = NSPackageModel.objects.filter(nsPackageId=csar_id)
 122         if not ns_pkg:
 123             raise CatalogException("NS CSAR(%s) does not exist." % csar_id)
 124         csar_path = ns_pkg[0].localFilePath
 125         ret = {"model": toscaparser.parse_nsd(csar_path, inputs)}
 126     except CatalogException as e:
 127         return [1, e.args[0]]
 128     except Exception as e:
 129         logger.error(e.args[0])
 130         logger.error(traceback.format_exc())
 131         return [1, str(sys.exc_info())]
 132     return [0, ret]
 133
 134
 135 class NsPackage(object):
 136     """
 137     Actions for sdc ns package.
 138     """
 139
 140     def __init__(self):
 141         pass
 142
 143     def on_distribute(self, csar_id):
 144         """
 145         Fetch NS package csar from SDC
 146         :param csar_id:
 147         :return:
 148         """
 149         if NSPackageModel.objects.filter(nsPackageId=csar_id):
 150             return [1, "NS CSAR(%s) already exists." % csar_id]
 151
 152         ns = sdc.get_asset(sdc.ASSETTYPE_SERVICES, csar_id)
 153         # check if the related resources exist
 154         resources = ns.get('resources', None)
 155         if resources:
 156             for resource in resources:
 157                 if resource['resoucreType'].upper == 'VF' and not VnfPackageModel.objects.filter(
 158                         vnfPackageId=resource['resourceUUID']):
 159                     logger.error("VF [%s] is not distributed.", resource['resourceUUID'])
 160                     raise CatalogException("VF (%s) is not distributed." % resource['resourceUUID'])
 161                 # if resource['resoucreType'] == 'PNF' and not PnfPackageModel.objects.filter(
 162                 #         pnfPackageId=resource['resourceUUID']):
 163                 #     logger.error("PNF [%s] is not distributed.", resource['resourceUUID'])
 164                 #     raise CatalogException("PNF (%s) is not distributed." % resource['resourceUUID'])
 165
 166         # download csar package
 167         local_path = os.path.join(CATALOG_ROOT_PATH, csar_id)
 168         csar_name = "%s.csar" % ns.get("name", csar_id)
 169         local_file_name = sdc.download_artifacts(ns["toscaModelURL"], local_path, csar_name)
 170         if local_file_name.endswith(".csar") or local_file_name.endswith(".zip"):
 171             artifact_vnf_file = fileutil.unzip_file(local_file_name, local_path, "Artifacts/Deployment/OTHER/ns.csar")
 172             if os.path.exists(artifact_vnf_file):
 173                 local_file_name = artifact_vnf_file
 174
 175         data = {
 176             'userDefinedData': {}
 177         }
 178         nsd = NsDescriptor()
 179         nsd.create(data, csar_id)
 180         nsd.parse_nsd_and_save(csar_id, local_file_name)
 181         return [0, "CSAR(%s) distributed successfully." % csar_id]
 182
 183     def delete_csar(self, csar_id):
 184         """
 185         Delete NS package by id
 186         :param csar_id:
 187         :return:
 188         """
 189         nsd = NsDescriptor()
 190         nsd.delete_single(csar_id)
 191         return [0, "Delete CSAR(%s) successfully." % csar_id]
 192
 193     def get_csars(self):
 194         """
 195         Get ns packages
 196         :return:
 197         """
 198         csars = []
 199         nss = NSPackageModel.objects.filter()
 200         for ns in nss:
 201             ret = self.get_csar(ns.nsPackageId)
 202             csars.append(ret[1])
 203         return [0, csars]
 204
 205     def get_csar(self, csar_id):
 206         """
 207         Get NS package by id
 208         :param csar_id:
 209         :return:
 210         """
 211         package_info = {}
 212         csars = NSPackageModel.objects.filter(nsPackageId=csar_id)
 213         if csars:
 214             package_info["nsdId"] = csars[0].nsdId
 215             package_info["nsPackageId"] = csars[0].nsPackageId
 216             package_info["nsdProvider"] = csars[0].nsdDesginer
 217             package_info["nsdVersion"] = csars[0].nsdVersion
 218             package_info["csarName"] = csars[0].nsPackageUri
 219             package_info["nsdModel"] = csars[0].nsdModel
 220             package_info["nsdInvariantId"] = csars[0].invariantId
 221             package_info["downloadUrl"] = "http://%s:%s/%s/%s/%s" % (
 222                 REG_TO_MSB_REG_PARAM[0]["nodes"][0]["ip"],
 223                 REG_TO_MSB_REG_PARAM[0]["nodes"][0]["port"],
 224                 CATALOG_URL_PATH,
 225                 csar_id,
 226                 csars[0].nsPackageUri)
 227         else:
 228             raise CatalogException("Ns package[%s] not Found." % csar_id)
 229
 230         return [0, {"csarId": csar_id, "packageInfo": package_info}]