mirror of https://github.com/chriskiehl/Gooey.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
3.4 KiB
95 lines
3.4 KiB
from argparse import ArgumentParser, _SubParsersAction, _MutuallyExclusiveGroup
|
|
|
|
|
|
class GooeySubParser(_SubParsersAction):
|
|
def __init__(self, *args, **kwargs):
|
|
super(GooeySubParser, self).__init__(*args, **kwargs)
|
|
|
|
class GooeyMutuallyExclusiveGroup(_MutuallyExclusiveGroup):
|
|
def __init__(self, parser, widgets, *args, **kwargs):
|
|
self.parser = parser
|
|
self.widgets = widgets
|
|
super(GooeyMutuallyExclusiveGroup, self).__init__(self.parser, *args, **kwargs)
|
|
|
|
def add_argument(self, *args, **kwargs):
|
|
widget = kwargs.pop('widget', None)
|
|
metavar = kwargs.pop('metavar', None)
|
|
super(GooeyMutuallyExclusiveGroup, self).add_argument(*args, **kwargs)
|
|
self.parser._actions[-1].metavar = metavar
|
|
self.widgets[self.parser._actions[-1].dest] = widget
|
|
|
|
|
|
class GooeyParser(object):
|
|
def __init__(self, **kwargs):
|
|
self.__dict__['parser'] = ArgumentParser(**kwargs)
|
|
self.widgets = {}
|
|
|
|
@property
|
|
def _mutually_exclusive_groups(self):
|
|
return self.parser._mutually_exclusive_groups
|
|
|
|
@property
|
|
def _actions(self):
|
|
return self.parser._actions
|
|
|
|
@property
|
|
def description(self):
|
|
return self.parser.description
|
|
|
|
def add_argument(self, *args, **kwargs):
|
|
widget = kwargs.pop('widget', None)
|
|
metavar = kwargs.pop('metavar', None)
|
|
self.parser.add_argument(*args, **kwargs)
|
|
self.parser._actions[-1].metavar = metavar
|
|
self.widgets[self.parser._actions[-1].dest] = widget
|
|
|
|
# def add_mutually_exclusive_group(self, **kwargs):
|
|
# return self.parser.add_mutually_exclusive_group(**kwargs)
|
|
|
|
def add_mutually_exclusive_group(self, **kwargs):
|
|
group = GooeyMutuallyExclusiveGroup(self.parser, self.widgets, **kwargs)
|
|
self.parser._mutually_exclusive_groups.append(group)
|
|
return group
|
|
|
|
def add_argument_group(self, *args, **kwargs):
|
|
return self.parser.add_argument_group(*args, **kwargs)
|
|
|
|
def parse_args(self, args=None, namespace=None):
|
|
return self.parser.parse_args(args, namespace)
|
|
|
|
def add_subparsers(self, **kwargs):
|
|
if self._subparsers is not None:
|
|
self.error(_('cannot have multiple subparser arguments'))
|
|
|
|
# add the parser class to the arguments if it's not present
|
|
kwargs.setdefault('parser_class', type(self))
|
|
|
|
if 'title' in kwargs or 'description' in kwargs:
|
|
title = _(kwargs.pop('title', 'subcommands'))
|
|
description = _(kwargs.pop('description', None))
|
|
self._subparsers = self.add_argument_group(title, description)
|
|
else:
|
|
self._subparsers = self._positionals
|
|
|
|
# prog defaults to the usage message of this parser, skipping
|
|
# optional arguments and with no "usage:" prefix
|
|
if kwargs.get('prog') is None:
|
|
formatter = self._get_formatter()
|
|
positionals = self._get_positional_actions()
|
|
groups = self._mutually_exclusive_groups
|
|
formatter.add_usage(self.usage, positionals, groups, '')
|
|
kwargs['prog'] = formatter.format_help().strip()
|
|
|
|
# create the parsers action and add it to the positionals list
|
|
parsers_class = self._pop_action_class(kwargs, 'parsers')
|
|
action = parsers_class(option_strings=[], **kwargs)
|
|
self._subparsers._add_action(action)
|
|
|
|
# return the created parsers action
|
|
return action
|
|
|
|
def __getattr__(self, item):
|
|
return getattr(self.parser, item)
|
|
|
|
def __setattr__(self, key, value):
|
|
return setattr(self.parser, key, value)
|