diff options
Diffstat (limited to 'Lib/distutils/fancy_getopt.py')
| -rw-r--r-- | Lib/distutils/fancy_getopt.py | 48 |
1 files changed, 36 insertions, 12 deletions
diff --git a/Lib/distutils/fancy_getopt.py b/Lib/distutils/fancy_getopt.py index cfebab4..629da29 100644 --- a/Lib/distutils/fancy_getopt.py +++ b/Lib/distutils/fancy_getopt.py @@ -66,6 +66,10 @@ class FancyGetopt: if self.option_table: self._build_index() + # 'alias' records (duh) alias options; {'foo': 'bar'} means + # --foo is an alias for --bar + self.alias = {} + # 'negative_alias' keeps track of options that are the boolean # opposite of some other option self.negative_alias = {} @@ -118,23 +122,29 @@ class FancyGetopt: return string.translate (long_option, longopt_xlate) + def _check_alias_dict (self, aliases, what): + assert type(aliases) is DictionaryType + for (alias, opt) in aliases.items(): + if not self.option_index.has_key(alias): + raise DistutilsGetoptError, \ + ("invalid %s '%s': " + "option '%s' not defined") % (what, alias, alias) + if not self.option_index.has_key(opt): + raise DistutilsGetoptError, \ + ("invalid %s '%s': " + "aliased option '%s' not defined") % (what, alias, opt) + + def set_aliases (self, alias): + """Set the aliases for this option parser.""" + self._check_alias_dict (alias, "alias") + self.alias = alias + def set_negative_aliases (self, negative_alias): """Set the negative aliases for this option parser. 'negative_alias' should be a dictionary mapping option names to option names, both the key and value must already be defined in the option table.""" - - assert type(negative_alias) is DictionaryType - for (negopt, opt) in negative_alias.items(): - if not self.option_index.has_key(negopt): - raise DistutilsGetoptError, \ - ("invalid negative alias '%s': " - "option '%s' not defined") % (negopt, negopt) - if not self.option_index.has_key(opt): - raise DistutilsGetoptError, \ - ("invalid negative alias '%s': " - "aliased option '%s' not defined") % (negopt, opt) - + self._check_alias_dict (negative_alias, "negative alias") self.negative_alias = negative_alias @@ -186,6 +196,16 @@ class FancyGetopt: else: self.takes_arg[long] = 0 + # If this is an alias option, make sure its "takes arg" flag is + # the same as the option it's aliased to. + alias_to = self.alias.get(long) + if alias_to is not None: + if self.takes_arg[long] != self.takes_arg[alias_to]: + raise DistutilsGetoptError, \ + ("invalid alias '%s': inconsistent with " + "aliased option '%s' (one of them takes a value, " + "the other doesn't") % (long, alias_to) + # Now enforce some bondage on the long option name, so we can # later translate it to an attribute name on some object. Have @@ -243,6 +263,10 @@ class FancyGetopt: raise DistutilsInternalError, \ "this can't happen: bad option string '%s'" % opt + alias = self.alias.get(opt) + if alias: + opt = alias + if not self.takes_arg[opt]: # boolean option? if val != '': # shouldn't have a value! raise DistutilsInternalError, \ |