Gooey/gooey/python_bindings/gooey_parser.py

from argparse import ArgumentParser, _SubParsersAction, _MutuallyExclusiveGroup

from gooey.gui.lang.i18n import _


class GooeySubParser(_SubParsersAction):
    def __init__(self, *args, **kwargs):
        super(GooeySubParser, self).__init__(*args, **kwargs)


class GooeyMutuallyExclusiveGroup(_MutuallyExclusiveGroup):
    def __init__(self, parser, metadata, *args, **kwargs):
        self.parser = parser
        self.metadata = metadata
        title = kwargs.pop('title', None) # todo inc group num?
        super(GooeyMutuallyExclusiveGroup, self).__init__(self.parser, *args, **kwargs)
        self.title = title

    def add_argument(self, *args, **kwargs):
        widget = kwargs.pop('widget', None)
        validator = kwargs.pop('validate', None)
        metavar = kwargs.pop('metavar', None)
        super(GooeyMutuallyExclusiveGroup, self).add_argument(*args, **kwargs)
        self.parser._actions[-1].metavar = metavar
        self.metadata['widgets'][self.parser._actions[-1].dest] = widget
        self.metadata['validators'][self.parser._actions[-1].dest] = validator


class GooeyParser(object):
    def __init__(self, **kwargs):
        self.__dict__['parser'] = ArgumentParser(**kwargs)
        # self.widgets = {}
        # self.validators = {}
        self.metadata = {
            'widgets': {},
            'validators': {}
        }

    @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)
        validator = kwargs.pop('validate', None)
        metavar = kwargs.pop('metavar', None)
        self.parser.add_argument(*args, **kwargs)
        self.parser._actions[-1].metavar = metavar
        self.metadata['widgets'][self.parser._actions[-1].dest] = widget
        self.metadata['validators'][self.parser._actions[-1].dest] = validator


    def add_mutually_exclusive_group(self, **kwargs):
        group = GooeyMutuallyExclusiveGroup(self.parser, self.metadata, **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)