Support of new OSM NB API following SOL005

[osm/osmclient.git] / osmclient / sol005 / package.py

diff --git a/osmclient/sol005/package.py b/osmclient/sol005/package.py
new file mode 100644 (file)
index 0000000..f0a9e89
--- /dev/null
+++ b/osmclient/sol005/package.py
@@ -0,0 +1,68 @@
+# Copyright 2018 Telefonica
+#
+# All Rights Reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+"""
+OSM package API handling
+"""
+
+import tarfile
+import re
+import yaml
+#from os import stat
+#from os.path import basename
+from osmclient.common.exceptions import ClientException
+from osmclient.common.exceptions import NotFound
+from osmclient.common import utils
+
+
+class Package(object):
+    def __init__(self, http=None, client=None):
+        self._client = client
+        self._http = http
+
+    def get_key_val_from_pkg(self, descriptor_file):
+        utils.get_key_val_from_pkg(descriptor_file)
+
+    def upload(self, filename):
+        pkg_type = utils.get_key_val_from_pkg(filename)
+        if pkg_type is None:
+            raise ClientException("Cannot determine package type")
+        if pkg_type['type'] == 'nsd':
+            endpoint = '/nsd/v1/ns_descriptors'
+        else:
+            endpoint = '/vnfpkgm/v1/vnf_packages'
+        #endpoint = '/nsds' if pkg_type['type'] == 'nsd' else '/vnfds'
+        #print 'Endpoint: {}'.format(endpoint)
+        headers = self._client._headers
+        headers['Content-Type'] = 'application/gzip'
+        #headers['Content-Type'] = 'application/binary'
+        # Next three lines are to be removed in next version
+        #headers['Content-Filename'] = basename(filename)
+        #file_size = stat(filename).st_size
+        #headers['Content-Range'] = 'bytes 0-{}/{}'.format(file_size - 1, file_size)
+        headers["Content-File-MD5"] = utils.md5(filename)
+        http_header = ['{}: {}'.format(key,val)
+                      for (key,val) in headers.items()]
+        self._http.set_http_header(http_header)
+        resp = self._http.post_cmd(endpoint=endpoint,
+                                   filename=filename)
+        #print 'RESP: {}'.format(yaml.safe_dump(resp))
+        if not resp or 'id' not in resp:
+            raise ClientException("failed to upload package")
+        else:
+            print resp['id']
+
+