osmclient/sol005/nsd.py

   1 # Copyright 2018 Telefonica
   2 #
   3 # All Rights Reserved.
   4 #
   5 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
   6 #    not use this file except in compliance with the License. You may obtain
   7 #    a copy of the License at
   8 #
   9 #         http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 #    Unless required by applicable law or agreed to in writing, software
  12 #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  13 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  14 #    License for the specific language governing permissions and limitations
  15 #    under the License.
  16
  17 """
  18 OSM nsd API handling
  19 """
  20
  21 from osmclient.common.exceptions import NotFound
  22 from osmclient.common.exceptions import ClientException
  23 from osmclient.common import utils
  24 import yaml
  25 import magic
  26 #from os import stat
  27 #from os.path import basename
  28
  29 class Nsd(object):
  30
  31     def __init__(self, http=None, client=None):
  32         self._http = http
  33         self._client = client
  34         self._apiName = '/nsd'
  35         self._apiVersion = '/v1'
  36         self._apiResource = '/ns_descriptors'
  37         self._apiBase = '{}{}{}'.format(self._apiName,
  38                                         self._apiVersion, self._apiResource)
  39         #self._apiBase='/nsds'
  40
  41     def list(self, filter=None):
  42         filter_string = ''
  43         if filter:
  44             filter_string = '?{}'.format(filter)
  45         resp = self._http.get_cmd('{}{}'.format(self._apiBase, filter_string))
  46         #print yaml.safe_dump(resp)
  47         if resp:
  48             return resp
  49         return list()
  50
  51     def get(self, name):
  52         if utils.validate_uuid4(name):
  53             for nsd in self.list():
  54                 if name == nsd['_id']:
  55                     return nsd
  56         else:
  57             for nsd in self.list():
  58                 if 'name' in nsd and name == nsd['name']:
  59                     return nsd
  60         raise NotFound("nsd {} not found".format(name))
  61
  62     def get_individual(self, name):
  63         nsd = self.get(name)
  64         # It is redundant, since the previous one already gets the whole nsdinfo
  65         # The only difference is that a different primitive is exercised
  66         resp = self._http.get_cmd('{}/{}'.format(self._apiBase, nsd['_id']))
  67         #print yaml.safe_dump(resp)
  68         if resp:
  69             return resp
  70         raise NotFound("nsd {} not found".format(name))
  71
  72     def get_thing(self, name, thing, filename):
  73         nsd = self.get(name)
  74         headers = self._client._headers
  75         headers['Accept'] = 'application/binary'
  76         resp2 = self._http.get2_cmd('{}/{}/{}'.format(self._apiBase, nsd['_id'], thing))
  77         #print yaml.safe_dump(resp2)
  78         if resp2:
  79             #store in a file
  80             return resp2
  81         raise NotFound("nsd {} not found".format(name))
  82
  83     def get_descriptor(self, name, filename):
  84         self.get_thing(name, 'nsd', filename)
  85
  86     def get_package(self, name, filename):
  87         self.get_thing(name, 'package_content', filename)
  88
  89     def get_artifact(self, name, artifact, filename):
  90         self.get_thing(name, 'artifacts/{}'.format(artifact), filename)
  91
  92     def delete(self, name):
  93         nsd = self.get(name)
  94         resp = self._http.delete_cmd('{}/{}'.format(self._apiBase, nsd['_id']))
  95         #print 'RESP: '.format(resp)
  96         if resp is None:
  97             print 'Deleted'
  98         else:
  99             raise ClientException("failed to delete nsd {}: {}".format(name, resp))
 100
 101     def create(self, filename, overwrite=None, update_endpoint=None):
 102         mime_type = magic.from_file(filename, mime=True)
 103         if mime_type is None:
 104             raise ClientException(
 105                      "failed to guess MIME type for file '{}'".format(filename))
 106         headers= self._client._headers
 107         if mime_type in ['application/yaml', 'text/plain']:
 108             headers['Content-Type'] = 'application/yaml'
 109         elif mime_type == 'application/gzip':
 110             headers['Content-Type'] = 'application/gzip'
 111             #headers['Content-Type'] = 'application/binary'
 112             # Next three lines are to be removed in next version
 113             #headers['Content-Filename'] = basename(filename)
 114             #file_size = stat(filename).st_size
 115             #headers['Content-Range'] = 'bytes 0-{}/{}'.format(file_size - 1, file_size)
 116         else:
 117             raise ClientException(
 118                      "Unexpected MIME type for file {}: MIME type {}".format(
 119                          filename, mime_type)
 120                   )
 121         headers["Content-File-MD5"] = utils.md5(filename)
 122         http_header = ['{}: {}'.format(key,val)
 123                       for (key,val) in headers.items()]
 124         self._http.set_http_header(http_header)
 125         if update_endpoint:
 126             resp = self._http.put_cmd(endpoint=update_endpoint, filename=filename)
 127         else:
 128             ow_string = ''
 129             if overwrite:
 130                 ow_string = '?{}'.format(overwrite)
 131             self._apiResource = '/ns_descriptors_content'
 132             self._apiBase = '{}{}{}'.format(self._apiName,
 133                                             self._apiVersion, self._apiResource)
 134             endpoint = '{}{}'.format(self._apiBase,ow_string)
 135             resp = self._http.post_cmd(endpoint=endpoint, filename=filename)
 136         #print resp
 137         if not resp or 'id' not in resp:
 138             raise ClientException("failed to upload package")
 139         else:
 140             print resp['id']
 141
 142     def update(self, name, filename):
 143         nsd = self.get(name)
 144         endpoint = '{}/{}/nsd_content'.format(self._apiBase, nsd['_id'])
 145         self.create(filename=filename, update_endpoint=endpoint)
 146