# Copyright 2013 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2

__all__ = ['ArgumentParser']

try:
	from argparse import ArgumentParser
except ImportError:
	# Compatibility with Python 2.6 and 3.1
	from optparse import OptionGroup, OptionParser

	from portage.localization import _

	class ArgumentParser(object):
		def __init__(self, **kwargs):
			add_help = kwargs.pop("add_help", None)
			if add_help is not None:
				kwargs["add_help_option"] = add_help
			parser = OptionParser(**kwargs)
			self._parser = parser
			self.add_argument = parser.add_option
			self.print_help = parser.print_help
			self.error = parser.error

		def add_argument_group(self, title=None, **kwargs):
			optiongroup = OptionGroup(self._parser, title, **kwargs)
			self._parser.add_option_group(optiongroup)
			return _ArgumentGroup(optiongroup)

		def parse_known_args(self, args=None, namespace=None):
			return self._parser.parse_args(args, namespace)

		def parse_args(self, args=None, namespace=None):
			args, argv = self.parse_known_args(args, namespace)
			if argv:
				msg = _('unrecognized arguments: %s')
				self.error(msg % ' '.join(argv))
			return args

	class _ArgumentGroup(object):
		def __init__(self, optiongroup):
			self.add_argument = optiongroup.add_option