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 json
  26 import magic
  27 #from os import stat
  28 #from os.path import basename
  29
  30 class Nsd(object):
  31
  32     def __init__(self, http=None, client=None):
  33         self._http = http
  34         self._client = client
  35         self._apiName = '/nsd'
  36         self._apiVersion = '/v1'
  37         self._apiResource = '/ns_descriptors'
  38         self._apiBase = '{}{}{}'.format(self._apiName,
  39                                         self._apiVersion, self._apiResource)
  40         #self._apiBase='/nsds'
  41
  42     def list(self, filter=None):
  43         filter_string = ''
  44         if filter:
  45             filter_string = '?{}'.format(filter)
  46         resp = self._http.get_cmd('{}{}'.format(self._apiBase, filter_string))
  47         #print yaml.safe_dump(resp)
  48         if resp:
  49             return resp
  50         return list()
  51
  52     def get(self, name):
  53         if utils.validate_uuid4(name):
  54             for nsd in self.list():
  55                 if name == nsd['_id']:
  56                     return nsd
  57         else:
  58             for nsd in self.list():
  59                 if 'name' in nsd and name == nsd['name']:
  60                     return nsd
  61         raise NotFound("nsd {} not found".format(name))
  62
  63     def get_individual(self, name):
  64         nsd = self.get(name)
  65         # It is redundant, since the previous one already gets the whole nsdinfo
  66         # The only difference is that a different primitive is exercised
  67         resp = self._http.get_cmd('{}/{}'.format(self._apiBase, nsd['_id']))
  68         #print yaml.safe_dump(resp)
  69         if resp:
  70             return resp
  71         raise NotFound("nsd {} not found".format(name))
  72
  73     def get_thing(self, name, thing, filename):
  74         nsd = self.get(name)
  75         headers = self._client._headers
  76         headers['Accept'] = 'application/binary'
  77         http_code, resp = self._http.get2_cmd('{}/{}/{}'.format(self._apiBase, nsd['_id'], thing))
  78         #print yaml.safe_dump(resp2)
  79         if resp:
  80             #store in a file
  81             return resp
  82         raise NotFound("nsd {} not found".format(name))
  83
  84     def get_descriptor(self, name, filename):
  85         self.get_thing(name, 'nsd', filename)
  86
  87     def get_package(self, name, filename):
  88         self.get_thing(name, 'package_content', filename)
  89
  90     def get_artifact(self, name, artifact, filename):
  91         self.get_thing(name, 'artifacts/{}'.format(artifact), filename)
  92
  93     def delete(self, name, force=False):
  94         nsd = self.get(name)
  95         querystring = ''
  96         if force:
  97             querystring = '?FORCE=True'
  98         http_code, resp = self._http.delete_cmd('{}/{}{}'.format(self._apiBase,
  99                                          nsd['_id'], querystring))
 100         if resp:
 101             resp = json.loads(resp)
 102         #print 'RESP: {}'.format(resp)
 103         if http_code == 202:
 104             print 'Deletion in progress'
 105         elif http_code == 204:
 106             print 'Deleted'
 107         else:
 108             raise ClientException("failed to delete nsd {}: {}".format(name, resp))
 109
 110     def create(self, filename, overwrite=None, update_endpoint=None):
 111         mime_type = magic.from_file(filename, mime=True)
 112         if mime_type is None:
 113             raise ClientException(
 114                      "failed to guess MIME type for file '{}'".format(filename))
 115         headers= self._client._headers
 116         if mime_type in ['application/yaml', 'text/plain']:
 117             headers['Content-Type'] = 'application/yaml'
 118         elif mime_type == 'application/gzip':
 119             headers['Content-Type'] = 'application/gzip'
 120             #headers['Content-Type'] = 'application/binary'
 121             # Next three lines are to be removed in next version
 122             #headers['Content-Filename'] = basename(filename)
 123             #file_size = stat(filename).st_size
 124             #headers['Content-Range'] = 'bytes 0-{}/{}'.format(file_size - 1, file_size)
 125         else:
 126             raise ClientException(
 127                      "Unexpected MIME type for file {}: MIME type {}".format(
 128                          filename, mime_type)
 129                   )
 130         headers["Content-File-MD5"] = utils.md5(filename)
 131         http_header = ['{}: {}'.format(key,val)
 132                       for (key,val) in headers.items()]
 133         self._http.set_http_header(http_header)
 134         if update_endpoint:
 135             http_code, resp = self._http.put_cmd(endpoint=update_endpoint, filename=filename)
 136         else:
 137             ow_string = ''
 138             if overwrite:
 139                 ow_string = '?{}'.format(overwrite)
 140             self._apiResource = '/ns_descriptors_content'
 141             self._apiBase = '{}{}{}'.format(self._apiName,
 142                                             self._apiVersion, self._apiResource)
 143             endpoint = '{}{}'.format(self._apiBase,ow_string)
 144             http_code, resp = self._http.post_cmd(endpoint=endpoint, filename=filename)
 145         if resp:
 146             resp = json.loads(resp)
 147         #print resp
 148         if not resp or 'id' not in resp:
 149             raise ClientException("failed to upload package")
 150         else:
 151             print resp['id']
 152
 153     def update(self, name, filename):
 154         nsd = self.get(name)
 155         endpoint = '{}/{}/nsd_content'.format(self._apiBase, nsd['_id'])
 156         self.create(filename=filename, update_endpoint=endpoint)
 157