3288e954d0f489841b8e1d86951f8201ea998690
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
27 #from os.path import basename
31 def __init__(self
, http
=None, client
=None):
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'
41 def list(self
, filter=None):
44 filter_string
= '?{}'.format(filter)
45 resp
= self
._http
.get_cmd('{}{}'.format(self
._apiBase
, filter_string
))
46 #print yaml.safe_dump(resp)
52 if utils
.validate_uuid4(name
):
53 for nsd
in self
.list():
54 if name
== nsd
['_id']:
57 for nsd
in self
.list():
58 if 'name' in nsd
and name
== nsd
['name']:
60 raise NotFound("nsd {} not found".format(name
))
62 def get_individual(self
, 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)
70 raise NotFound("nsd {} not found".format(name
))
72 def get_thing(self
, name
, thing
, filename
):
74 headers
= self
._client
._headers
75 headers
['Accept'] = 'application/binary'
76 resp2
= self
._http
.get2_cmd('{}/{}/{}'.format(self
._apiBase
, nsd
['_id'], thing
))
77 #print yaml.safe_dump(resp2)
81 raise NotFound("nsd {} not found".format(name
))
83 def get_descriptor(self
, name
, filename
):
84 self
.get_thing(name
, 'nsd', filename
)
86 def get_package(self
, name
, filename
):
87 self
.get_thing(name
, 'package_content', filename
)
89 def get_artifact(self
, name
, artifact
, filename
):
90 self
.get_thing(name
, 'artifacts/{}'.format(artifact
), filename
)
92 def delete(self
, name
):
94 resp
= self
._http
.delete_cmd('{}/{}'.format(self
._apiBase
, nsd
['_id']))
95 #print 'RESP: '.format(resp)
99 raise ClientException("failed to delete nsd {}: {}".format(name
, resp
))
101 def create(self
, filename
, overwrite
=None, update_endpoint
=None):
102 mime_type
= magic
.from_file(filename
, mime
=True)
103 if mime_type
is None:
104 raise ClientException(
105 "failed to guess MIME type for file '{}'".format(filename
))
106 headers
= self
._client
._headers
107 if mime_type
in ['application/yaml', 'text/plain']:
108 headers
['Content-Type'] = 'application/yaml'
109 elif mime_type
== 'application/gzip':
110 headers
['Content-Type'] = 'application/gzip'
111 #headers['Content-Type'] = 'application/binary'
112 # Next three lines are to be removed in next version
113 #headers['Content-Filename'] = basename(filename)
114 #file_size = stat(filename).st_size
115 #headers['Content-Range'] = 'bytes 0-{}/{}'.format(file_size - 1, file_size)
117 raise ClientException(
118 "Unexpected MIME type for file {}: MIME type {}".format(
121 headers
["Content-File-MD5"] = utils
.md5(filename
)
122 http_header
= ['{}: {}'.format(key
,val
)
123 for (key
,val
) in headers
.items()]
124 self
._http
.set_http_header(http_header
)
126 resp
= self
._http
.put_cmd(endpoint
=update_endpoint
, filename
=filename
)
130 ow_string
= '?{}'.format(overwrite
)
131 self
._apiResource
= '/ns_descriptors_content'
132 self
._apiBase
= '{}{}{}'.format(self
._apiName
,
133 self
._apiVersion
, self
._apiResource
)
134 endpoint
= '{}{}'.format(self
._apiBase
,ow_string
)
135 resp
= self
._http
.post_cmd(endpoint
=endpoint
, filename
=filename
)
137 if not resp
or 'id' not in resp
:
138 raise ClientException("failed to upload package")
142 def update(self
, name
, filename
):
144 endpoint
= '{}/{}/nsd_content'.format(self
._apiBase
, nsd
['_id'])
145 self
.create(filename
=filename
, update_endpoint
=endpoint
)