diff options
Diffstat (limited to 'dnsviz/commands/probe.py')
-rw-r--r-- | dnsviz/commands/probe.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/dnsviz/commands/probe.py b/dnsviz/commands/probe.py index f337870..bb01c0a 100644 --- a/dnsviz/commands/probe.py +++ b/dnsviz/commands/probe.py @@ -910,7 +910,7 @@ class ArgHelper: self._logger = logger self._zones_to_serve = [] - def build_parser(self, prog, args): + def build_parser(self, prog): self.parser = argparse.ArgumentParser(description='Issue diagnostic DNS queries', prog=prog) helper = DomainListArgHelper(self._resolver) @@ -1027,6 +1027,8 @@ class ArgHelper: help='Domain names') self._arg_mapping = dict([(a.dest, '/'.join(a.option_strings)) for a in self.parser._actions]) + + def parse_args(self, args): self.args = self.parser.parse_args(args) @classmethod @@ -1459,7 +1461,8 @@ def main(argv): sys.exit(1) arghelper = ArgHelper(bootstrap_resolver, logger) - arghelper.build_parser('%s %s' % (sys.argv[0], argv[0]), argv[1:]) + arghelper.build_parser('%s %s' % (sys.argv[0], argv[0])) + arghelper.parse_args(argv[1:]) logger.setLevel(arghelper.get_log_level()) try: |