Merge "Add qos rule type filtering"

1 files changed, 24 insertions, 1 deletions

diff --git a/openstackclient/network/v2/network_qos_rule_type.py b/openstackclient/network/v2/network_qos_rule_type.py
index 9af22876..3f4f6a19 100644
--- a/openstackclient/network/v2/network_qos_rule_type.py
+++ b/openstackclient/network/v2/network_qos_rule_type.py
@@ -32,6 +32,23 @@ def _get_columns(item):
 class ListNetworkQosRuleType(command.Lister):
     _description = _("List QoS rule types")
 
+    def get_parser(self, prog_name):
+        parser = super().get_parser(prog_name)
+        supported = parser.add_mutually_exclusive_group()
+        supported.add_argument(
+            '--all-supported',
+            action='store_true',
+            help=_("List all the QoS rule types supported by any loaded "
+                   "mechanism drivers (the union of all sets of supported "
+                   "rules)")
+        )
+        supported.add_argument(
+            '--all-rules',
+            action='store_true',
+            help=_("List all QoS rule types implemented in Neutron QoS driver")
+        )
+        return parser
+
     def take_action(self, parsed_args):
         client = self.app.client_manager.network
         columns = (
@@ -40,7 +57,13 @@ class ListNetworkQosRuleType(command.Lister):
         column_headers = (
             'Type',
         )
-        data = client.qos_rule_types()
+
+        args = {}
+        if parsed_args.all_supported:
+            args['all_supported'] = True
+        elif parsed_args.all_rules:
+            args['all_rules'] = True
+        data = client.qos_rule_types(**args)
 
         return (column_headers,
                 (utils.get_item_properties(