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
._logger
= logging
.getLogger("osmclient")
33 self
._apiName
= "/pdu"
34 self
._apiVersion
= "/v1"
35 self
._apiResource
= "/pdu_descriptors"
36 self
._apiBase
= "{}{}{}".format(
37 self
._apiName
, self
._apiVersion
, self
._apiResource
40 def _get_vim_account(self
, vim_account
):
41 vim
= self
._client
.vim
.get(vim_account
)
43 raise NotFound("cannot find vim account '{}'".format(vim_account
))
46 def list(self
, filter=None):
47 self
._logger
.debug("")
48 self
._client
.get_token()
51 filter_string
= "?{}".format(filter)
52 _
, resp
= self
._http
.get2_cmd("{}{}".format(self
._apiBase
, filter_string
))
54 return json
.loads(resp
)
58 self
._logger
.debug("")
59 self
._client
.get_token()
60 if utils
.validate_uuid4(name
):
61 for pdud
in self
.list():
62 if name
== pdud
["_id"]:
65 for pdud
in self
.list():
66 if "name" in pdud
and name
== pdud
["name"]:
68 raise NotFound("pdud {} not found".format(name
))
70 def get_individual(self
, name
):
71 self
._logger
.debug("")
73 # It is redundant, since the previous one already gets the whole pdudInfo
74 # The only difference is that a different primitive is exercised
76 _
, resp
= self
._http
.get2_cmd("{}/{}".format(self
._apiBase
, pdud
["_id"]))
78 raise NotFound("pdu '{}' not found".format(name
))
79 # print(yaml.safe_dump(resp))
81 return json
.loads(resp
)
82 raise NotFound("pdu '{}' not found".format(name
))
84 def delete(self
, name
, force
=False):
85 self
._logger
.debug("")
89 querystring
= "?FORCE=True"
90 http_code
, resp
= self
._http
.delete_cmd(
91 "{}/{}{}".format(self
._apiBase
, pdud
["_id"], querystring
)
93 # print('HTTP CODE: {}'.format(http_code))
94 # print('RESP: {}'.format(resp))
96 print("Deletion in progress")
97 elif http_code
== 204:
103 # msg = json.loads(resp)
106 raise ClientException("failed to delete pdu {} - {}".format(name
, msg
))
108 def create(self
, pdu
, update_endpoint
=None):
109 self
._logger
.debug("")
111 if pdu
["vim_accounts"]:
112 vim_account_list
= []
113 for vim_account
in pdu
["vim_accounts"]:
114 vim
= self
._get
_vim
_account
(vim_account
)
115 vim_account_list
.append(vim
["_id"])
116 pdu
["vim_accounts"] = vim_account_list
118 self
._client
.get_token()
119 headers
= self
._client
._headers
120 headers
["Content-Type"] = "application/yaml"
122 "{}: {}".format(key
, val
) for (key
, val
) in list(headers
.items())
124 self
._http
.set_http_header(http_header
)
126 http_code
, resp
= self
._http
.patch_cmd(
127 endpoint
=update_endpoint
, postfields_dict
=pdu
130 endpoint
= self
._apiBase
131 # endpoint = '{}{}'.format(self._apiBase,ow_string)
132 http_code
, resp
= self
._http
.post_cmd(
133 endpoint
=endpoint
, postfields_dict
=pdu
135 if http_code
in (200, 201, 202):
137 resp
= json
.loads(resp
)
138 if not resp
or "id" not in resp
:
139 raise ClientException(
140 "unexpected response from server: {}".format(resp
)
143 elif http_code
== 204:
146 def update(self
, name
, pdu
):
147 self
._logger
.debug("")
148 pdud
= self
.get(name
)
149 endpoint
= "{}/{}".format(self
._apiBase
, pdud
["_id"])
150 self
.create(pdu
=pdu
, update_endpoint
=endpoint
)