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
28 #from os.path import basename
32 def __init__(self
, http
=None, client
=None):
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'
42 def list(self
, filter=None):
45 filter_string
= '?{}'.format(filter)
46 resp
= self
._http
.get_cmd('{}{}'.format(self
._apiBase
,filter_string
))
52 if utils
.validate_uuid4(name
):
53 for vnfd
in self
.list():
54 if name
== vnfd
['_id']:
57 for vnfd
in self
.list():
58 if 'name' in vnfd
and name
== vnfd
['name']:
60 raise NotFound("vnfd {} not found".format(name
))
62 def get_individual(self
, 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)
70 raise NotFound("vnfd {} not found".format(name
))
72 def get_thing(self
, name
, thing
, filename
):
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)
81 raise NotFound("vnfd {} not found".format(name
))
83 def get_descriptor(self
, name
, filename
):
84 self
.get_thing(name
, 'vnfd', 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 http_code
, resp
= self
._http
.delete_cmd('{}/{}'.format(self
._apiBase
,vnfd
['_id']))
96 resp
= json
.loads(resp
)
97 #print 'RESP: {}'.format(resp)
99 print 'Deletion in progress'
100 elif http_code
== 204:
103 raise ClientException("failed to delete vnfd {}: {}".format(name
, resp
))
105 def create(self
, filename
, overwrite
=None, update_endpoint
=None):
106 mime_type
= magic
.from_file(filename
, mime
=True)
107 if mime_type
is None:
108 raise ClientException(
109 "failed to guess MIME type for file '{}'".format(filename
))
110 headers
= self
._client
._headers
111 if mime_type
in ['application/yaml', 'text/plain']:
112 headers
['Content-Type'] = 'application/yaml'
113 elif mime_type
== 'application/gzip':
114 headers
['Content-Type'] = 'application/gzip'
115 #headers['Content-Type'] = 'application/binary'
116 # Next three lines are to be removed in next version
117 #headers['Content-Filename'] = basename(filename)
118 #file_size = stat(filename).st_size
119 #headers['Content-Range'] = 'bytes 0-{}/{}'.format(file_size - 1, file_size)
121 raise ClientException(
122 "Unexpected MIME type for file {}: MIME type {}".format(
125 headers
["Content-File-MD5"] = utils
.md5(filename
)
126 http_header
= ['{}: {}'.format(key
,val
)
127 for (key
,val
) in headers
.items()]
128 self
._http
.set_http_header(http_header
)
130 http_code
, resp
= self
._http
.put_cmd(endpoint
=update_endpoint
, filename
=filename
)
134 ow_string
= '?{}'.format(overwrite
)
135 self
._apiResource
= '/vnf_packages_content'
136 self
._apiBase
= '{}{}{}'.format(self
._apiName
,
137 self
._apiVersion
, self
._apiResource
)
138 endpoint
= '{}{}'.format(self
._apiBase
,ow_string
)
139 http_code
, resp
= self
._http
.post_cmd(endpoint
=endpoint
, filename
=filename
)
141 resp
= json
.loads(resp
)
143 if not resp
or 'id' not in resp
:
144 raise ClientException("failed to upload package")
148 def update(self
, name
, filename
):
149 vnfd
= self
.get(name
)
150 endpoint
= '{}/{}/vnfd_content'.format(self
._apiBase
, vnfd
['_id'])
151 self
.create(filename
=filename
, update_endpoint
=endpoint
)