catalog/packages/biz/pnf_descriptor.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
  16 import json
  17 import logging
  18 import os
  19 import uuid
  20
  21 from catalog.pub.config.config import CATALOG_ROOT_PATH
  22 from catalog.pub.utils import fileutil
  23 from catalog.pub.utils.values import ignore_case_get
  24 from catalog.pub.database.models import NSPackageModel, PnfPackageModel
  25 from catalog.pub.exceptions import CatalogException
  26 from catalog.pub.utils import toscaparser
  27
  28 logger = logging.getLogger(__name__)
  29
  30
  31 def create(data):
  32     user_defined_data = ignore_case_get(data, 'userDefinedData')
  33     data = {
  34         'id': str(uuid.uuid4()),
  35         'pnfdOnboardingState': 'CREATED',
  36         'pnfdUsageState': 'NOT_IN_USE',
  37         'userDefinedData': user_defined_data,
  38         '_links': None  # TODO
  39     }
  40     PnfPackageModel(
  41         pnfPackageId=data['id'],
  42         onboardingState=data['pnfdOnboardingState'],
  43         usageState=data['pnfdUsageState'],
  44         userDefinedData=data['userDefinedData']
  45     ).save()
  46     return data
  47
  48
  49 def query_multiple():
  50     pnf_pkgs = PnfPackageModel.objects.all()
  51     if not pnf_pkgs.exists():
  52         raise CatalogException('PNF descriptors do not exist.')
  53     response_data = []
  54     for pnf_pkg in pnf_pkgs:
  55         data = fill_response_data(pnf_pkg)
  56         response_data.append(data)
  57     return response_data
  58
  59
  60 def query_single(pnfd_info_id):
  61     pnf_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
  62     if not pnf_pkgs.exists():
  63         raise CatalogException('PNF descriptor (%s) does not exist.' % pnfd_info_id)
  64     return fill_response_data(pnf_pkgs[0])
  65
  66
  67 def process(pnfd_info_id, local_file_name):  # TODO: onboardingState changes
  68     pnfd_json = toscaparser.parse_pnfd(local_file_name)
  69     pnfd = json.JSONDecoder().decode(pnfd_json)
  70
  71     pnfd_id = pnfd["metadata"]["id"]
  72     if pnfd_id and PnfPackageModel.objects.filter(pnfdId=pnfd_id):  # pnfd_id may not exist
  73         raise CatalogException("NS Descriptor (%s) already exists." % pnfd_id)
  74
  75     PnfPackageModel(
  76         pnfPackageId=pnfd_info_id,
  77         pnfdId=pnfd_id,
  78         pnfdName=pnfd["metadata"].get("name", pnfd_id),
  79         pnfdDesginer=pnfd["metadata"].get("vendor", "undefined"),
  80         pnfdDescription=pnfd["metadata"].get("description", ""),
  81         pnfdVersion=pnfd["metadata"].get("version", "undefined"),
  82         nsPackageUri=local_file_name,  # TODO
  83         sdcCsarId=pnfd_info_id,
  84         localFilePath=local_file_name,
  85         pnfdModel=pnfd_json
  86     ).save()
  87
  88
  89 def upload(files, pnfd_info_id):
  90     ns_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
  91     if not ns_pkgs.exists():
  92         raise CatalogException('The NS descriptor (%s) does not exist.' % pnfd_info_id)
  93
  94     remote_files = files
  95     for remote_file in remote_files:
  96         local_file_name = remote_file.name
  97         local_file_dir = os.path.join(CATALOG_ROOT_PATH, pnfd_info_id)
  98         local_file_name = os.path.join(local_file_dir, local_file_name)
  99         if not os.path.exists(local_file_dir):
 100             fileutil.make_dirs(local_file_dir)
 101         with open(local_file_name, 'wb') as local_file:
 102             if remote_file.multiple_chunks(chunk_size=None):  # TODO: chunk_size
 103                 for chunk in remote_file.chunks():
 104                     local_file.write(chunk)
 105             else:
 106                 data = remote_file.read()
 107                 local_file.write(data)
 108
 109
 110 def download(pnfd_info_id):
 111     pnf_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
 112     if not pnf_pkgs.exists():
 113         raise CatalogException('The PNF Descriptor (%s) does not exist.' % pnfd_info_id)
 114     if pnf_pkgs[0].onboardingState != 'ONBOARDED':
 115         raise CatalogException('The PNF Descriptor (%s) is not ONBOARDED.' % pnfd_info_id)
 116     local_file_path = pnf_pkgs[0].localFilePath
 117     return local_file_path
 118
 119
 120 def delete_pnf(pnfd_info_id):
 121     # TODO
 122     pnf_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
 123     if not pnf_pkgs.exists():
 124         logger.debug('PNF descriptor (%s) is deleted.' % pnfd_info_id)
 125         return
 126     if pnf_pkgs[0].usageState != 'NOT_IN_USE':
 127         raise CatalogException('The PNF descriptor (%s) shall be NOT_IN_USE.' % pnfd_info_id)
 128     ns_pkgs = NSPackageModel.objects.all()
 129     for ns_pkg in ns_pkgs:
 130         if pnfd_info_id in ns_pkg.pnfdInfoIds:
 131             raise CatalogException('The PNF descriptor (%s) is referenced.' % pnfd_info_id)
 132             break
 133     pnf_pkgs.delete()
 134     vnf_pkg_path = os.path.join(CATALOG_ROOT_PATH, pnfd_info_id)
 135     fileutil.delete_dirs(vnf_pkg_path)
 136     logger.debug('PNF descriptor (%s) is deleted.' % pnfd_info_id)
 137
 138
 139 def fill_response_data(pnf_pkg):
 140     data = {
 141         'id': pnf_pkg.pnfPackageId,
 142         'pnfdId': pnf_pkg.pnfdId,
 143         'pnfdName': pnf_pkg.pnfdProductName,  # TODO: check
 144         'pnfdVersion': pnf_pkg.pnfdVersion,
 145         'pnfdProvider': pnf_pkg.pnfVendor,  # TODO: check
 146         'pnfdInvariantId': None,  # TODO
 147         'pnfdOnboardingState': pnf_pkg.onboardingState,
 148         'onboardingFailureDetails': None,  # TODO
 149         'pnfdUsageState': pnf_pkg.usageState,
 150         'userDefinedData': {},
 151         '_links': None  # TODO
 152     }
 153     if pnf_pkg.userDefinedData:
 154         user_defined_data = json.JSONDecoder().decode(pnf_pkg.userDefinedData)
 155         data['userDefinedData'] = user_defined_data
 156
 157     return data