summaryrefslogtreecommitdiffstats
path: root/client/escalatorclient/v1/update.py
diff options
context:
space:
mode:
Diffstat (limited to 'client/escalatorclient/v1/update.py')
-rw-r--r--client/escalatorclient/v1/update.py105
1 files changed, 105 insertions, 0 deletions
diff --git a/client/escalatorclient/v1/update.py b/client/escalatorclient/v1/update.py
new file mode 100644
index 0000000..6989a0a
--- /dev/null
+++ b/client/escalatorclient/v1/update.py
@@ -0,0 +1,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))