1 # Copyright 2018 Telefonica
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
9 # http://www.apache.org/licenses/LICENSE-2.0
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
21 from osmclient
.common
.exceptions
import NotFound
22 from osmclient
.common
.exceptions
import ClientException
23 from osmclient
.common
import utils
26 from os
.path
import basename
33 def __init__(self
, http
=None, client
=None):
36 self
._logger
= logging
.getLogger('osmclient')
37 self
._apiName
= '/nsd'
38 self
._apiVersion
= '/v1'
39 self
._apiResource
= '/ns_descriptors'
40 self
._apiBase
= '{}{}{}'.format(self
._apiName
,
41 self
._apiVersion
, self
._apiResource
)
42 #self._apiBase='/nsds'
44 def list(self
, filter=None):
45 self
._logger
.debug("")
46 self
._client
.get_token()
49 filter_string
= '?{}'.format(filter)
50 _
, resp
= self
._http
.get2_cmd('{}{}'.format(self
._apiBase
, filter_string
))
51 #print(yaml.safe_dump(resp))
53 return json
.loads(resp
)
57 self
._logger
.debug("")
58 self
._client
.get_token()
59 if utils
.validate_uuid4(name
):
60 for nsd
in self
.list():
61 if name
== nsd
['_id']:
64 for nsd
in self
.list():
65 if 'name' in nsd
and name
== nsd
['name']:
67 raise NotFound("nsd {} not found".format(name
))
69 def get_individual(self
, name
):
70 self
._logger
.debug("")
71 # Call to get_token not required, because will be implicitly called by get.
74 # It is redundant, since the previous one already gets the whole nsdinfo
75 # The only difference is that a different primitive is exercised
76 _
, resp
= self
._http
.get2_cmd('{}/{}'.format(self
._apiBase
, nsd
['_id']))
77 #print(yaml.safe_dump(resp))
79 return json
.loads(resp
)
81 raise NotFound("nsd '{}' not found".format(name
))
82 raise NotFound("nsd '{}' not found".format(name
))
84 def get_thing(self
, name
, thing
, filename
):
85 self
._logger
.debug("")
86 # Call to get_token not required, because will be implicitly called by get.
88 headers
= self
._client
._headers
89 headers
['Accept'] = 'application/binary'
90 http_code
, resp
= self
._http
.get2_cmd('{}/{}/{}'.format(self
._apiBase
, nsd
['_id'], thing
))
91 #print('HTTP CODE: {}'.format(http_code))
92 #print('RESP: {}'.format(resp))
93 #if http_code in (200, 201, 202, 204):
96 return json
.loads(resp
)
101 # msg = json.loads(resp)
104 raise ClientException("failed to get {} from {} - {}".format(thing
, name
, msg
))
106 def get_descriptor(self
, name
, filename
):
107 self
._logger
.debug("")
108 self
.get_thing(name
, 'nsd', filename
)
110 def get_package(self
, name
, filename
):
111 self
._logger
.debug("")
112 self
.get_thing(name
, 'package_content', filename
)
114 def get_artifact(self
, name
, artifact
, filename
):
115 self
._logger
.debug("")
116 self
.get_thing(name
, 'artifacts/{}'.format(artifact
), filename
)
118 def delete(self
, name
, force
=False):
119 self
._logger
.debug("")
123 querystring
= '?FORCE=True'
124 http_code
, resp
= self
._http
.delete_cmd('{}/{}{}'.format(self
._apiBase
,
125 nsd
['_id'], querystring
))
126 #print('HTTP CODE: {}'.format(http_code))
127 #print('RESP: {}'.format(resp))
129 print('Deletion in progress')
130 elif http_code
== 204:
136 # msg = json.loads(resp)
139 raise ClientException("failed to delete nsd {} - {}".format(name
, msg
))
141 def create(self
, filename
, overwrite
=None, update_endpoint
=None):
142 self
._logger
.debug("")
143 self
._client
.get_token()
144 mime_type
= magic
.from_file(filename
, mime
=True)
145 if mime_type
is None:
146 raise ClientException(
147 "failed to guess MIME type for file '{}'".format(filename
))
148 headers
= self
._client
._headers
149 headers
['Content-Filename'] = basename(filename
)
150 if mime_type
in ['application/yaml', 'text/plain', 'application/json']:
151 headers
['Content-Type'] = 'text/plain'
152 elif mime_type
in ['application/gzip', 'application/x-gzip']:
153 headers
['Content-Type'] = 'application/gzip'
154 #headers['Content-Type'] = 'application/binary'
155 # Next three lines are to be removed in next version
156 #headers['Content-Filename'] = basename(filename)
157 #file_size = stat(filename).st_size
158 #headers['Content-Range'] = 'bytes 0-{}/{}'.format(file_size - 1, file_size)
160 raise ClientException(
161 "Unexpected MIME type for file {}: MIME type {}".format(
164 headers
["Content-File-MD5"] = utils
.md5(filename
)
165 http_header
= ['{}: {}'.format(key
,val
)
166 for (key
,val
) in list(headers
.items())]
167 self
._http
.set_http_header(http_header
)
169 http_code
, resp
= self
._http
.put_cmd(endpoint
=update_endpoint
, filename
=filename
)
173 ow_string
= '?{}'.format(overwrite
)
174 self
._apiResource
= '/ns_descriptors_content'
175 self
._apiBase
= '{}{}{}'.format(self
._apiName
,
176 self
._apiVersion
, self
._apiResource
)
177 endpoint
= '{}{}'.format(self
._apiBase
,ow_string
)
178 http_code
, resp
= self
._http
.post_cmd(endpoint
=endpoint
, filename
=filename
)
179 #print('HTTP CODE: {}'.format(http_code))
180 #print('RESP: {}'.format(resp))
181 if http_code
in (200, 201, 202):
183 resp
= json
.loads(resp
)
184 if not resp
or 'id' not in resp
:
185 raise ClientException('unexpected response from server - {}'.format(resp
))
187 elif http_code
== 204:
190 # msg = "Error {}".format(http_code)
193 # msg = "{} - {}".format(msg, json.loads(resp))
195 # msg = "{} - {}".format(msg, resp)
196 # raise ClientException("failed to create/update nsd - {}".format(msg))
198 def update(self
, name
, filename
):
199 self
._logger
.debug("")
201 endpoint
= '{}/{}/nsd_content'.format(self
._apiBase
, nsd
['_id'])
202 self
.create(filename
=filename
, update_endpoint
=endpoint
)