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