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))