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 yaml
  25 import json
  26 import magic
  27 #from os import stat
  28 #from os.path import basename
  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 yaml.safe_dump(resp2)
  78         if resp:
  79             #store in a file
  80             return resp
  81         raise NotFound("vnfd {} not found".format(name))
  82
  83     def get_descriptor(self, name, filename):
  84         self.get_thing(name, 'vnfd', 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, force=False):
  93         vnfd = self.get(name)
  94         querystring = ''
  95         if force:
  96             querystring = '?FORCE=True'
  97         http_code, resp = self._http.delete_cmd('{}/{}{}'.format(self._apiBase,
  98                                          vnfd['_id'], querystring))
  99         if resp:
 100             resp = json.loads(resp)
 101         #print 'RESP: {}'.format(resp)
 102         if http_code == 202:
 103             print 'Deletion in progress'
 104         elif http_code == 204:
 105             print 'Deleted'
 106         else:
 107             raise ClientException("failed to delete vnfd {}: {}".format(name, resp))
 108
 109     def create(self, filename, overwrite=None, update_endpoint=None):
 110         mime_type = magic.from_file(filename, mime=True)
 111         if mime_type is None:
 112             raise ClientException(
 113                      "failed to guess MIME type for file '{}'".format(filename))
 114         headers= self._client._headers
 115         if mime_type in ['application/yaml', 'text/plain']:
 116             headers['Content-Type'] = 'application/yaml'
 117         elif mime_type == 'application/gzip':
 118             headers['Content-Type'] = 'application/gzip'
 119             #headers['Content-Type'] = 'application/binary'
 120             # Next three lines are to be removed in next version
 121             #headers['Content-Filename'] = basename(filename)
 122             #file_size = stat(filename).st_size
 123             #headers['Content-Range'] = 'bytes 0-{}/{}'.format(file_size - 1, file_size)
 124         else:
 125             raise ClientException(
 126                      "Unexpected MIME type for file {}: MIME type {}".format(
 127                          filename, mime_type)
 128                   )
 129         headers["Content-File-MD5"] = utils.md5(filename)
 130         http_header = ['{}: {}'.format(key,val)
 131                       for (key,val) in headers.items()]
 132         self._http.set_http_header(http_header)
 133         if update_endpoint:
 134             http_code, resp = self._http.put_cmd(endpoint=update_endpoint, filename=filename)
 135         else:
 136             ow_string = ''
 137             if overwrite:
 138                 ow_string = '?{}'.format(overwrite)
 139             self._apiResource = '/vnf_packages_content'
 140             self._apiBase = '{}{}{}'.format(self._apiName,
 141                                             self._apiVersion, self._apiResource)
 142             endpoint = '{}{}'.format(self._apiBase,ow_string)
 143             http_code, resp = self._http.post_cmd(endpoint=endpoint, filename=filename)
 144         if resp:
 145             resp = json.loads(resp)
 146         #print resp
 147         if not resp or 'id' not in resp:
 148             raise ClientException("failed to upload package")
 149         else:
 150             print resp['id']
 151
 152     def update(self, name, filename):
 153         vnfd = self.get(name)
 154         endpoint = '{}/{}/vnfd_content'.format(self._apiBase, vnfd['_id'])
 155         self.create(filename=filename, update_endpoint=endpoint)
 156