From 7266c695b3bc592558a1170bd00e5904fa3dd303 Mon Sep 17 00:00:00 2001
From: Josh Kearney <josh@jk0.org>
Date: Tue, 5 Mar 2013 10:30:36 -0600
Subject: [PATCH] Added compute service support.

Change-Id: I548e35396534b5498f31a45c752984f6d33357c3
---
 openstackclient/compute/v2/service.py | 111 ++++++++++++++++++++++++++
 setup.py                              |   4 +
 2 files changed, 115 insertions(+)
 create mode 100644 openstackclient/compute/v2/service.py

diff --git a/openstackclient/compute/v2/service.py b/openstackclient/compute/v2/service.py
new file mode 100644
index 0000000000..3bd7481499
--- /dev/null
+++ b/openstackclient/compute/v2/service.py
@@ -0,0 +1,111 @@
+#   Copyright 2013 OpenStack, LLC.
+#
+#   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.
+#
+
+"""Service action implementations"""
+
+import logging
+
+from cliff import lister
+
+from novaclient.v1_1 import services
+from openstackclient.common import utils
+
+
+class ListService(lister.Lister):
+    """List service command"""
+
+    api = "compute"
+    log = logging.getLogger(__name__ + ".ListService")
+
+    def get_parser(self, prog_name):
+        parser = super(ListService, self).get_parser(prog_name)
+        parser.add_argument(
+            "--host",
+            metavar="<host>",
+            help="Name of host")
+        parser.add_argument(
+            "--service",
+            metavar="<service>",
+            help="Name of service")
+        return parser
+
+    def take_action(self, parsed_args):
+        self.log.debug("take_action(%s)" % parsed_args)
+        compute_client = self.app.client_manager.compute
+        columns = (
+            "Binary",
+            "Host",
+            "Zone",
+            "Status",
+            "State",
+            "Updated At"
+        )
+        data = compute_client.services.list(parsed_args.host,
+                                            parsed_args.service)
+        return (columns,
+                (utils.get_item_properties(
+                    s, columns,
+                ) for s in data))
+
+
+class SetService(lister.Lister):
+    """Set service command"""
+
+    api = "compute"
+    log = logging.getLogger(__name__ + ".SetService")
+
+    def get_parser(self, prog_name):
+        parser = super(SetService, self).get_parser(prog_name)
+        parser.add_argument(
+            "host",
+            metavar="<host>",
+            help="Name of host")
+        parser.add_argument(
+            "service",
+            metavar="<service>",
+            help="Name of service")
+        enabled_group = parser.add_mutually_exclusive_group()
+        enabled_group.add_argument(
+            "--enable",
+            dest="enabled",
+            default=True,
+            help="Enable a service",
+            action="store_true")
+        enabled_group.add_argument(
+            "--disable",
+            dest="enabled",
+            help="Disable a service",
+            action="store_false")
+        return parser
+
+    def take_action(self, parsed_args):
+        self.log.debug("take_action(%s)" % parsed_args)
+        compute_client = self.app.client_manager.compute
+        columns = (
+            "Host",
+            "Service",
+            "Disabled"
+        )
+
+        if parsed_args.enabled:
+            action = compute_client.services.enable
+        else:
+            action = compute_client.services.disable
+
+        data = action(parsed_args.host, parsed_args.service)
+        return (columns,
+                (utils.get_item_properties(
+                    s, columns,
+                ) for s in data))
diff --git a/setup.py b/setup.py
index 3ad427f5bc..bf4c8bb96c 100644
--- a/setup.py
+++ b/setup.py
@@ -133,12 +133,16 @@ setuptools.setup(
             'list_agent=openstackclient.compute.v2.agent:ListAgent',
             'list_flavor=openstackclient.compute.v2.flavor:ListFlavor',
             'list_server=openstackclient.compute.v2.server:ListServer',
+            'list_compute-service=' +
+            'openstackclient.compute.v2.service:ListService',
             'pause_server=openstackclient.compute.v2.server:PauseServer',
             'reboot_server=openstackclient.compute.v2.server:RebootServer',
             'rebuild_server=openstackclient.compute.v2.server:RebuildServer',
             'resume_server=openstackclient.compute.v2.server:ResumeServer',
             'set_agent=openstackclient.compute.v2.agent:SetAgent',
             'show_flavor=openstackclient.compute.v2.flavor:ShowFlavor',
+            'set_compute-service=' +
+            'openstackclient.compute.v2.service:SetService',
             'show_server=openstackclient.compute.v2.server:ShowServer',
             'suspend_server=openstackclient.compute.v2.server:SuspendServer',
             'unpause_server=openstackclient.compute.v2.server:UnpauseServer',