1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
# Copyright 2012 OpenStack Foundation
# All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import copy
import six
from escalatorclient.common import utils
from escalatorclient.openstack.common.apiclient import base
CREATE_PARAMS = (
'cluster_id', 'version_id', 'version_patch_id', 'hosts',
'update_object', 'update_script')
OS_REQ_ID_HDR = 'x-openstack-request-id'
class Update(base.Resource):
def __repr__(self):
return "<Update %s>" % self._info
def update(self, **fields):
self.manager.update(self, **fields)
def data(self, **kwargs):
return self.manager.data(self, **kwargs)
class UpdateManager(base.ManagerWithFind):
resource_class = Update
def _Update_meta_to_headers(self, fields):
headers = {}
fields_copy = copy.deepcopy(fields)
# NOTE(flaper87): Convert to str, headers
# that are not instance of basestring. All
# headers will be encoded later, before the
# request is sent.
for key, value in six.iteritems(fields_copy):
headers['%s' % key] = utils.to_str(value)
return headers
@staticmethod
def _format_update_meta_for_user(meta):
for key in ['size', 'min_ram', 'min_disk']:
if key in meta:
try:
meta[key] = int(meta[key]) if meta[key] else 0
except ValueError:
pass
return meta
def list(self, **kwargs):
pass
def query_progress(self, **kwargs):
fields = {}
for field in kwargs:
if field in CREATE_PARAMS:
fields[field] = kwargs[field]
else:
msg = 'update() got an unexpected argument \'%s\''
raise TypeError(msg % field)
if "cluster" in fields:
url = '/v1/update/%s' % fields['cluster_id']
resp, body = self.client.get(url)
return Update(self, self._format_update_meta_for_user(body))
def update(self, **kwargs):
"""Update a cluster
TODO(bcwaldon): document accepted params
"""
fields = {}
for field in kwargs:
if field in CREATE_PARAMS:
fields[field] = kwargs[field]
elif field == 'return_req_id':
continue
else:
msg = 'update() got an unexpected argument \'%s\''
raise TypeError(msg % field)
if "cluster_id" in fields:
url = '/v1/update/%s' % fields['cluster_id']
hdrs = self._Update_meta_to_headers(fields)
resp, body = self.client.post(url, headers=None, data=hdrs)
return Update(self, self._format_update_meta_for_user(body))
|
