63308d6c5e63d1de0525a3bce3662f194ca1464b
[osm/osmclient.git] / 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 json
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 http_code, resp = self._http.get2_cmd('{}/{}/{}'.format(self._apiBase, nsd['_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, 'nsd', 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 nsd = self.get(name)
103 querystring = ''
104 if force:
105 querystring = '?FORCE=True'
106 http_code, resp = self._http.delete_cmd('{}/{}{}'.format(self._apiBase,
107 nsd['_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 resp = json.loads(resp)
119 except ValueError:
120 msg = resp
121 raise ClientException("failed to delete nsd {} - {}".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 if mime_type in ['application/yaml', 'text/plain']:
130 headers['Content-Type'] = 'application/yaml'
131 elif mime_type in ['application/gzip', 'application/x-gzip']:
132 headers['Content-Type'] = 'application/gzip'
133 #headers['Content-Type'] = 'application/binary'
134 # Next three lines are to be removed in next version
135 #headers['Content-Filename'] = basename(filename)
136 #file_size = stat(filename).st_size
137 #headers['Content-Range'] = 'bytes 0-{}/{}'.format(file_size - 1, file_size)
138 else:
139 raise ClientException(
140 "Unexpected MIME type for file {}: MIME type {}".format(
141 filename, mime_type)
142 )
143 headers["Content-File-MD5"] = utils.md5(filename)
144 http_header = ['{}: {}'.format(key,val)
145 for (key,val) in list(headers.items())]
146 self._http.set_http_header(http_header)
147 if update_endpoint:
148 http_code, resp = self._http.put_cmd(endpoint=update_endpoint, filename=filename)
149 else:
150 ow_string = ''
151 if overwrite:
152 ow_string = '?{}'.format(overwrite)
153 self._apiResource = '/ns_descriptors_content'
154 self._apiBase = '{}{}{}'.format(self._apiName,
155 self._apiVersion, self._apiResource)
156 endpoint = '{}{}'.format(self._apiBase,ow_string)
157 http_code, resp = self._http.post_cmd(endpoint=endpoint, filename=filename)
158 #print 'HTTP CODE: {}'.format(http_code)
159 #print 'RESP: {}'.format(resp)
160 if http_code in (200, 201, 202, 204):
161 if resp:
162 resp = json.loads(resp)
163 if not resp or 'id' not in resp:
164 raise ClientException('unexpected response from server - {}'.format(
165 resp))
166 print(resp['id'])
167 else:
168 msg = "Error {}".format(http_code)
169 if resp:
170 try:
171 msg = "{} - {}".format(msg, json.loads(resp))
172 except ValueError:
173 msg = "{} - {}".format(msg, resp)
174 raise ClientException("failed to create/update nsd - {}".format(msg))
175
176 def update(self, name, filename):
177 nsd = self.get(name)
178 endpoint = '{}/{}/nsd_content'.format(self._apiBase, nsd['_id'])
179 self.create(filename=filename, update_endpoint=endpoint)
180