1 # Copyright 2018 ZTE Corporation.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
21 from catalog.packages.biz.common import read, save
22 from catalog.packages.const import PKG_STATUS
23 from catalog.pub.config.config import CATALOG_ROOT_PATH
24 from catalog.pub.database.models import NSPackageModel, PnfPackageModel
25 from catalog.pub.exceptions import CatalogException, ResourceNotFoundException
26 from catalog.pub.utils import fileutil, toscaparser
27 from catalog.pub.utils.values import ignore_case_get
29 logger = logging.getLogger(__name__)
32 class PnfDescriptor(object):
37 def create(self, data):
38 logger.info('Start to create a PNFD...')
39 user_defined_data = ignore_case_get(data, 'userDefinedData', {})
41 'id': str(uuid.uuid4()),
42 'pnfdOnboardingState': PKG_STATUS.CREATED,
43 'pnfdUsageState': PKG_STATUS.NOT_IN_USE,
44 'userDefinedData': user_defined_data,
47 PnfPackageModel.objects.create(
48 pnfPackageId=data['id'],
49 onboardingState=data['pnfdOnboardingState'],
50 usageState=data['pnfdUsageState'],
51 userDefinedData=json.dumps(user_defined_data)
53 logger.info('A PNFD(%s) has been created.' % data['id'])
56 def query_multiple(self, request):
57 pnfdId = request.query_params.get('pnfdId')
59 pnf_pkgs = PnfPackageModel.objects.filter(pnfdId=pnfdId)
61 pnf_pkgs = PnfPackageModel.objects.all()
63 for pnf_pkg in pnf_pkgs:
64 data = self.fill_response_data(pnf_pkg)
65 response_data.append(data)
68 def query_single(self, pnfd_info_id):
69 pnf_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
70 if not pnf_pkgs.exists():
71 logger.error('PNFD(%s) does not exist.' % pnfd_info_id)
72 raise ResourceNotFoundException('PNFD(%s) does not exist.' % pnfd_info_id)
73 return self.fill_response_data(pnf_pkgs[0])
75 def upload(self, remote_file, pnfd_info_id):
76 logger.info('Start to upload PNFD(%s)...' % pnfd_info_id)
77 pnf_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
78 if not pnf_pkgs.exists():
79 logger.info('PNFD(%s) does not exist.' % pnfd_info_id)
80 raise CatalogException('PNFD (%s) does not exist.' % pnfd_info_id)
81 pnf_pkgs.update(onboardingState=PKG_STATUS.UPLOADING)
83 local_file_name = save(remote_file, pnfd_info_id)
84 logger.info('PNFD(%s) content has been uploaded.' % pnfd_info_id)
85 return local_file_name
87 def delete_single(self, pnfd_info_id):
88 logger.info('Start to delete PNFD(%s)...' % pnfd_info_id)
89 pnf_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
90 if not pnf_pkgs.exists():
91 logger.info('PNFD(%s) has been deleted.' % pnfd_info_id)
94 if pnf_pkgs[0].usageState != PKG_STATUS.NOT_IN_USE:
95 logger.info('PNFD(%s) shall be NOT_IN_USE.' % pnfd_info_id)
96 raise CatalogException('PNFD(%s) shall be NOT_IN_USE.' % pnfd_info_id)
98 del_pnfd_id = pnf_pkgs[0].pnfdId
99 ns_pkgs = NSPackageModel.objects.all()
100 for ns_pkg in ns_pkgs:
103 nsd_model = json.JSONDecoder().decode(ns_pkg.nsdModel)
106 for pnf in nsd_model['pnfs']:
107 if del_pnfd_id == pnf["properties"]["id"]:
108 logger.warn("PNFD(%s) is referenced in NSD", del_pnfd_id)
109 raise CatalogException('PNFD(%s) is referenced.' % pnfd_info_id)
111 pnf_pkg_path = os.path.join(CATALOG_ROOT_PATH, pnfd_info_id)
112 fileutil.delete_dirs(pnf_pkg_path)
113 logger.debug('PNFD(%s) has been deleted.' % pnfd_info_id)
115 def download(self, pnfd_info_id):
116 logger.info('Start to download PNFD(%s)...' % pnfd_info_id)
117 pnf_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
118 if not pnf_pkgs.exists():
119 logger.error('PNFD(%s) does not exist.' % pnfd_info_id)
120 raise ResourceNotFoundException('PNFD(%s) does not exist.' % pnfd_info_id)
121 if pnf_pkgs[0].onboardingState != PKG_STATUS.ONBOARDED:
122 logger.error('PNFD(%s) is not ONBOARDED.' % pnfd_info_id)
123 raise CatalogException('PNFD(%s) is not ONBOARDED.' % pnfd_info_id)
125 local_file_path = pnf_pkgs[0].localFilePath
126 start, end = 0, os.path.getsize(local_file_path)
127 logger.info('PNFD(%s) has been downloaded.' % pnfd_info_id)
128 return read(local_file_path, start, end)
130 def parse_pnfd_and_save(self, pnfd_info_id, local_file_name):
131 logger.info('Start to process PNFD(%s)...' % pnfd_info_id)
132 pnf_pkgs = PnfPackageModel.objects.filter(pnfPackageId=pnfd_info_id)
133 pnf_pkgs.update(onboardingState=PKG_STATUS.PROCESSING)
134 pnfd_json = toscaparser.parse_pnfd(local_file_name)
135 pnfd = json.JSONDecoder().decode(pnfd_json)
137 logger.debug("pnfd_json is %s" % pnfd_json)
142 if pnfd.get("pnf", "") != "":
143 if pnfd["pnf"].get("properties", "") != "":
144 pnfd_id = pnfd["pnf"].get("properties", {}).get("descriptor_id", "")
145 pnfdVersion = pnfd["pnf"].get("properties", {}).get("version", "")
146 pnfdProvider = pnfd["pnf"].get("properties", {}).get("provider", "")
147 pnfdName = pnfd["pnf"].get("properties", {}).get("name", "")
149 pnfd_id = pnfd["metadata"].get("descriptor_id", "")
151 pnfd_id = pnfd["metadata"].get("id", "")
153 pnfd_id = pnfd["metadata"].get("UUID", "")
155 raise CatalogException('pnfd_id is Null.')
157 if pnfdVersion == "":
158 pnfdVersion = pnfd["metadata"].get("template_version", "")
159 if pnfdVersion == "":
160 pnfdVersion = pnfd["metadata"].get("version", "")
162 if pnfdProvider == "":
163 pnfdProvider = pnfd["metadata"].get("template_author", "")
164 if pnfdVersion == "":
165 pnfdVersion = pnfd["metadata"].get("provider", "")
168 pnfdName = pnfd["metadata"].get("template_name", "")
169 if pnfdVersion == "":
170 pnfdName = pnfd["metadata"].get("name", "")
172 other_pnf = PnfPackageModel.objects.filter(pnfdId=pnfd_id)
173 if other_pnf and other_pnf[0].pnfPackageId != pnfd_info_id:
174 logger.info('PNFD(%s) already exists.' % pnfd_id)
175 raise CatalogException("PNFD(%s) already exists." % pnfd_id)
180 pnfdVersion=pnfdVersion,
181 pnfVendor=pnfdProvider,
182 pnfPackageUri=local_file_name,
183 onboardingState=PKG_STATUS.ONBOARDED,
184 usageState=PKG_STATUS.NOT_IN_USE,
185 localFilePath=local_file_name,
188 logger.info('PNFD(%s) has been processed.' % pnfd_info_id)
190 def fill_response_data(self, pnf_pkg):
192 'id': pnf_pkg.pnfPackageId,
193 'pnfdId': pnf_pkg.pnfdId,
194 'pnfdName': pnf_pkg.pnfdName,
195 'pnfdVersion': pnf_pkg.pnfdVersion,
196 'pnfdProvider': pnf_pkg.pnfVendor,
197 'pnfdInvariantId': None, # TODO
198 'pnfdOnboardingState': pnf_pkg.onboardingState,
199 'onboardingFailureDetails': None, # TODO
200 'pnfdUsageState': pnf_pkg.usageState,
201 'userDefinedData': {},
202 '_links': None # TODO
204 if pnf_pkg.userDefinedData:
205 user_defined_data = json.JSONDecoder().decode(pnf_pkg.userDefinedData)
206 data['userDefinedData'] = user_defined_data
210 def handle_upload_failed(self, pnf_pkg_id):
211 pnf_pkg = PnfPackageModel.objects.filter(pnfPackageId=pnf_pkg_id)
212 pnf_pkg.update(onboardingState=PKG_STATUS.CREATED)
214 def parse_pnfd(self, csar_id, inputs):
216 pnf_pkg = PnfPackageModel.objects.filter(pnfPackageId=csar_id)
218 raise CatalogException("PNF CSAR(%s) does not exist." % csar_id)
219 csar_path = pnf_pkg[0].localFilePath
220 ret = {"model": toscaparser.parse_pnfd(csar_path, inputs)}
221 except CatalogException as e:
222 return [1, e.args[0]]
223 except Exception as e:
224 logger.error(e.args[0])
225 return [1, e.args[0]]