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
29 def __init__(self
, http
=None, client
=None):
32 self
._apiName
= '/pdu'
33 self
._apiVersion
= '/v1'
34 self
._apiResource
= '/pdu_descriptors'
35 self
._apiBase
= '{}{}{}'.format(self
._apiName
,
36 self
._apiVersion
, self
._apiResource
)
38 def list(self
, filter=None):
39 self
._client
.get_token()
42 filter_string
= '?{}'.format(filter)
43 resp
= self
._http
.get_cmd('{}{}'.format(self
._apiBase
,filter_string
))
49 self
._client
.get_token()
50 if utils
.validate_uuid4(name
):
51 for pdud
in self
.list():
52 if name
== pdud
['_id']:
55 for pdud
in self
.list():
56 if 'name' in pdud
and name
== pdud
['name']:
58 raise NotFound("pdud {} not found".format(name
))
60 def get_individual(self
, name
):
62 # It is redundant, since the previous one already gets the whole pdudInfo
63 # The only difference is that a different primitive is exercised
64 resp
= self
._http
.get_cmd('{}/{}'.format(self
._apiBase
, pdud
['_id']))
65 #print(yaml.safe_dump(resp))
68 raise NotFound("pdu {} not found".format(name
))
70 def delete(self
, name
, force
=False):
74 querystring
= '?FORCE=True'
75 http_code
, resp
= self
._http
.delete_cmd('{}/{}{}'.format(self
._apiBase
,
76 pdud
['_id'], querystring
))
77 #print('HTTP CODE: {}'.format(http_code))
78 #print('RESP: {}'.format(resp))
80 print('Deletion in progress')
81 elif http_code
== 204:
87 msg
= json
.loads(resp
)
90 raise ClientException("failed to delete pdu {} - {}".format(name
, msg
))
92 def create(self
, pdu
, update_endpoint
=None):
93 self
._client
.get_token()
94 headers
= self
._client
._headers
95 headers
['Content-Type'] = 'application/yaml'
96 http_header
= ['{}: {}'.format(key
,val
)
97 for (key
,val
) in list(headers
.items())]
98 self
._http
.set_http_header(http_header
)
100 http_code
, resp
= self
._http
.put_cmd(endpoint
=update_endpoint
, postfields_dict
=pdu
)
102 endpoint
= self
._apiBase
103 #endpoint = '{}{}'.format(self._apiBase,ow_string)
104 http_code
, resp
= self
._http
.post_cmd(endpoint
=endpoint
, postfields_dict
=pdu
)
105 #print('HTTP CODE: {}'.format(http_code))
106 #print('RESP: {}'.format(resp))
107 if http_code
in (200, 201, 202, 204):
109 resp
= json
.loads(resp
)
110 if not resp
or 'id' not in resp
:
111 raise ClientException('unexpected response from server: '.format(
115 msg
= "Error {}".format(http_code
)
118 msg
= "{} - {}".format(msg
, json
.loads(resp
))
120 msg
= "{} - {}".format(msg
, resp
)
121 raise ClientException("failed to create/update pdu - {}".format(msg
))
123 def update(self
, name
, filename
):
124 pdud
= self
.get(name
)
125 endpoint
= '{}/{}'.format(self
._apiBase
, pdud
['_id'])
126 self
.create(filename
=filename
, update_endpoint
=endpoint
)