Gooey/gooey/python_bindings/gooey_parser.py

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)