X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osmclient%2Fsol005%2Fvnfd.py;fp=osmclient%2Fsol005%2Fvnfd.py;h=4542d56c13d87a014fb6cfc84beba0d92a1da58c;hb=95686bbc69ded243c346f94dceb0bee567572fb7;hp=e32ac832375f128c37fb8f448fda723f261cab5e;hpb=52424a2ae26db69c5a97d01e84454ffdd4d31228;p=osm%2Fosmclient.git diff --git a/osmclient/sol005/vnfd.py b/osmclient/sol005/vnfd.py index e32ac83..4542d56 100644 --- a/osmclient/sol005/vnfd.py +++ b/osmclient/sol005/vnfd.py @@ -33,24 +33,24 @@ from osm_im.validation import Validation as validation_im class Vnfd(object): - def __init__(self, http=None, client=None): self._http = http self._client = client - self._logger = logging.getLogger('osmclient') - self._apiName = '/vnfpkgm' - self._apiVersion = '/v1' - self._apiResource = '/vnf_packages' - self._apiBase = '{}{}{}'.format(self._apiName, - self._apiVersion, self._apiResource) + self._logger = logging.getLogger("osmclient") + self._apiName = "/vnfpkgm" + self._apiVersion = "/v1" + self._apiResource = "/vnf_packages" + self._apiBase = "{}{}{}".format( + self._apiName, self._apiVersion, self._apiResource + ) def list(self, filter=None): self._logger.debug("") self._client.get_token() - filter_string = '' + filter_string = "" if filter: - filter_string = '?{}'.format(filter) - _, resp = self._http.get2_cmd('{}{}'.format(self._apiBase, filter_string)) + filter_string = "?{}".format(filter) + _, resp = self._http.get2_cmd("{}{}".format(self._apiBase, filter_string)) if resp: return json.loads(resp) @@ -61,13 +61,13 @@ class Vnfd(object): self._client.get_token() if utils.validate_uuid4(name): for vnfd in self.list(): - if name == vnfd['_id']: + if name == vnfd["_id"]: return vnfd else: for vnfd in self.list(): - if 'product-name' in vnfd and name == vnfd['product-name']: + if "product-name" in vnfd and name == vnfd["product-name"]: return vnfd - elif 'name' in vnfd and name == vnfd['name']: + elif "name" in vnfd and name == vnfd["name"]: return vnfd raise NotFound("vnfd {} not found".format(name)) @@ -77,7 +77,7 @@ class Vnfd(object): # It is redundant, since the previous one already gets the whole vnfpkginfo # The only difference is that a different primitive is exercised try: - _, resp = self._http.get2_cmd('{}/{}'.format(self._apiBase, vnfd['_id'])) + _, resp = self._http.get2_cmd("{}/{}".format(self._apiBase, vnfd["_id"])) if resp: return json.loads(resp) except NotFound: @@ -88,97 +88,122 @@ class Vnfd(object): self._logger.debug("") vnfd = self.get(name) headers = self._client._headers - headers['Accept'] = 'application/binary' - http_code, resp = self._http.get2_cmd('{}/{}/{}'.format(self._apiBase, vnfd['_id'], thing)) + headers["Accept"] = "application/binary" + http_code, resp = self._http.get2_cmd( + "{}/{}/{}".format(self._apiBase, vnfd["_id"], thing) + ) if resp: return json.loads(resp) def get_descriptor(self, name, filename): self._logger.debug("") - self.get_thing(name, 'vnfd', filename) + self.get_thing(name, "vnfd", filename) def get_package(self, name, filename): self._logger.debug("") - self.get_thing(name, 'package_content', filename) + self.get_thing(name, "package_content", filename) def get_artifact(self, name, artifact, filename): self._logger.debug("") - self.get_thing(name, 'artifacts/{}'.format(artifact), filename) + self.get_thing(name, "artifacts/{}".format(artifact), filename) def delete(self, name, force=False): self._logger.debug("") self._client.get_token() vnfd = self.get(name) - querystring = '' + querystring = "" if force: - querystring = '?FORCE=True' - http_code, resp = self._http.delete_cmd('{}/{}{}'.format(self._apiBase, - vnfd['_id'], querystring)) + querystring = "?FORCE=True" + http_code, resp = self._http.delete_cmd( + "{}/{}{}".format(self._apiBase, vnfd["_id"], querystring) + ) if http_code == 202: - print('Deletion in progress') + print("Deletion in progress") elif http_code == 204: - print('Deleted') + print("Deleted") else: msg = resp or "" raise ClientException("failed to delete vnfd {} - {}".format(name, msg)) - def create(self, filename, overwrite=None, update_endpoint=None, skip_charm_build=False, - override_epa=False, override_nonepa=False, override_paravirt=False): + def create( + self, + filename, + overwrite=None, + update_endpoint=None, + skip_charm_build=False, + override_epa=False, + override_nonepa=False, + override_paravirt=False, + ): self._logger.debug("") if os.path.isdir(filename): - filename = filename.rstrip('/') - filename = self._client.package_tool.build(filename, skip_validation=False, - skip_charm_build=skip_charm_build) + filename = filename.rstrip("/") + filename = self._client.package_tool.build( + filename, skip_validation=False, skip_charm_build=skip_charm_build + ) - print('Uploading package {}'.format(filename)) - self.create(filename, overwrite=overwrite, update_endpoint=update_endpoint, - override_epa=override_epa, override_nonepa=override_nonepa, - override_paravirt=override_paravirt) + print("Uploading package {}".format(filename)) + self.create( + filename, + overwrite=overwrite, + update_endpoint=update_endpoint, + override_epa=override_epa, + override_nonepa=override_nonepa, + override_paravirt=override_paravirt, + ) else: self._client.get_token() mime_type = magic.from_file(filename, mime=True) if mime_type is None: raise ClientException( "Unexpected MIME type for file {}: MIME type {}".format( - filename, mime_type) + filename, mime_type + ) ) headers = self._client._headers - headers['Content-Filename'] = basename(filename) - if mime_type in ['application/yaml', 'text/plain', 'application/json']: - headers['Content-Type'] = 'text/plain' - elif mime_type in ['application/gzip', 'application/x-gzip']: - headers['Content-Type'] = 'application/gzip' + headers["Content-Filename"] = basename(filename) + if mime_type in ["application/yaml", "text/plain", "application/json"]: + headers["Content-Type"] = "text/plain" + elif mime_type in ["application/gzip", "application/x-gzip"]: + headers["Content-Type"] = "application/gzip" else: raise ClientException( "Unexpected MIME type for file {}: MIME type {}".format( - filename, mime_type) + filename, mime_type + ) ) - special_override_string = '' + special_override_string = "" if override_epa or override_nonepa or override_paravirt: # If override for EPA, non-EPA or paravirt is required, get the descriptor data descriptor_data = None - if mime_type in ['application/yaml', 'text/plain', 'application/json']: + if mime_type in ["application/yaml", "text/plain", "application/json"]: with open(filename) as df: descriptor_data = df.read() - elif mime_type in ['application/gzip', 'application/x-gzip']: + elif mime_type in ["application/gzip", "application/x-gzip"]: tar_object = tarfile.open(filename, "r:gz") descriptor_list = [] for member in tar_object: if member.isreg(): - if '/' not in os.path.dirname(member.name) and member.name.endswith('.yaml'): + if "/" not in os.path.dirname( + member.name + ) and member.name.endswith(".yaml"): descriptor_list.append(member.name) if len(descriptor_list) > 1: - raise ClientException('Found more than one potential descriptor in the tar.gz file') + raise ClientException( + "Found more than one potential descriptor in the tar.gz file" + ) elif len(descriptor_list) == 0: - raise ClientException('No descriptor was found in the tar.gz file') + raise ClientException( + "No descriptor was found in the tar.gz file" + ) with tar_object.extractfile(descriptor_list[0]) as df: descriptor_data = df.read() tar_object.close() if not descriptor_data: - raise ClientException('Descriptor could not be read') + raise ClientException("Descriptor could not be read") desc_type, vnfd = validation_im().yaml_validation(descriptor_data) validation_im().pyangbind_validation(desc_type, vnfd) @@ -189,8 +214,8 @@ class Vnfd(object): # Get only the first descriptor in case there are many in the yaml file # k can be vnfd or etsi-nfv-vnfd:vnfd. This check is skipped first_vnfd = vnfd.get(k, {}) - vcd_list = first_vnfd.get('virtual-compute-desc', []) - vdu_list = first_vnfd.get('vdu', []) + vcd_list = first_vnfd.get("virtual-compute-desc", []) + vdu_list = first_vnfd.get("vdu", []) break for vcd_number, vcd in enumerate(vcd_list): @@ -200,22 +225,31 @@ class Vnfd(object): virtual_memory["numa-enabled"] = True virtual_memory["numa-node-policy"] = { "node-cnt": 1, - "mem-policy": "STRICT" + "mem-policy": "STRICT", } virtual_cpu = vcd["virtual-cpu"] virtual_cpu["pinning"] = { "policy": "static", - "thread-policy": "PREFER" + "thread-policy": "PREFER", } - cpu_override_string = "virtual-compute-desc.{}.virtual-cpu={};"\ - .format(vcd_number, quote(yaml.safe_dump(virtual_cpu))) - memory_override_string = "virtual-compute-desc.{}.virtual-memory={};"\ - .format(vcd_number, quote(yaml.safe_dump(virtual_memory))) - special_override_string = "{}{}{}".format(special_override_string, - cpu_override_string, memory_override_string) + cpu_override_string = ( + "virtual-compute-desc.{}.virtual-cpu={};".format( + vcd_number, quote(yaml.safe_dump(virtual_cpu)) + ) + ) + memory_override_string = ( + "virtual-compute-desc.{}.virtual-memory={};".format( + vcd_number, quote(yaml.safe_dump(virtual_memory)) + ) + ) + special_override_string = "{}{}{}".format( + special_override_string, + cpu_override_string, + memory_override_string, + ) - headers['Query-String-Format'] = 'yaml' + headers["Query-String-Format"] = "yaml" if override_nonepa: virtual_memory = vcd["virtual-memory"] virtual_memory["mempage-size"] = "" @@ -224,32 +258,52 @@ class Vnfd(object): virtual_cpu = vcd["virtual-cpu"] virtual_cpu["pinning"] = {} - cpu_override_string = "virtual-compute-desc.{}.virtual-cpu={};"\ - .format(vcd_number, quote(yaml.safe_dump(virtual_cpu))) - memory_override_string = "virtual-compute-desc.{}.virtual-memory={};"\ - .format(vcd_number, quote(yaml.safe_dump(virtual_memory))) - special_override_string = "{}{}{}".format(special_override_string, - cpu_override_string, memory_override_string) + cpu_override_string = ( + "virtual-compute-desc.{}.virtual-cpu={};".format( + vcd_number, quote(yaml.safe_dump(virtual_cpu)) + ) + ) + memory_override_string = ( + "virtual-compute-desc.{}.virtual-memory={};".format( + vcd_number, quote(yaml.safe_dump(virtual_memory)) + ) + ) + special_override_string = "{}{}{}".format( + special_override_string, + cpu_override_string, + memory_override_string, + ) if override_paravirt: for vdu_number, vdu in enumerate(vdu_list): for cpd_number, cpd in enumerate(vdu["int-cpd"]): - for vnir_number, vnir in enumerate(cpd['virtual-network-interface-requirement']): - special_override_string = "{}vdu.{}.int-cpd.{}.virtual-network-interface-" \ - "requirement.{}.virtual-interface.type=PARAVIRT;"\ - .format(special_override_string, vdu_number, cpd_number, vnir_number) + for vnir_number, vnir in enumerate( + cpd["virtual-network-interface-requirement"] + ): + special_override_string = ( + "{}vdu.{}.int-cpd.{}.virtual-network-interface-" + "requirement.{}.virtual-interface.type=PARAVIRT;".format( + special_override_string, + vdu_number, + cpd_number, + vnir_number, + ) + ) special_override_string = special_override_string.rstrip(";") headers["Content-File-MD5"] = utils.md5(filename) - http_header = ['{}: {}'.format(key, val) - for (key, val) in list(headers.items())] + http_header = [ + "{}: {}".format(key, val) for (key, val) in list(headers.items()) + ] self._http.set_http_header(http_header) if update_endpoint: - http_code, resp = self._http.put_cmd(endpoint=update_endpoint, filename=filename) + http_code, resp = self._http.put_cmd( + endpoint=update_endpoint, filename=filename + ) else: - ow_string = '' + ow_string = "" if special_override_string: if overwrite: overwrite = "{};{}".format(overwrite, special_override_string) @@ -257,25 +311,30 @@ class Vnfd(object): overwrite = special_override_string if overwrite: - ow_string = '?{}'.format(overwrite) - self._apiResource = '/vnf_packages_content' - self._apiBase = '{}{}{}'.format(self._apiName, - self._apiVersion, self._apiResource) - endpoint = '{}{}'.format(self._apiBase, ow_string) - http_code, resp = self._http.post_cmd(endpoint=endpoint, filename=filename) + ow_string = "?{}".format(overwrite) + self._apiResource = "/vnf_packages_content" + self._apiBase = "{}{}{}".format( + self._apiName, self._apiVersion, self._apiResource + ) + endpoint = "{}{}".format(self._apiBase, ow_string) + http_code, resp = self._http.post_cmd( + endpoint=endpoint, filename=filename + ) if http_code in (200, 201, 202): if resp: resp = json.loads(resp) - if not resp or 'id' not in resp: - raise ClientException('unexpected response from server: {}'.format(resp)) - print(resp['id']) + if not resp or "id" not in resp: + raise ClientException( + "unexpected response from server: {}".format(resp) + ) + print(resp["id"]) elif http_code == 204: - print('Updated') + print("Updated") def update(self, name, filename): self._logger.debug("") self._client.get_token() vnfd = self.get(name) - endpoint = '{}/{}/package_content'.format(self._apiBase, vnfd['_id']) + endpoint = "{}/{}/package_content".format(self._apiBase, vnfd["_id"]) self.create(filename=filename, update_endpoint=endpoint)