catalog-be/src/main/resources/scripts/sdcBePy/common/normative/toscaElements.py

   1 import json
   2 import zipfile
   3
   4 import pycurl
   5
   6 from sdcBePy.common.logger import debug, log, print_name_and_return_code, print_and_exit
   7 from sdcBePy.common.sdcBeProxy import SdcBeProxy
   8 from sdcBePy.common.errors import ResourceCreationError
   9
  10
  11 def process_and_create_normative_element(normative_element,
  12                                          scheme=None, be_host=None, be_port=None, admin_user=None, sdc_be_proxy=None,
  13                                          debug=False,
  14                                          exit_on_success=False):
  15     if sdc_be_proxy is None:
  16         sdc_be_proxy = SdcBeProxy(be_host, be_port, scheme, admin_user, debug=debug)
  17
  18     file_dir, url_suffix, element_name, element_from_name, with_metadata = normative_element.get_parameters()
  19     _create_normative_element(sdc_be_proxy,
  20                               file_dir,
  21                               url_suffix,
  22                               element_name,
  23                               element_from_name,
  24                               with_metadata,
  25                               exit_on_success)
  26
  27
  28 def _create_normative_element(sdc_be_proxy, file_dir,
  29                               url_suffix, element_name, element_form_name, with_metadata=False,
  30                               exit_on_success=False):
  31     result = _send_request(sdc_be_proxy,
  32                            file_dir,
  33                            url_suffix,
  34                            element_name,
  35                            element_form_name,
  36                            with_metadata)
  37     print_and_check_result(result, exit_on_success)
  38
  39
  40 def _send_request(sdc_be_proxy, file_dir, url_suffix, element_name,
  41                   element_form_name,
  42                   with_metadata=False):
  43     try:
  44         log("create normative element ", element_name)
  45
  46         type_file_name = file_dir + element_name
  47         multi_part_form_data = _create_multipart_form_data(element_form_name, type_file_name, with_metadata,
  48                                                            element_name)
  49
  50         http_res = sdc_be_proxy.post_file(url_suffix, multi_part_form_data)
  51         if http_res is not None:
  52             debug("http response =", http_res)
  53
  54         response = sdc_be_proxy.get_response_from_buffer()
  55         debug("response buffer", response)
  56         # c.close()
  57         return element_name, http_res, response
  58
  59     except Exception as inst:
  60         print("ERROR=" + str(inst))
  61         return element_name, None, None
  62
  63
  64 def _create_multipart_form_data(element_form_name, type_file_name, with_metadata, element_name):
  65     tosca_type_zip_part = _create_zip_file_multi_part(element_form_name, type_file_name, element_name)
  66     multi_part_form_data = [tosca_type_zip_part]
  67     if with_metadata:
  68         metadata_type_part = _create_metadata_multipart(type_file_name)
  69         multi_part_form_data.append(metadata_type_part)
  70     debug(multi_part_form_data)
  71     return multi_part_form_data
  72
  73
  74 def _create_metadata_multipart(type_file_name):
  75     metadata = _create_json_metadata_str(type_file_name)
  76     return "toscaTypeMetadata", metadata
  77
  78
  79 def _create_zip_file_multi_part(element_form_name, type_file_name, element_name):
  80     zf = zipfile.ZipFile(type_file_name + ".zip", "w")
  81     zf.write(type_file_name + '.yml', element_name + '.yml')
  82     zf.close()
  83
  84     tosca_type_zip_path = type_file_name + ".zip"
  85     tosca_type_zip_part = (element_form_name, (pycurl.FORM_FILE, tosca_type_zip_path))
  86     return tosca_type_zip_part
  87
  88
  89 def _create_json_metadata_str(file_name):
  90     type_metadata_json_file = file_name + ".json"
  91     debug(type_metadata_json_file)
  92     json_file = open(type_metadata_json_file)
  93
  94     debug("before load json")
  95     json_data = json.load(json_file, strict=False)
  96     debug(json_data)
  97
  98     return json.dumps(json_data)
  99
 100
 101 def print_and_check_result(result, exit_on_success):
 102     if result is not None:
 103         print_name_and_return_code(result[0], result[1])
 104         if result[1] is None or result[1] not in [200, 201, 409]:
 105             raise ResourceCreationError("Failed to create the normatives elements!!", 1)
 106         else:
 107             if exit_on_success is True:
 108                 print_and_exit(0, "All normatives elements created successfully!!")
 109     else:
 110         raise ResourceCreationError("Results is None!", 1)