Updated .gitignore: .idea
[osm/osmclient.git] / 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