From f37eda3a27dc88d3186d21eca328cca086ee3647 Mon Sep 17 00:00:00 2001 From: Tang Chen Date: Wed, 17 Feb 2016 10:07:50 +0800 Subject: Make SetFlavor and UnsetFlavor inherit from cliff.Command set/unset comamnd classes should inherit from cliff.Command class. Change-Id: I54e5608ac0768d7d94b7f7d516ea1948daefdc1b Partial-Bug: 1546065 --- openstackclient/compute/v2/flavor.py | 24 ++++++------------------ 1 file changed, 6 insertions(+), 18 deletions(-) (limited to 'openstackclient/compute/v2') diff --git a/openstackclient/compute/v2/flavor.py b/openstackclient/compute/v2/flavor.py index 0308d940..e106bd65 100644 --- a/openstackclient/compute/v2/flavor.py +++ b/openstackclient/compute/v2/flavor.py @@ -242,7 +242,7 @@ class ShowFlavor(command.ShowOne): return zip(*sorted(six.iteritems(flavor))) -class SetFlavor(command.ShowOne): +class SetFlavor(command.Command): """Set flavor properties""" def get_parser(self, prog_name): @@ -263,17 +263,11 @@ class SetFlavor(command.ShowOne): def take_action(self, parsed_args): compute_client = self.app.client_manager.compute - resource_flavor = compute_client.flavors.find(name=parsed_args.flavor) + flavor = compute_client.flavors.find(name=parsed_args.flavor) + flavor.set_keys(parsed_args.property) - resource_flavor.set_keys(parsed_args.property) - flavor = resource_flavor._info.copy() - flavor['properties'] = utils.format_dict(resource_flavor.get_keys()) - flavor.pop("links", None) - return zip(*sorted(six.iteritems(flavor))) - - -class UnsetFlavor(command.ShowOne): +class UnsetFlavor(command.Command): """Unset flavor properties""" def get_parser(self, prog_name): @@ -295,11 +289,5 @@ class UnsetFlavor(command.ShowOne): def take_action(self, parsed_args): compute_client = self.app.client_manager.compute - resource_flavor = compute_client.flavors.find(name=parsed_args.flavor) - - resource_flavor.unset_keys(parsed_args.property) - - flavor = resource_flavor._info.copy() - flavor['properties'] = utils.format_dict(resource_flavor.get_keys()) - flavor.pop("links", None) - return zip(*sorted(six.iteritems(flavor))) + flavor = compute_client.flavors.find(name=parsed_args.flavor) + flavor.unset_keys(parsed_args.property) -- cgit v1.2.1