1 files changed, 136 insertions, 0 deletions
diff --git a/troveclient/management.py b/troveclient/management.py
new file mode 100644
index 0000000..1e17f92
--- /dev/null
+++ b/troveclient/management.py
@@ -0,0 +1,136 @@
+# Copyright (c) 2011 OpenStack, LLC.
+# 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.
+
+from troveclient import base
+import urlparse
+
+from troveclient.common import check_for_exceptions
+from troveclient.common import limit_url
+from troveclient.common import Paginated
+from troveclient.instances import Instance
+
+
+class RootHistory(base.Resource):
+    def __repr__(self):
+        return ("<Root History: Instance %s enabled at %s by %s>"
+                % (self.id, self.created, self.user))
+
+
+class Management(base.ManagerWithFind):
+    """
+    Manage :class:`Instances` resources.
+    """
+    resource_class = Instance
+
+    def _list(self, url, response_key, limit=None, marker=None):
+        resp, body = self.api.client.get(limit_url(url, limit, marker))
+        if not body:
+            raise Exception("Call to " + url + " did not return a body.")
+        links = body.get('links', [])
+        next_links = [link['href'] for link in links if link['rel'] == 'next']
+        next_marker = None
+        for link in next_links:
+            # Extract the marker from the url.
+            parsed_url = urlparse.urlparse(link)
+            query_dict = dict(urlparse.parse_qsl(parsed_url.query))
+            next_marker = query_dict.get('marker', None)
+        instances = body[response_key]
+        instances = [self.resource_class(self, res) for res in instances]
+        return Paginated(instances, next_marker=next_marker, links=links)
+
+    def show(self, instance):
+        """
+        Get details of one instance.
+
+        :rtype: :class:`Instance`.
+        """
+
+        return self._get("/mgmt/instances/%s" % base.getid(instance),
+                         'instance')
+
+    def index(self, deleted=None, limit=None, marker=None):
+        """
+        Show an overview of all local instances.
+        Optionally, filter by deleted status.
+
+        :rtype: list of :class:`Instance`.
+        """
+        form = ''
+        if deleted is not None:
+            if deleted:
+                form = "?deleted=true"
+            else:
+                form = "?deleted=false"
+
+        url = "/mgmt/instances%s" % form
+        return self._list(url, "instances", limit, marker)
+
+    def root_enabled_history(self, instance):
+        """
+        Get root access history of one instance.
+
+        """
+        url = "/mgmt/instances/%s/root" % base.getid(instance)
+        resp, body = self.api.client.get(url)
+        if not body:
+            raise Exception("Call to " + url + " did not return a body.")
+        return RootHistory(self, body['root_history'])
+
+    def _action(self, instance_id, body):
+        """
+        Perform a server "action" -- reboot/rebuild/resize/etc.
+        """
+        url = "/mgmt/instances/%s/action" % instance_id
+        resp, body = self.api.client.post(url, body=body)
+        check_for_exceptions(resp, body)
+
+    def stop(self, instance_id):
+        body = {'stop': {}}
+        self._action(instance_id, body)
+
+    def reboot(self, instance_id):
+        """
+        Reboot the underlying OS.
+
+        :param instance_id: The :class:`Instance` (or its ID) to share onto.
+        """
+        body = {'reboot': {}}
+        self._action(instance_id, body)
+
+    def migrate(self, instance_id, host=None):
+        """
+        Migrate the instance.
+
+        :param instance_id: The :class:`Instance` (or its ID) to share onto.
+        """
+        if host:
+            body = {'migrate': {'host': host}}
+        else:
+            body = {'migrate': {}}
+        self._action(instance_id, body)
+
+    def update(self, instance_id):
+        """
+        Update the guest agent via apt-get.
+        """
+        body = {'update': {}}
+        self._action(instance_id, body)
+
+    def reset_task_status(self, instance_id):
+        """
+        Set the task status to NONE.
+        """
+        body = {'reset-task-status': {}}
+        self._action(instance_id, body)