genericparser/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 genericparser.pub.config.config import GENERICPARSER_ROOT_PATH, GENERICPARSER_URL_PATH
  21 from genericparser.pub.config.config import REG_TO_MSB_REG_PARAM
  22 from genericparser.pub.database.models import NSPackageModel
  23 from genericparser.pub.exceptions import GenericparserException
  24 from genericparser.pub.msapi import sdc
  25 from genericparser.pub.utils import toscaparsers
  26 from genericparser.packages.biz.ns_descriptor import NsDescriptor
  27 from genericparser.pub.utils import fileutil
  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 GenericparserException as e:
  45         NsPackage().delete_csar(csar_id)
  46         return fmt_ns_pkg_rsp(STATUS_FAILED, e.message)
  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 GenericparserException as e:
  61         return fmt_ns_pkg_rsp(STATUS_FAILED, e.message)
  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 GenericparserException as e:
  73         return [1, e.message]
  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 GenericparserException as e:
  85         return [1, e.message]
  86     except Exception as e:
  87         logger.error(e.message)
  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 GenericparserException("NS CSAR(%s) does not exist." % csar_id)
  99         csar_path = ns_pkg[0].localFilePath
 100         ret = {"model": toscaparsers.parse_nsd(csar_path, inputs)}
 101     except GenericparserException as e:
 102         return [1, e.message]
 103     except Exception as e:
 104         logger.error(e.message)
 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         artifact = sdc.get_artifact(sdc.ASSETTYPE_SERVICES, csar_id)
 123         local_path = os.path.join(GENERICPARSER_ROOT_PATH, csar_id)
 124         csar_name = "%s.csar" % artifact.get("name", csar_id)
 125         local_file_name = sdc.download_artifacts(artifact["toscaModelURL"], local_path, csar_name)
 126         if local_file_name.endswith(".csar") or local_file_name.endswith(".zip"):
 127             artifact_vnf_file = fileutil.unzip_file(local_file_name, local_path, "Artifacts/Deployment/OTHER/ns.csar")
 128             if os.path.exists(artifact_vnf_file):
 129                 local_file_name = artifact_vnf_file
 130
 131         data = {
 132             'userDefinedData': {}
 133         }
 134         nsd = NsDescriptor()
 135         nsd.create(data, csar_id)
 136         nsd.parse_nsd_and_save(csar_id, local_file_name)
 137         return [0, "CSAR(%s) distributed successfully." % csar_id]
 138
 139     def delete_csar(self, csar_id):
 140         nsd = NsDescriptor()
 141         nsd.delete_single(csar_id)
 142         return [0, "Delete CSAR(%s) successfully." % csar_id]
 143
 144     def get_csars(self):
 145         csars = []
 146         nss = NSPackageModel.objects.filter()
 147         for ns in nss:
 148             ret = self.get_csar(ns.nsPackageId)
 149             csars.append(ret[1])
 150         return [0, csars]
 151
 152     def get_csar(self, csar_id):
 153         package_info = {}
 154         csars = NSPackageModel.objects.filter(nsPackageId=csar_id)
 155         if csars:
 156             package_info["nsdId"] = csars[0].nsdId
 157             package_info["nsPackageId"] = csars[0].nsPackageId
 158             package_info["nsdProvider"] = csars[0].nsdDesginer
 159             package_info["nsdVersion"] = csars[0].nsdVersion
 160             package_info["csarName"] = csars[0].nsPackageUri
 161             package_info["nsdModel"] = csars[0].nsdModel
 162             package_info["nsdInvariantId"] = csars[0].invariantId
 163             package_info["downloadUrl"] = "http://%s:%s/%s/%s/%s" % (
 164                 REG_TO_MSB_REG_PARAM[0]["nodes"][0]["ip"],
 165                 REG_TO_MSB_REG_PARAM[0]["nodes"][0]["port"],
 166                 GENERICPARSER_URL_PATH,
 167                 csar_id,
 168                 csars[0].nsPackageUri)
 169         else:
 170             raise GenericparserException("Ns package[%s] not Found." % csar_id)
 171
 172         return [0, {"csarId": csar_id, "packageInfo": package_info}]