Forráskód Böngészése

Add subparser handling to parse_qubes_app

Bahtiar `kalkin-` Gadimov 8 éve
szülő
commit
2f0df26606
1 módosított fájl, 7 hozzáadás és 0 törlés
  1. 7 0
      qubes/tools/__init__.py

+ 7 - 0
qubes/tools/__init__.py

@@ -360,6 +360,13 @@ class QubesArgumentParser(argparse.ArgumentParser):
         for action in self._actions:
             if issubclass(action.__class__, QubesAction):
                 action.parse_qubes_app(self, namespace)
+            elif issubclass(action.__class__, argparse._SubParsersAction):
+                assert hasattr(namespace, 'command')
+                command = namespace.command
+                subparser = action._name_parser_map[command]
+                for subaction in subparser._actions:
+                    if issubclass(subaction.__class__, QubesAction):
+                        subaction.parse_qubes_app(self, namespace)
 
         return namespace