Merge "Add "consistency group create" command"

1 files changed, 77 insertions, 0 deletions

diff --git a/openstackclient/volume/v2/consistency_group.py b/openstackclient/volume/v2/consistency_group.py
index 0754fdc7..9316f287 100644
--- a/openstackclient/volume/v2/consistency_group.py
+++ b/openstackclient/volume/v2/consistency_group.py
@@ -14,12 +14,89 @@
 
 """Volume v2 consistency group action implementations"""
 
+import logging
+
 from osc_lib.command import command
 from osc_lib import utils
+import six
 
 from openstackclient.i18n import _
 
 
+LOG = logging.getLogger(__name__)
+
+
+class CreateConsistencyGroup(command.ShowOne):
+    _description = _("Create new consistency group.")
+
+    def get_parser(self, prog_name):
+        parser = super(CreateConsistencyGroup, self).get_parser(prog_name)
+        parser.add_argument(
+            "name",
+            metavar="<name>",
+            nargs="?",
+            help=_("Name of new consistency group (default to None)")
+        )
+        exclusive_group = parser.add_mutually_exclusive_group(required=True)
+        exclusive_group.add_argument(
+            "--volume-type",
+            metavar="<volume-type>",
+            help=_("Volume type of this consistency group (name or ID)")
+        )
+        exclusive_group.add_argument(
+            "--consistency-group-source",
+            metavar="<consistency-group>",
+            help=_("Existing consistency group (name or ID)")
+        )
+        parser.add_argument(
+            "--description",
+            metavar="<description>",
+            help=_("Description of this consistency group")
+        )
+        parser.add_argument(
+            "--availability-zone",
+            metavar="<availability-zone>",
+            help=_("Availability zone for this consistency group "
+                   "(not available if creating consistency group "
+                   "from source)"),
+        )
+        return parser
+
+    def take_action(self, parsed_args):
+        volume_client = self.app.client_manager.volume
+        if parsed_args.volume_type:
+            volume_type_id = utils.find_resource(
+                volume_client.volume_types,
+                parsed_args.volume_type).id
+            consistency_group = volume_client.consistencygroups.create(
+                volume_type_id,
+                name=parsed_args.name,
+                description=parsed_args.description,
+                availability_zone=parsed_args.availability_zone
+            )
+        elif parsed_args.consistency_group_source:
+            if parsed_args.availability_zone:
+                msg = _("'--availability-zone' option will not work "
+                        "if creating consistency group from source")
+                LOG.warning(msg)
+            consistency_group_id = utils.find_resource(
+                volume_client.consistencygroups,
+                parsed_args.consistency_group_source).id
+            consistency_group_snapshot = None
+            # TODO(Huanxuan Ao): Support for creating from consistency group
+            # snapshot after adding "consistency_group_snapshot" resource
+            consistency_group = (
+                volume_client.consistencygroups.create_from_src(
+                    consistency_group_snapshot,
+                    consistency_group_id,
+                    name=parsed_args.name,
+                    description=parsed_args.description
+                )
+            )
+
+        return zip(*sorted(six.iteritems(consistency_group._info)))
+
+
 class ListConsistencyGroup(command.Lister):
     _description = _("List consistency groups.")