blob: c2077136f05771f340a483ec810922fe0504a8b7 [file] [log] [blame]
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00001"""A powerful, extensible, and easy-to-use option parser.
Guido van Rossumb9ba4582002-11-14 22:00:19 +00002
3By Greg Ward <gward@python.net>
4
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00005Originally distributed as Optik.
Greg Ward2492fcf2003-04-21 02:40:34 +00006
7For support, use the optik-users@lists.sourceforge.net mailing list
8(http://lists.sourceforge.net/lists/listinfo/optik-users).
Georg Brandlf3532df2009-04-27 16:41:41 +00009
10Simple usage example:
11
12 from optparse import OptionParser
13
14 parser = OptionParser()
15 parser.add_option("-f", "--file", dest="filename",
16 help="write report to FILE", metavar="FILE")
17 parser.add_option("-q", "--quiet",
18 action="store_false", dest="verbose", default=True,
19 help="don't print status messages to stdout")
20
21 (options, args) = parser.parse_args()
Guido van Rossumb9ba4582002-11-14 22:00:19 +000022"""
23
Thomas Wouters0e3f5912006-08-11 14:57:12 +000024__version__ = "1.5.3"
Greg Ward2492fcf2003-04-21 02:40:34 +000025
Greg Ward4656ed42003-05-08 01:38:52 +000026__all__ = ['Option',
Benjamin Petersond23f8222009-04-05 19:13:16 +000027 'make_option',
Greg Ward4656ed42003-05-08 01:38:52 +000028 'SUPPRESS_HELP',
29 'SUPPRESS_USAGE',
Greg Ward4656ed42003-05-08 01:38:52 +000030 'Values',
31 'OptionContainer',
32 'OptionGroup',
33 'OptionParser',
34 'HelpFormatter',
35 'IndentedHelpFormatter',
36 'TitledHelpFormatter',
37 'OptParseError',
38 'OptionError',
39 'OptionConflictError',
40 'OptionValueError',
41 'BadOptionError']
Greg Ward2492fcf2003-04-21 02:40:34 +000042
Guido van Rossumb9ba4582002-11-14 22:00:19 +000043__copyright__ = """
Thomas Wouters477c8d52006-05-27 19:21:47 +000044Copyright (c) 2001-2006 Gregory P. Ward. All rights reserved.
45Copyright (c) 2002-2006 Python Software Foundation. All rights reserved.
Guido van Rossumb9ba4582002-11-14 22:00:19 +000046
47Redistribution and use in source and binary forms, with or without
48modification, are permitted provided that the following conditions are
49met:
50
51 * Redistributions of source code must retain the above copyright
52 notice, this list of conditions and the following disclaimer.
53
54 * Redistributions in binary form must reproduce the above copyright
55 notice, this list of conditions and the following disclaimer in the
56 documentation and/or other materials provided with the distribution.
57
58 * Neither the name of the author nor the names of its
59 contributors may be used to endorse or promote products derived from
60 this software without specific prior written permission.
61
62THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
63IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
64TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
65PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR
66CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
67EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
68PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
69PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
70LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
71NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
72SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
73"""
74
75import sys, os
Guido van Rossumb9ba4582002-11-14 22:00:19 +000076import textwrap
Greg Wardeba20e62004-07-31 16:15:44 +000077
78def _repr(self):
79 return "<%s at 0x%x: %s>" % (self.__class__.__name__, id(self), self)
80
81
82# This file was generated from:
Thomas Wouters0e3f5912006-08-11 14:57:12 +000083# Id: option_parser.py 527 2006-07-23 15:21:30Z greg
84# Id: option.py 522 2006-06-11 16:22:03Z gward
85# Id: help.py 527 2006-07-23 15:21:30Z greg
Thomas Wouters477c8d52006-05-27 19:21:47 +000086# Id: errors.py 509 2006-04-20 00:58:24Z gward
87
88try:
Éric Araujo6a1454f2011-03-20 19:59:25 +010089 from gettext import gettext, ngettext
Thomas Wouters477c8d52006-05-27 19:21:47 +000090except ImportError:
91 def gettext(message):
92 return message
Éric Araujo6a1454f2011-03-20 19:59:25 +010093
94 def ngettext(singular, plural, n):
95 if n == 1:
96 return singular
97 return plural
98
Thomas Wouters477c8d52006-05-27 19:21:47 +000099_ = gettext
100
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000101
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000102class OptParseError (Exception):
Greg Wardeba20e62004-07-31 16:15:44 +0000103 def __init__(self, msg):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000104 self.msg = msg
105
Greg Wardeba20e62004-07-31 16:15:44 +0000106 def __str__(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000107 return self.msg
108
Greg Ward2492fcf2003-04-21 02:40:34 +0000109
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000110class OptionError (OptParseError):
111 """
112 Raised if an Option instance is created with invalid or
113 inconsistent arguments.
114 """
115
Greg Wardeba20e62004-07-31 16:15:44 +0000116 def __init__(self, msg, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000117 self.msg = msg
118 self.option_id = str(option)
119
Greg Wardeba20e62004-07-31 16:15:44 +0000120 def __str__(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000121 if self.option_id:
122 return "option %s: %s" % (self.option_id, self.msg)
123 else:
124 return self.msg
125
126class OptionConflictError (OptionError):
127 """
128 Raised if conflicting options are added to an OptionParser.
129 """
130
131class OptionValueError (OptParseError):
132 """
133 Raised if an invalid option value is encountered on the command
134 line.
135 """
136
137class BadOptionError (OptParseError):
138 """
Thomas Wouters477c8d52006-05-27 19:21:47 +0000139 Raised if an invalid option is seen on the command line.
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000140 """
Thomas Wouters477c8d52006-05-27 19:21:47 +0000141 def __init__(self, opt_str):
142 self.opt_str = opt_str
143
144 def __str__(self):
145 return _("no such option: %s") % self.opt_str
146
147class AmbiguousOptionError (BadOptionError):
148 """
149 Raised if an ambiguous option is seen on the command line.
150 """
151 def __init__(self, opt_str, possibilities):
152 BadOptionError.__init__(self, opt_str)
153 self.possibilities = possibilities
154
155 def __str__(self):
156 return (_("ambiguous option: %s (%s?)")
157 % (self.opt_str, ", ".join(self.possibilities)))
Greg Ward2492fcf2003-04-21 02:40:34 +0000158
159
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000160class HelpFormatter:
161
162 """
163 Abstract base class for formatting option help. OptionParser
164 instances should use one of the HelpFormatter subclasses for
165 formatting help; by default IndentedHelpFormatter is used.
166
167 Instance attributes:
Greg Wardeba20e62004-07-31 16:15:44 +0000168 parser : OptionParser
169 the controlling OptionParser instance
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000170 indent_increment : int
171 the number of columns to indent per nesting level
172 max_help_position : int
173 the maximum starting column for option help text
174 help_position : int
175 the calculated starting column for option help text;
176 initially the same as the maximum
177 width : int
Greg Wardeba20e62004-07-31 16:15:44 +0000178 total number of columns for output (pass None to constructor for
179 this value to be taken from the $COLUMNS environment variable)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000180 level : int
181 current indentation level
182 current_indent : int
183 current indentation level (in columns)
184 help_width : int
185 number of columns available for option help text (calculated)
Greg Wardeba20e62004-07-31 16:15:44 +0000186 default_tag : str
187 text to replace with each option's default value, "%default"
188 by default. Set to false value to disable default value expansion.
189 option_strings : { Option : str }
190 maps Option instances to the snippet of help text explaining
191 the syntax of that option, e.g. "-h, --help" or
192 "-fFILE, --file=FILE"
193 _short_opt_fmt : str
194 format string controlling how short options with values are
195 printed in help text. Must be either "%s%s" ("-fFILE") or
196 "%s %s" ("-f FILE"), because those are the two syntaxes that
197 Optik supports.
198 _long_opt_fmt : str
199 similar but for long options; must be either "%s %s" ("--file FILE")
200 or "%s=%s" ("--file=FILE").
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000201 """
202
Greg Wardeba20e62004-07-31 16:15:44 +0000203 NO_DEFAULT_VALUE = "none"
204
205 def __init__(self,
206 indent_increment,
207 max_help_position,
208 width,
209 short_first):
210 self.parser = None
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000211 self.indent_increment = indent_increment
212 self.help_position = self.max_help_position = max_help_position
Greg Wardeba20e62004-07-31 16:15:44 +0000213 if width is None:
214 try:
215 width = int(os.environ['COLUMNS'])
216 except (KeyError, ValueError):
217 width = 80
218 width -= 2
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000219 self.width = width
220 self.current_indent = 0
221 self.level = 0
Greg Wardeba20e62004-07-31 16:15:44 +0000222 self.help_width = None # computed later
Greg Ward2492fcf2003-04-21 02:40:34 +0000223 self.short_first = short_first
Greg Wardeba20e62004-07-31 16:15:44 +0000224 self.default_tag = "%default"
225 self.option_strings = {}
226 self._short_opt_fmt = "%s %s"
227 self._long_opt_fmt = "%s=%s"
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000228
Greg Wardeba20e62004-07-31 16:15:44 +0000229 def set_parser(self, parser):
230 self.parser = parser
231
232 def set_short_opt_delimiter(self, delim):
233 if delim not in ("", " "):
234 raise ValueError(
235 "invalid metavar delimiter for short options: %r" % delim)
236 self._short_opt_fmt = "%s" + delim + "%s"
237
238 def set_long_opt_delimiter(self, delim):
239 if delim not in ("=", " "):
240 raise ValueError(
241 "invalid metavar delimiter for long options: %r" % delim)
242 self._long_opt_fmt = "%s" + delim + "%s"
243
244 def indent(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000245 self.current_indent += self.indent_increment
246 self.level += 1
247
Greg Wardeba20e62004-07-31 16:15:44 +0000248 def dedent(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000249 self.current_indent -= self.indent_increment
250 assert self.current_indent >= 0, "Indent decreased below 0."
251 self.level -= 1
252
Greg Wardeba20e62004-07-31 16:15:44 +0000253 def format_usage(self, usage):
Collin Winterce36ad82007-08-30 01:19:48 +0000254 raise NotImplementedError("subclasses must implement")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000255
Greg Wardeba20e62004-07-31 16:15:44 +0000256 def format_heading(self, heading):
Collin Winterce36ad82007-08-30 01:19:48 +0000257 raise NotImplementedError("subclasses must implement")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000258
Thomas Wouters477c8d52006-05-27 19:21:47 +0000259 def _format_text(self, text):
260 """
261 Format a paragraph of free-form text for inclusion in the
262 help output at the current indentation level.
263 """
264 text_width = self.width - self.current_indent
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000265 indent = " "*self.current_indent
Thomas Wouters477c8d52006-05-27 19:21:47 +0000266 return textwrap.fill(text,
267 text_width,
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000268 initial_indent=indent,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000269 subsequent_indent=indent)
270
271 def format_description(self, description):
272 if description:
273 return self._format_text(description) + "\n"
274 else:
275 return ""
276
277 def format_epilog(self, epilog):
278 if epilog:
279 return "\n" + self._format_text(epilog) + "\n"
280 else:
281 return ""
282
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000283
Greg Wardeba20e62004-07-31 16:15:44 +0000284 def expand_default(self, option):
285 if self.parser is None or not self.default_tag:
286 return option.help
287
288 default_value = self.parser.defaults.get(option.dest)
289 if default_value is NO_DEFAULT or default_value is None:
290 default_value = self.NO_DEFAULT_VALUE
291
292 return option.help.replace(self.default_tag, str(default_value))
293
294 def format_option(self, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000295 # The help for each option consists of two parts:
296 # * the opt strings and metavars
297 # eg. ("-x", or "-fFILENAME, --file=FILENAME")
298 # * the user-supplied help string
299 # eg. ("turn on expert mode", "read data from FILENAME")
300 #
301 # If possible, we write both of these on the same line:
302 # -x turn on expert mode
303 #
304 # But if the opt string list is too long, we put the help
305 # string on a second line, indented to the same column it would
306 # start in if it fit on the first line.
307 # -fFILENAME, --file=FILENAME
308 # read data from FILENAME
309 result = []
Greg Wardeba20e62004-07-31 16:15:44 +0000310 opts = self.option_strings[option]
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000311 opt_width = self.help_position - self.current_indent - 2
312 if len(opts) > opt_width:
313 opts = "%*s%s\n" % (self.current_indent, "", opts)
314 indent_first = self.help_position
315 else: # start help on same line as opts
316 opts = "%*s%-*s " % (self.current_indent, "", opt_width, opts)
317 indent_first = 0
318 result.append(opts)
319 if option.help:
Greg Wardeba20e62004-07-31 16:15:44 +0000320 help_text = self.expand_default(option)
321 help_lines = textwrap.wrap(help_text, self.help_width)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000322 result.append("%*s%s\n" % (indent_first, "", help_lines[0]))
323 result.extend(["%*s%s\n" % (self.help_position, "", line)
324 for line in help_lines[1:]])
325 elif opts[-1] != "\n":
326 result.append("\n")
327 return "".join(result)
328
Greg Wardeba20e62004-07-31 16:15:44 +0000329 def store_option_strings(self, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000330 self.indent()
331 max_len = 0
332 for opt in parser.option_list:
333 strings = self.format_option_strings(opt)
Greg Wardeba20e62004-07-31 16:15:44 +0000334 self.option_strings[opt] = strings
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000335 max_len = max(max_len, len(strings) + self.current_indent)
336 self.indent()
337 for group in parser.option_groups:
338 for opt in group.option_list:
339 strings = self.format_option_strings(opt)
Greg Wardeba20e62004-07-31 16:15:44 +0000340 self.option_strings[opt] = strings
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000341 max_len = max(max_len, len(strings) + self.current_indent)
342 self.dedent()
343 self.dedent()
344 self.help_position = min(max_len + 2, self.max_help_position)
Greg Wardeba20e62004-07-31 16:15:44 +0000345 self.help_width = self.width - self.help_position
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000346
Greg Wardeba20e62004-07-31 16:15:44 +0000347 def format_option_strings(self, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000348 """Return a comma-separated list of option strings & metavariables."""
Greg Ward2492fcf2003-04-21 02:40:34 +0000349 if option.takes_value():
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000350 metavar = option.metavar or option.dest.upper()
Greg Wardeba20e62004-07-31 16:15:44 +0000351 short_opts = [self._short_opt_fmt % (sopt, metavar)
352 for sopt in option._short_opts]
353 long_opts = [self._long_opt_fmt % (lopt, metavar)
354 for lopt in option._long_opts]
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000355 else:
Greg Ward2492fcf2003-04-21 02:40:34 +0000356 short_opts = option._short_opts
357 long_opts = option._long_opts
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000358
Greg Ward2492fcf2003-04-21 02:40:34 +0000359 if self.short_first:
360 opts = short_opts + long_opts
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000361 else:
Greg Ward2492fcf2003-04-21 02:40:34 +0000362 opts = long_opts + short_opts
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000363
Greg Ward2492fcf2003-04-21 02:40:34 +0000364 return ", ".join(opts)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000365
366class IndentedHelpFormatter (HelpFormatter):
367 """Format help with indented section bodies.
368 """
369
Greg Wardeba20e62004-07-31 16:15:44 +0000370 def __init__(self,
371 indent_increment=2,
372 max_help_position=24,
373 width=None,
374 short_first=1):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000375 HelpFormatter.__init__(
376 self, indent_increment, max_help_position, width, short_first)
377
Greg Wardeba20e62004-07-31 16:15:44 +0000378 def format_usage(self, usage):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000379 return _("Usage: %s\n") % usage
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000380
Greg Wardeba20e62004-07-31 16:15:44 +0000381 def format_heading(self, heading):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000382 return "%*s%s:\n" % (self.current_indent, "", heading)
383
384
385class TitledHelpFormatter (HelpFormatter):
386 """Format help with underlined section headers.
387 """
388
Greg Wardeba20e62004-07-31 16:15:44 +0000389 def __init__(self,
390 indent_increment=0,
391 max_help_position=24,
392 width=None,
393 short_first=0):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000394 HelpFormatter.__init__ (
395 self, indent_increment, max_help_position, width, short_first)
396
Greg Wardeba20e62004-07-31 16:15:44 +0000397 def format_usage(self, usage):
398 return "%s %s\n" % (self.format_heading(_("Usage")), usage)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000399
Greg Wardeba20e62004-07-31 16:15:44 +0000400 def format_heading(self, heading):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000401 return "%s\n%s\n" % (heading, "=-"[self.level] * len(heading))
Greg Ward2492fcf2003-04-21 02:40:34 +0000402
403
Thomas Wouters477c8d52006-05-27 19:21:47 +0000404def _parse_num(val, type):
405 if val[:2].lower() == "0x": # hexadecimal
406 radix = 16
407 elif val[:2].lower() == "0b": # binary
408 radix = 2
409 val = val[2:] or "0" # have to remove "0b" prefix
410 elif val[:1] == "0": # octal
411 radix = 8
412 else: # decimal
413 radix = 10
414
415 return type(val, radix)
416
417def _parse_int(val):
418 return _parse_num(val, int)
419
420def _parse_long(val):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000421 return _parse_num(val, int)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000422
423_builtin_cvt = { "int" : (_parse_int, _("integer")),
424 "long" : (_parse_long, _("long integer")),
Greg Wardeba20e62004-07-31 16:15:44 +0000425 "float" : (float, _("floating-point")),
426 "complex" : (complex, _("complex")) }
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000427
Greg Wardeba20e62004-07-31 16:15:44 +0000428def check_builtin(option, opt, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000429 (cvt, what) = _builtin_cvt[option.type]
430 try:
431 return cvt(value)
432 except ValueError:
433 raise OptionValueError(
Greg Wardeba20e62004-07-31 16:15:44 +0000434 _("option %s: invalid %s value: %r") % (opt, what, value))
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000435
436def check_choice(option, opt, value):
437 if value in option.choices:
438 return value
439 else:
440 choices = ", ".join(map(repr, option.choices))
441 raise OptionValueError(
Greg Wardeba20e62004-07-31 16:15:44 +0000442 _("option %s: invalid choice: %r (choose from %s)")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000443 % (opt, value, choices))
444
445# Not supplying a default is different from a default of None,
446# so we need an explicit "not supplied" value.
Greg Wardeba20e62004-07-31 16:15:44 +0000447NO_DEFAULT = ("NO", "DEFAULT")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000448
449
450class Option:
451 """
452 Instance attributes:
453 _short_opts : [string]
454 _long_opts : [string]
455
456 action : string
457 type : string
458 dest : string
459 default : any
460 nargs : int
461 const : any
462 choices : [string]
463 callback : function
464 callback_args : (any*)
465 callback_kwargs : { string : any }
466 help : string
467 metavar : string
468 """
469
470 # The list of instance attributes that may be set through
471 # keyword args to the constructor.
472 ATTRS = ['action',
473 'type',
474 'dest',
475 'default',
476 'nargs',
477 'const',
478 'choices',
479 'callback',
480 'callback_args',
481 'callback_kwargs',
482 'help',
483 'metavar']
484
485 # The set of actions allowed by option parsers. Explicitly listed
486 # here so the constructor can validate its arguments.
487 ACTIONS = ("store",
488 "store_const",
489 "store_true",
490 "store_false",
491 "append",
Thomas Wouters477c8d52006-05-27 19:21:47 +0000492 "append_const",
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000493 "count",
494 "callback",
495 "help",
496 "version")
497
498 # The set of actions that involve storing a value somewhere;
499 # also listed just for constructor argument validation. (If
500 # the action is one of these, there must be a destination.)
501 STORE_ACTIONS = ("store",
502 "store_const",
503 "store_true",
504 "store_false",
505 "append",
Thomas Wouters477c8d52006-05-27 19:21:47 +0000506 "append_const",
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000507 "count")
508
509 # The set of actions for which it makes sense to supply a value
Greg Ward48aa84b2004-10-27 02:20:04 +0000510 # type, ie. which may consume an argument from the command line.
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000511 TYPED_ACTIONS = ("store",
512 "append",
513 "callback")
514
Greg Ward48aa84b2004-10-27 02:20:04 +0000515 # The set of actions which *require* a value type, ie. that
516 # always consume an argument from the command line.
517 ALWAYS_TYPED_ACTIONS = ("store",
518 "append")
519
Thomas Wouters477c8d52006-05-27 19:21:47 +0000520 # The set of actions which take a 'const' attribute.
521 CONST_ACTIONS = ("store_const",
522 "append_const")
523
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000524 # The set of known types for option parsers. Again, listed here for
525 # constructor argument validation.
526 TYPES = ("string", "int", "long", "float", "complex", "choice")
527
528 # Dictionary of argument checking functions, which convert and
529 # validate option arguments according to the option type.
530 #
531 # Signature of checking functions is:
532 # check(option : Option, opt : string, value : string) -> any
533 # where
534 # option is the Option instance calling the checker
535 # opt is the actual option seen on the command-line
536 # (eg. "-a", "--file")
537 # value is the option argument seen on the command-line
538 #
539 # The return value should be in the appropriate Python type
540 # for option.type -- eg. an integer if option.type == "int".
541 #
542 # If no checker is defined for a type, arguments will be
543 # unchecked and remain strings.
544 TYPE_CHECKER = { "int" : check_builtin,
545 "long" : check_builtin,
546 "float" : check_builtin,
Greg Wardeba20e62004-07-31 16:15:44 +0000547 "complex": check_builtin,
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000548 "choice" : check_choice,
549 }
550
551
552 # CHECK_METHODS is a list of unbound method objects; they are called
553 # by the constructor, in order, after all attributes are
554 # initialized. The list is created and filled in later, after all
555 # the methods are actually defined. (I just put it here because I
556 # like to define and document all class attributes in the same
557 # place.) Subclasses that add another _check_*() method should
558 # define their own CHECK_METHODS list that adds their check method
559 # to those from this class.
560 CHECK_METHODS = None
561
562
563 # -- Constructor/initialization methods ----------------------------
564
Greg Wardeba20e62004-07-31 16:15:44 +0000565 def __init__(self, *opts, **attrs):
Greg Ward2492fcf2003-04-21 02:40:34 +0000566 # Set _short_opts, _long_opts attrs from 'opts' tuple.
567 # Have to be set now, in case no option strings are supplied.
568 self._short_opts = []
569 self._long_opts = []
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000570 opts = self._check_opt_strings(opts)
571 self._set_opt_strings(opts)
572
573 # Set all other attrs (action, type, etc.) from 'attrs' dict
574 self._set_attrs(attrs)
575
576 # Check all the attributes we just set. There are lots of
577 # complicated interdependencies, but luckily they can be farmed
578 # out to the _check_*() methods listed in CHECK_METHODS -- which
579 # could be handy for subclasses! The one thing these all share
580 # is that they raise OptionError if they discover a problem.
581 for checker in self.CHECK_METHODS:
582 checker(self)
583
Greg Wardeba20e62004-07-31 16:15:44 +0000584 def _check_opt_strings(self, opts):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000585 # Filter out None because early versions of Optik had exactly
586 # one short option and one long option, either of which
587 # could be None.
Guido van Rossumc1f779c2007-07-03 08:25:58 +0000588 opts = [opt for opt in opts if opt]
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000589 if not opts:
Greg Ward2492fcf2003-04-21 02:40:34 +0000590 raise TypeError("at least one option string must be supplied")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000591 return opts
592
Greg Wardeba20e62004-07-31 16:15:44 +0000593 def _set_opt_strings(self, opts):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000594 for opt in opts:
595 if len(opt) < 2:
596 raise OptionError(
597 "invalid option string %r: "
598 "must be at least two characters long" % opt, self)
599 elif len(opt) == 2:
600 if not (opt[0] == "-" and opt[1] != "-"):
601 raise OptionError(
602 "invalid short option string %r: "
603 "must be of the form -x, (x any non-dash char)" % opt,
604 self)
605 self._short_opts.append(opt)
606 else:
607 if not (opt[0:2] == "--" and opt[2] != "-"):
608 raise OptionError(
609 "invalid long option string %r: "
610 "must start with --, followed by non-dash" % opt,
611 self)
612 self._long_opts.append(opt)
613
Greg Wardeba20e62004-07-31 16:15:44 +0000614 def _set_attrs(self, attrs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000615 for attr in self.ATTRS:
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000616 if attr in attrs:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000617 setattr(self, attr, attrs[attr])
618 del attrs[attr]
619 else:
620 if attr == 'default':
621 setattr(self, attr, NO_DEFAULT)
622 else:
623 setattr(self, attr, None)
624 if attrs:
Georg Brandlc2d9d7f2007-02-11 23:06:17 +0000625 attrs = sorted(attrs.keys())
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000626 raise OptionError(
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000627 "invalid keyword arguments: %s" % ", ".join(attrs),
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000628 self)
629
630
631 # -- Constructor validation methods --------------------------------
632
Greg Wardeba20e62004-07-31 16:15:44 +0000633 def _check_action(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000634 if self.action is None:
635 self.action = "store"
636 elif self.action not in self.ACTIONS:
637 raise OptionError("invalid action: %r" % self.action, self)
638
Greg Wardeba20e62004-07-31 16:15:44 +0000639 def _check_type(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000640 if self.type is None:
Greg Ward48aa84b2004-10-27 02:20:04 +0000641 if self.action in self.ALWAYS_TYPED_ACTIONS:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000642 if self.choices is not None:
643 # The "choices" attribute implies "choice" type.
644 self.type = "choice"
645 else:
646 # No type given? "string" is the most sensible default.
647 self.type = "string"
648 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000649 # Allow type objects or builtin type conversion functions
650 # (int, str, etc.) as an alternative to their names. (The
Georg Brandl1a3284e2007-12-02 09:40:06 +0000651 # complicated check of builtins is only necessary for
Thomas Wouters477c8d52006-05-27 19:21:47 +0000652 # Python 2.1 and earlier, and is short-circuited by the
653 # first check on modern Pythons.)
Georg Brandl1a3284e2007-12-02 09:40:06 +0000654 import builtins
Guido van Rossum13257902007-06-07 23:15:56 +0000655 if ( isinstance(self.type, type) or
Thomas Wouters477c8d52006-05-27 19:21:47 +0000656 (hasattr(self.type, "__name__") and
Georg Brandl1a3284e2007-12-02 09:40:06 +0000657 getattr(builtins, self.type.__name__, None) is self.type) ):
Greg Wardeba20e62004-07-31 16:15:44 +0000658 self.type = self.type.__name__
Thomas Wouters477c8d52006-05-27 19:21:47 +0000659
Greg Wardeba20e62004-07-31 16:15:44 +0000660 if self.type == "str":
661 self.type = "string"
662
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000663 if self.type not in self.TYPES:
664 raise OptionError("invalid option type: %r" % self.type, self)
665 if self.action not in self.TYPED_ACTIONS:
666 raise OptionError(
667 "must not supply a type for action %r" % self.action, self)
668
669 def _check_choice(self):
670 if self.type == "choice":
671 if self.choices is None:
672 raise OptionError(
673 "must supply a list of choices for type 'choice'", self)
Guido van Rossum13257902007-06-07 23:15:56 +0000674 elif not isinstance(self.choices, (tuple, list)):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000675 raise OptionError(
676 "choices must be a list of strings ('%s' supplied)"
677 % str(type(self.choices)).split("'")[1], self)
678 elif self.choices is not None:
679 raise OptionError(
680 "must not supply choices for type %r" % self.type, self)
681
Greg Wardeba20e62004-07-31 16:15:44 +0000682 def _check_dest(self):
683 # No destination given, and we need one for this action. The
684 # self.type check is for callbacks that take a value.
685 takes_value = (self.action in self.STORE_ACTIONS or
686 self.type is not None)
687 if self.dest is None and takes_value:
688
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000689 # Glean a destination from the first long option string,
690 # or from the first short option string if no long options.
691 if self._long_opts:
692 # eg. "--foo-bar" -> "foo_bar"
693 self.dest = self._long_opts[0][2:].replace('-', '_')
694 else:
695 self.dest = self._short_opts[0][1]
696
Greg Wardeba20e62004-07-31 16:15:44 +0000697 def _check_const(self):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000698 if self.action not in self.CONST_ACTIONS and self.const is not None:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000699 raise OptionError(
700 "'const' must not be supplied for action %r" % self.action,
701 self)
702
Greg Wardeba20e62004-07-31 16:15:44 +0000703 def _check_nargs(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000704 if self.action in self.TYPED_ACTIONS:
705 if self.nargs is None:
706 self.nargs = 1
707 elif self.nargs is not None:
708 raise OptionError(
709 "'nargs' must not be supplied for action %r" % self.action,
710 self)
711
Greg Wardeba20e62004-07-31 16:15:44 +0000712 def _check_callback(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000713 if self.action == "callback":
Guido van Rossumd59da4b2007-05-22 18:11:13 +0000714 if not hasattr(self.callback, '__call__'):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000715 raise OptionError(
716 "callback not callable: %r" % self.callback, self)
717 if (self.callback_args is not None and
Guido van Rossum13257902007-06-07 23:15:56 +0000718 not isinstance(self.callback_args, tuple)):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000719 raise OptionError(
720 "callback_args, if supplied, must be a tuple: not %r"
721 % self.callback_args, self)
722 if (self.callback_kwargs is not None and
Guido van Rossum13257902007-06-07 23:15:56 +0000723 not isinstance(self.callback_kwargs, dict)):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000724 raise OptionError(
725 "callback_kwargs, if supplied, must be a dict: not %r"
726 % self.callback_kwargs, self)
727 else:
728 if self.callback is not None:
729 raise OptionError(
730 "callback supplied (%r) for non-callback option"
731 % self.callback, self)
732 if self.callback_args is not None:
733 raise OptionError(
734 "callback_args supplied for non-callback option", self)
735 if self.callback_kwargs is not None:
736 raise OptionError(
737 "callback_kwargs supplied for non-callback option", self)
738
739
740 CHECK_METHODS = [_check_action,
741 _check_type,
742 _check_choice,
743 _check_dest,
744 _check_const,
745 _check_nargs,
746 _check_callback]
747
748
749 # -- Miscellaneous methods -----------------------------------------
750
Greg Wardeba20e62004-07-31 16:15:44 +0000751 def __str__(self):
Greg Ward2492fcf2003-04-21 02:40:34 +0000752 return "/".join(self._short_opts + self._long_opts)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000753
Greg Wardeba20e62004-07-31 16:15:44 +0000754 __repr__ = _repr
755
756 def takes_value(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000757 return self.type is not None
758
Greg Wardeba20e62004-07-31 16:15:44 +0000759 def get_opt_string(self):
760 if self._long_opts:
761 return self._long_opts[0]
762 else:
763 return self._short_opts[0]
764
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000765
766 # -- Processing methods --------------------------------------------
767
Greg Wardeba20e62004-07-31 16:15:44 +0000768 def check_value(self, opt, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000769 checker = self.TYPE_CHECKER.get(self.type)
770 if checker is None:
771 return value
772 else:
773 return checker(self, opt, value)
774
Greg Wardeba20e62004-07-31 16:15:44 +0000775 def convert_value(self, opt, value):
776 if value is not None:
777 if self.nargs == 1:
778 return self.check_value(opt, value)
779 else:
780 return tuple([self.check_value(opt, v) for v in value])
781
782 def process(self, opt, value, values, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000783
784 # First, convert the value(s) to the right type. Howl if any
785 # value(s) are bogus.
Greg Wardeba20e62004-07-31 16:15:44 +0000786 value = self.convert_value(opt, value)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000787
788 # And then take whatever action is expected of us.
789 # This is a separate method to make life easier for
790 # subclasses to add new actions.
791 return self.take_action(
792 self.action, self.dest, opt, value, values, parser)
793
Greg Wardeba20e62004-07-31 16:15:44 +0000794 def take_action(self, action, dest, opt, value, values, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000795 if action == "store":
796 setattr(values, dest, value)
797 elif action == "store_const":
798 setattr(values, dest, self.const)
799 elif action == "store_true":
Greg Ward2492fcf2003-04-21 02:40:34 +0000800 setattr(values, dest, True)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000801 elif action == "store_false":
Greg Ward2492fcf2003-04-21 02:40:34 +0000802 setattr(values, dest, False)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000803 elif action == "append":
804 values.ensure_value(dest, []).append(value)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000805 elif action == "append_const":
806 values.ensure_value(dest, []).append(self.const)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000807 elif action == "count":
808 setattr(values, dest, values.ensure_value(dest, 0) + 1)
809 elif action == "callback":
810 args = self.callback_args or ()
811 kwargs = self.callback_kwargs or {}
812 self.callback(self, opt, value, parser, *args, **kwargs)
813 elif action == "help":
814 parser.print_help()
Greg Ward48aa84b2004-10-27 02:20:04 +0000815 parser.exit()
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000816 elif action == "version":
817 parser.print_version()
Greg Ward48aa84b2004-10-27 02:20:04 +0000818 parser.exit()
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000819 else:
Benjamin Peterson4469d0c2008-11-30 22:46:23 +0000820 raise ValueError("unknown action %r" % self.action)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000821
822 return 1
823
824# class Option
Greg Ward2492fcf2003-04-21 02:40:34 +0000825
826
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000827SUPPRESS_HELP = "SUPPRESS"+"HELP"
828SUPPRESS_USAGE = "SUPPRESS"+"USAGE"
829
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000830class Values:
831
Greg Wardeba20e62004-07-31 16:15:44 +0000832 def __init__(self, defaults=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000833 if defaults:
834 for (attr, val) in defaults.items():
835 setattr(self, attr, val)
836
Greg Wardeba20e62004-07-31 16:15:44 +0000837 def __str__(self):
838 return str(self.__dict__)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000839
Greg Wardeba20e62004-07-31 16:15:44 +0000840 __repr__ = _repr
841
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000842 def __eq__(self, other):
Greg Wardeba20e62004-07-31 16:15:44 +0000843 if isinstance(other, Values):
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000844 return self.__dict__ == other.__dict__
Guido van Rossum13257902007-06-07 23:15:56 +0000845 elif isinstance(other, dict):
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000846 return self.__dict__ == other
Greg Wardeba20e62004-07-31 16:15:44 +0000847 else:
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000848 return NotImplemented
Greg Wardeba20e62004-07-31 16:15:44 +0000849
850 def _update_careful(self, dict):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000851 """
852 Update the option values from an arbitrary dictionary, but only
853 use keys from dict that already have a corresponding attribute
854 in self. Any keys in dict without a corresponding attribute
855 are silently ignored.
856 """
857 for attr in dir(self):
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000858 if attr in dict:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000859 dval = dict[attr]
860 if dval is not None:
861 setattr(self, attr, dval)
862
Greg Wardeba20e62004-07-31 16:15:44 +0000863 def _update_loose(self, dict):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000864 """
865 Update the option values from an arbitrary dictionary,
866 using all keys from the dictionary regardless of whether
867 they have a corresponding attribute in self or not.
868 """
869 self.__dict__.update(dict)
870
Greg Wardeba20e62004-07-31 16:15:44 +0000871 def _update(self, dict, mode):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000872 if mode == "careful":
873 self._update_careful(dict)
874 elif mode == "loose":
875 self._update_loose(dict)
876 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000877 raise ValueError("invalid update mode: %r" % mode)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000878
Greg Wardeba20e62004-07-31 16:15:44 +0000879 def read_module(self, modname, mode="careful"):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000880 __import__(modname)
881 mod = sys.modules[modname]
882 self._update(vars(mod), mode)
883
Greg Wardeba20e62004-07-31 16:15:44 +0000884 def read_file(self, filename, mode="careful"):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000885 vars = {}
Neal Norwitz01688022007-08-12 00:43:29 +0000886 exec(open(filename).read(), vars)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000887 self._update(vars, mode)
888
Greg Wardeba20e62004-07-31 16:15:44 +0000889 def ensure_value(self, attr, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000890 if not hasattr(self, attr) or getattr(self, attr) is None:
891 setattr(self, attr, value)
892 return getattr(self, attr)
893
894
895class OptionContainer:
896
897 """
898 Abstract base class.
899
900 Class attributes:
901 standard_option_list : [Option]
902 list of standard options that will be accepted by all instances
903 of this parser class (intended to be overridden by subclasses).
904
905 Instance attributes:
906 option_list : [Option]
907 the list of Option objects contained by this OptionContainer
908 _short_opt : { string : Option }
909 dictionary mapping short option strings, eg. "-f" or "-X",
910 to the Option instances that implement them. If an Option
911 has multiple short option strings, it will appears in this
912 dictionary multiple times. [1]
913 _long_opt : { string : Option }
914 dictionary mapping long option strings, eg. "--file" or
915 "--exclude", to the Option instances that implement them.
916 Again, a given Option can occur multiple times in this
917 dictionary. [1]
918 defaults : { string : any }
919 dictionary mapping option destination names to default
920 values for each destination [1]
921
922 [1] These mappings are common to (shared by) all components of the
923 controlling OptionParser, where they are initially created.
924
925 """
926
Greg Wardeba20e62004-07-31 16:15:44 +0000927 def __init__(self, option_class, conflict_handler, description):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000928 # Initialize the option list and related data structures.
929 # This method must be provided by subclasses, and it must
930 # initialize at least the following instance attributes:
931 # option_list, _short_opt, _long_opt, defaults.
932 self._create_option_list()
933
934 self.option_class = option_class
935 self.set_conflict_handler(conflict_handler)
936 self.set_description(description)
937
Greg Wardeba20e62004-07-31 16:15:44 +0000938 def _create_option_mappings(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000939 # For use by OptionParser constructor -- create the master
940 # option mappings used by this OptionParser and all
941 # OptionGroups that it owns.
942 self._short_opt = {} # single letter -> Option instance
943 self._long_opt = {} # long option -> Option instance
944 self.defaults = {} # maps option dest -> default value
945
946
Greg Wardeba20e62004-07-31 16:15:44 +0000947 def _share_option_mappings(self, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000948 # For use by OptionGroup constructor -- use shared option
949 # mappings from the OptionParser that owns this OptionGroup.
950 self._short_opt = parser._short_opt
951 self._long_opt = parser._long_opt
952 self.defaults = parser.defaults
953
Greg Wardeba20e62004-07-31 16:15:44 +0000954 def set_conflict_handler(self, handler):
Greg Ward48aa84b2004-10-27 02:20:04 +0000955 if handler not in ("error", "resolve"):
Collin Winterce36ad82007-08-30 01:19:48 +0000956 raise ValueError("invalid conflict_resolution value %r" % handler)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000957 self.conflict_handler = handler
958
Greg Wardeba20e62004-07-31 16:15:44 +0000959 def set_description(self, description):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000960 self.description = description
961
Greg Wardeba20e62004-07-31 16:15:44 +0000962 def get_description(self):
963 return self.description
964
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000965
Thomas Wouters477c8d52006-05-27 19:21:47 +0000966 def destroy(self):
967 """see OptionParser.destroy()."""
968 del self._short_opt
969 del self._long_opt
970 del self.defaults
971
972
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000973 # -- Option-adding methods -----------------------------------------
974
Greg Wardeba20e62004-07-31 16:15:44 +0000975 def _check_conflict(self, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000976 conflict_opts = []
977 for opt in option._short_opts:
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000978 if opt in self._short_opt:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000979 conflict_opts.append((opt, self._short_opt[opt]))
980 for opt in option._long_opts:
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000981 if opt in self._long_opt:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000982 conflict_opts.append((opt, self._long_opt[opt]))
983
984 if conflict_opts:
985 handler = self.conflict_handler
Greg Ward48aa84b2004-10-27 02:20:04 +0000986 if handler == "error":
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000987 raise OptionConflictError(
988 "conflicting option string(s): %s"
989 % ", ".join([co[0] for co in conflict_opts]),
990 option)
Greg Ward48aa84b2004-10-27 02:20:04 +0000991 elif handler == "resolve":
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000992 for (opt, c_option) in conflict_opts:
993 if opt.startswith("--"):
994 c_option._long_opts.remove(opt)
995 del self._long_opt[opt]
996 else:
997 c_option._short_opts.remove(opt)
998 del self._short_opt[opt]
999 if not (c_option._short_opts or c_option._long_opts):
1000 c_option.container.option_list.remove(c_option)
1001
Greg Wardeba20e62004-07-31 16:15:44 +00001002 def add_option(self, *args, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001003 """add_option(Option)
1004 add_option(opt_str, ..., kwarg=val, ...)
1005 """
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001006 if isinstance(args[0], str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001007 option = self.option_class(*args, **kwargs)
1008 elif len(args) == 1 and not kwargs:
1009 option = args[0]
1010 if not isinstance(option, Option):
Collin Winterce36ad82007-08-30 01:19:48 +00001011 raise TypeError("not an Option instance: %r" % option)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001012 else:
Collin Winterce36ad82007-08-30 01:19:48 +00001013 raise TypeError("invalid arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001014
1015 self._check_conflict(option)
1016
1017 self.option_list.append(option)
1018 option.container = self
1019 for opt in option._short_opts:
1020 self._short_opt[opt] = option
1021 for opt in option._long_opts:
1022 self._long_opt[opt] = option
1023
1024 if option.dest is not None: # option has a dest, we need a default
1025 if option.default is not NO_DEFAULT:
1026 self.defaults[option.dest] = option.default
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001027 elif option.dest not in self.defaults:
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001028 self.defaults[option.dest] = None
1029
1030 return option
1031
Greg Wardeba20e62004-07-31 16:15:44 +00001032 def add_options(self, option_list):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001033 for option in option_list:
1034 self.add_option(option)
1035
1036 # -- Option query/removal methods ----------------------------------
1037
Greg Wardeba20e62004-07-31 16:15:44 +00001038 def get_option(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001039 return (self._short_opt.get(opt_str) or
1040 self._long_opt.get(opt_str))
1041
Greg Wardeba20e62004-07-31 16:15:44 +00001042 def has_option(self, opt_str):
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001043 return (opt_str in self._short_opt or
Guido van Rossum93662412006-08-19 16:09:41 +00001044 opt_str in self._long_opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001045
Greg Wardeba20e62004-07-31 16:15:44 +00001046 def remove_option(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001047 option = self._short_opt.get(opt_str)
1048 if option is None:
1049 option = self._long_opt.get(opt_str)
1050 if option is None:
1051 raise ValueError("no such option %r" % opt_str)
1052
1053 for opt in option._short_opts:
1054 del self._short_opt[opt]
1055 for opt in option._long_opts:
1056 del self._long_opt[opt]
1057 option.container.option_list.remove(option)
1058
1059
1060 # -- Help-formatting methods ---------------------------------------
1061
Greg Wardeba20e62004-07-31 16:15:44 +00001062 def format_option_help(self, formatter):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001063 if not self.option_list:
1064 return ""
1065 result = []
1066 for option in self.option_list:
1067 if not option.help is SUPPRESS_HELP:
1068 result.append(formatter.format_option(option))
1069 return "".join(result)
1070
Greg Wardeba20e62004-07-31 16:15:44 +00001071 def format_description(self, formatter):
1072 return formatter.format_description(self.get_description())
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001073
Greg Wardeba20e62004-07-31 16:15:44 +00001074 def format_help(self, formatter):
1075 result = []
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001076 if self.description:
Greg Wardeba20e62004-07-31 16:15:44 +00001077 result.append(self.format_description(formatter))
1078 if self.option_list:
1079 result.append(self.format_option_help(formatter))
1080 return "\n".join(result)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001081
1082
1083class OptionGroup (OptionContainer):
1084
Greg Wardeba20e62004-07-31 16:15:44 +00001085 def __init__(self, parser, title, description=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001086 self.parser = parser
1087 OptionContainer.__init__(
1088 self, parser.option_class, parser.conflict_handler, description)
1089 self.title = title
1090
Greg Wardeba20e62004-07-31 16:15:44 +00001091 def _create_option_list(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001092 self.option_list = []
1093 self._share_option_mappings(self.parser)
1094
Greg Wardeba20e62004-07-31 16:15:44 +00001095 def set_title(self, title):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001096 self.title = title
1097
Thomas Wouters477c8d52006-05-27 19:21:47 +00001098 def destroy(self):
1099 """see OptionParser.destroy()."""
1100 OptionContainer.destroy(self)
1101 del self.option_list
1102
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001103 # -- Help-formatting methods ---------------------------------------
1104
Greg Wardeba20e62004-07-31 16:15:44 +00001105 def format_help(self, formatter):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001106 result = formatter.format_heading(self.title)
1107 formatter.indent()
1108 result += OptionContainer.format_help(self, formatter)
1109 formatter.dedent()
1110 return result
1111
1112
1113class OptionParser (OptionContainer):
1114
1115 """
1116 Class attributes:
1117 standard_option_list : [Option]
1118 list of standard options that will be accepted by all instances
1119 of this parser class (intended to be overridden by subclasses).
1120
1121 Instance attributes:
1122 usage : string
1123 a usage string for your program. Before it is displayed
1124 to the user, "%prog" will be expanded to the name of
Greg Ward2492fcf2003-04-21 02:40:34 +00001125 your program (self.prog or os.path.basename(sys.argv[0])).
1126 prog : string
1127 the name of the current program (to override
1128 os.path.basename(sys.argv[0])).
Thomas Wouters477c8d52006-05-27 19:21:47 +00001129 epilog : string
1130 paragraph of help text to print after option help
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001131
Greg Wardeba20e62004-07-31 16:15:44 +00001132 option_groups : [OptionGroup]
1133 list of option groups in this parser (option groups are
1134 irrelevant for parsing the command-line, but very useful
1135 for generating help)
1136
1137 allow_interspersed_args : bool = true
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001138 if true, positional arguments may be interspersed with options.
1139 Assuming -a and -b each take a single argument, the command-line
1140 -ablah foo bar -bboo baz
1141 will be interpreted the same as
1142 -ablah -bboo -- foo bar baz
1143 If this flag were false, that command line would be interpreted as
1144 -ablah -- foo bar -bboo baz
1145 -- ie. we stop processing options as soon as we see the first
1146 non-option argument. (This is the tradition followed by
1147 Python's getopt module, Perl's Getopt::Std, and other argument-
1148 parsing libraries, but it is generally annoying to users.)
1149
Greg Wardeba20e62004-07-31 16:15:44 +00001150 process_default_values : bool = true
1151 if true, option default values are processed similarly to option
1152 values from the command line: that is, they are passed to the
1153 type-checking function for the option's type (as long as the
1154 default value is a string). (This really only matters if you
1155 have defined custom types; see SF bug #955889.) Set it to false
1156 to restore the behaviour of Optik 1.4.1 and earlier.
1157
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001158 rargs : [string]
1159 the argument list currently being parsed. Only set when
1160 parse_args() is active, and continually trimmed down as
1161 we consume arguments. Mainly there for the benefit of
1162 callback options.
1163 largs : [string]
1164 the list of leftover arguments that we have skipped while
1165 parsing options. If allow_interspersed_args is false, this
1166 list is always empty.
1167 values : Values
1168 the set of option values currently being accumulated. Only
1169 set when parse_args() is active. Also mainly for callbacks.
1170
1171 Because of the 'rargs', 'largs', and 'values' attributes,
1172 OptionParser is not thread-safe. If, for some perverse reason, you
1173 need to parse command-line arguments simultaneously in different
1174 threads, use different OptionParser instances.
1175
1176 """
1177
1178 standard_option_list = []
1179
Greg Wardeba20e62004-07-31 16:15:44 +00001180 def __init__(self,
1181 usage=None,
1182 option_list=None,
1183 option_class=Option,
1184 version=None,
1185 conflict_handler="error",
1186 description=None,
1187 formatter=None,
1188 add_help_option=True,
Thomas Wouters477c8d52006-05-27 19:21:47 +00001189 prog=None,
1190 epilog=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001191 OptionContainer.__init__(
1192 self, option_class, conflict_handler, description)
1193 self.set_usage(usage)
Greg Ward2492fcf2003-04-21 02:40:34 +00001194 self.prog = prog
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001195 self.version = version
Greg Wardeba20e62004-07-31 16:15:44 +00001196 self.allow_interspersed_args = True
1197 self.process_default_values = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001198 if formatter is None:
1199 formatter = IndentedHelpFormatter()
1200 self.formatter = formatter
Greg Wardeba20e62004-07-31 16:15:44 +00001201 self.formatter.set_parser(self)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001202 self.epilog = epilog
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001203
1204 # Populate the option list; initial sources are the
1205 # standard_option_list class attribute, the 'option_list'
Greg Wardeba20e62004-07-31 16:15:44 +00001206 # argument, and (if applicable) the _add_version_option() and
1207 # _add_help_option() methods.
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001208 self._populate_option_list(option_list,
1209 add_help=add_help_option)
1210
1211 self._init_parsing_state()
1212
Thomas Wouters477c8d52006-05-27 19:21:47 +00001213
1214 def destroy(self):
1215 """
1216 Declare that you are done with this OptionParser. This cleans up
1217 reference cycles so the OptionParser (and all objects referenced by
1218 it) can be garbage-collected promptly. After calling destroy(), the
1219 OptionParser is unusable.
1220 """
1221 OptionContainer.destroy(self)
1222 for group in self.option_groups:
1223 group.destroy()
1224 del self.option_list
1225 del self.option_groups
1226 del self.formatter
1227
1228
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001229 # -- Private methods -----------------------------------------------
1230 # (used by our or OptionContainer's constructor)
1231
Greg Wardeba20e62004-07-31 16:15:44 +00001232 def _create_option_list(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001233 self.option_list = []
1234 self.option_groups = []
1235 self._create_option_mappings()
1236
Greg Wardeba20e62004-07-31 16:15:44 +00001237 def _add_help_option(self):
1238 self.add_option("-h", "--help",
1239 action="help",
1240 help=_("show this help message and exit"))
1241
1242 def _add_version_option(self):
1243 self.add_option("--version",
1244 action="version",
1245 help=_("show program's version number and exit"))
1246
1247 def _populate_option_list(self, option_list, add_help=True):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001248 if self.standard_option_list:
1249 self.add_options(self.standard_option_list)
1250 if option_list:
1251 self.add_options(option_list)
1252 if self.version:
Greg Wardeba20e62004-07-31 16:15:44 +00001253 self._add_version_option()
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001254 if add_help:
Greg Wardeba20e62004-07-31 16:15:44 +00001255 self._add_help_option()
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001256
Greg Wardeba20e62004-07-31 16:15:44 +00001257 def _init_parsing_state(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001258 # These are set in parse_args() for the convenience of callbacks.
1259 self.rargs = None
1260 self.largs = None
1261 self.values = None
1262
1263
1264 # -- Simple modifier methods ---------------------------------------
1265
Greg Wardeba20e62004-07-31 16:15:44 +00001266 def set_usage(self, usage):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001267 if usage is None:
Greg Wardeba20e62004-07-31 16:15:44 +00001268 self.usage = _("%prog [options]")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001269 elif usage is SUPPRESS_USAGE:
1270 self.usage = None
Greg Wardeba20e62004-07-31 16:15:44 +00001271 # For backwards compatibility with Optik 1.3 and earlier.
Thomas Wouters477c8d52006-05-27 19:21:47 +00001272 elif usage.lower().startswith("usage: "):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001273 self.usage = usage[7:]
1274 else:
1275 self.usage = usage
1276
Greg Wardeba20e62004-07-31 16:15:44 +00001277 def enable_interspersed_args(self):
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00001278 """Set parsing to not stop on the first non-option, allowing
1279 interspersing switches with command arguments. This is the
1280 default behavior. See also disable_interspersed_args() and the
1281 class documentation description of the attribute
1282 allow_interspersed_args."""
Greg Wardeba20e62004-07-31 16:15:44 +00001283 self.allow_interspersed_args = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001284
Greg Wardeba20e62004-07-31 16:15:44 +00001285 def disable_interspersed_args(self):
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00001286 """Set parsing to stop on the first non-option. Use this if
1287 you have a command processor which runs another command that
1288 has options of its own and you want to make sure these options
1289 don't get confused.
1290 """
Greg Wardeba20e62004-07-31 16:15:44 +00001291 self.allow_interspersed_args = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001292
Greg Wardeba20e62004-07-31 16:15:44 +00001293 def set_process_default_values(self, process):
1294 self.process_default_values = process
1295
1296 def set_default(self, dest, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001297 self.defaults[dest] = value
1298
Greg Wardeba20e62004-07-31 16:15:44 +00001299 def set_defaults(self, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001300 self.defaults.update(kwargs)
1301
Greg Wardeba20e62004-07-31 16:15:44 +00001302 def _get_all_options(self):
1303 options = self.option_list[:]
1304 for group in self.option_groups:
1305 options.extend(group.option_list)
1306 return options
1307
1308 def get_default_values(self):
1309 if not self.process_default_values:
1310 # Old, pre-Optik 1.5 behaviour.
1311 return Values(self.defaults)
1312
1313 defaults = self.defaults.copy()
1314 for option in self._get_all_options():
1315 default = defaults.get(option.dest)
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001316 if isinstance(default, str):
Greg Wardeba20e62004-07-31 16:15:44 +00001317 opt_str = option.get_opt_string()
1318 defaults[option.dest] = option.check_value(opt_str, default)
1319
1320 return Values(defaults)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001321
1322
1323 # -- OptionGroup methods -------------------------------------------
1324
Greg Wardeba20e62004-07-31 16:15:44 +00001325 def add_option_group(self, *args, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001326 # XXX lots of overlap with OptionContainer.add_option()
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001327 if isinstance(args[0], str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001328 group = OptionGroup(self, *args, **kwargs)
1329 elif len(args) == 1 and not kwargs:
1330 group = args[0]
1331 if not isinstance(group, OptionGroup):
Collin Winterce36ad82007-08-30 01:19:48 +00001332 raise TypeError("not an OptionGroup instance: %r" % group)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001333 if group.parser is not self:
Collin Winterce36ad82007-08-30 01:19:48 +00001334 raise ValueError("invalid OptionGroup (wrong parser)")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001335 else:
Collin Winterce36ad82007-08-30 01:19:48 +00001336 raise TypeError("invalid arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001337
1338 self.option_groups.append(group)
1339 return group
1340
Greg Wardeba20e62004-07-31 16:15:44 +00001341 def get_option_group(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001342 option = (self._short_opt.get(opt_str) or
1343 self._long_opt.get(opt_str))
1344 if option and option.container is not self:
1345 return option.container
1346 return None
1347
1348
1349 # -- Option-parsing methods ----------------------------------------
1350
Greg Wardeba20e62004-07-31 16:15:44 +00001351 def _get_args(self, args):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001352 if args is None:
1353 return sys.argv[1:]
1354 else:
1355 return args[:] # don't modify caller's list
1356
Greg Wardeba20e62004-07-31 16:15:44 +00001357 def parse_args(self, args=None, values=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001358 """
1359 parse_args(args : [string] = sys.argv[1:],
1360 values : Values = None)
1361 -> (values : Values, args : [string])
1362
1363 Parse the command-line options found in 'args' (default:
1364 sys.argv[1:]). Any errors result in a call to 'error()', which
1365 by default prints the usage message to stderr and calls
1366 sys.exit() with an error message. On success returns a pair
1367 (values, args) where 'values' is an Values instance (with all
1368 your option values) and 'args' is the list of arguments left
1369 over after parsing options.
1370 """
1371 rargs = self._get_args(args)
1372 if values is None:
1373 values = self.get_default_values()
1374
1375 # Store the halves of the argument list as attributes for the
1376 # convenience of callbacks:
1377 # rargs
1378 # the rest of the command-line (the "r" stands for
1379 # "remaining" or "right-hand")
1380 # largs
1381 # the leftover arguments -- ie. what's left after removing
1382 # options and their arguments (the "l" stands for "leftover"
1383 # or "left-hand")
1384 self.rargs = rargs
1385 self.largs = largs = []
1386 self.values = values
1387
1388 try:
1389 stop = self._process_args(largs, rargs, values)
Guido van Rossumb940e112007-01-10 16:19:56 +00001390 except (BadOptionError, OptionValueError) as err:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001391 self.error(str(err))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001392
1393 args = largs + rargs
1394 return self.check_values(values, args)
1395
Greg Wardeba20e62004-07-31 16:15:44 +00001396 def check_values(self, values, args):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001397 """
1398 check_values(values : Values, args : [string])
1399 -> (values : Values, args : [string])
1400
1401 Check that the supplied option values and leftover arguments are
1402 valid. Returns the option values and leftover arguments
1403 (possibly adjusted, possibly completely new -- whatever you
1404 like). Default implementation just returns the passed-in
1405 values; subclasses may override as desired.
1406 """
1407 return (values, args)
1408
Greg Wardeba20e62004-07-31 16:15:44 +00001409 def _process_args(self, largs, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001410 """_process_args(largs : [string],
1411 rargs : [string],
1412 values : Values)
1413
1414 Process command-line arguments and populate 'values', consuming
1415 options and arguments from 'rargs'. If 'allow_interspersed_args' is
1416 false, stop at the first non-option argument. If true, accumulate any
1417 interspersed non-option arguments in 'largs'.
1418 """
1419 while rargs:
1420 arg = rargs[0]
1421 # We handle bare "--" explicitly, and bare "-" is handled by the
1422 # standard arg handler since the short arg case ensures that the
1423 # len of the opt string is greater than 1.
1424 if arg == "--":
1425 del rargs[0]
1426 return
1427 elif arg[0:2] == "--":
1428 # process a single long option (possibly with value(s))
1429 self._process_long_opt(rargs, values)
1430 elif arg[:1] == "-" and len(arg) > 1:
1431 # process a cluster of short options (possibly with
1432 # value(s) for the last one only)
1433 self._process_short_opts(rargs, values)
1434 elif self.allow_interspersed_args:
1435 largs.append(arg)
1436 del rargs[0]
1437 else:
1438 return # stop now, leave this arg in rargs
1439
1440 # Say this is the original argument list:
1441 # [arg0, arg1, ..., arg(i-1), arg(i), arg(i+1), ..., arg(N-1)]
1442 # ^
1443 # (we are about to process arg(i)).
1444 #
1445 # Then rargs is [arg(i), ..., arg(N-1)] and largs is a *subset* of
1446 # [arg0, ..., arg(i-1)] (any options and their arguments will have
1447 # been removed from largs).
1448 #
1449 # The while loop will usually consume 1 or more arguments per pass.
1450 # If it consumes 1 (eg. arg is an option that takes no arguments),
1451 # then after _process_arg() is done the situation is:
1452 #
1453 # largs = subset of [arg0, ..., arg(i)]
1454 # rargs = [arg(i+1), ..., arg(N-1)]
1455 #
1456 # If allow_interspersed_args is false, largs will always be
1457 # *empty* -- still a subset of [arg0, ..., arg(i-1)], but
1458 # not a very interesting subset!
1459
Greg Wardeba20e62004-07-31 16:15:44 +00001460 def _match_long_opt(self, opt):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001461 """_match_long_opt(opt : string) -> string
1462
1463 Determine which long option string 'opt' matches, ie. which one
1464 it is an unambiguous abbrevation for. Raises BadOptionError if
1465 'opt' doesn't unambiguously match any long option string.
1466 """
1467 return _match_abbrev(opt, self._long_opt)
1468
Greg Wardeba20e62004-07-31 16:15:44 +00001469 def _process_long_opt(self, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001470 arg = rargs.pop(0)
1471
1472 # Value explicitly attached to arg? Pretend it's the next
1473 # argument.
1474 if "=" in arg:
1475 (opt, next_arg) = arg.split("=", 1)
1476 rargs.insert(0, next_arg)
Greg Wardeba20e62004-07-31 16:15:44 +00001477 had_explicit_value = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001478 else:
1479 opt = arg
Greg Wardeba20e62004-07-31 16:15:44 +00001480 had_explicit_value = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001481
1482 opt = self._match_long_opt(opt)
1483 option = self._long_opt[opt]
1484 if option.takes_value():
1485 nargs = option.nargs
1486 if len(rargs) < nargs:
Éric Araujo6a1454f2011-03-20 19:59:25 +01001487 self.error(ngettext(
1488 "%(option)s option requires %(number)d argument",
1489 "%(option)s option requires %(number)d arguments",
1490 nargs) % {"option": opt, "number": nargs})
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001491 elif nargs == 1:
1492 value = rargs.pop(0)
1493 else:
1494 value = tuple(rargs[0:nargs])
1495 del rargs[0:nargs]
1496
1497 elif had_explicit_value:
Greg Wardeba20e62004-07-31 16:15:44 +00001498 self.error(_("%s option does not take a value") % opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001499
1500 else:
1501 value = None
1502
1503 option.process(opt, value, values, self)
1504
Greg Wardeba20e62004-07-31 16:15:44 +00001505 def _process_short_opts(self, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001506 arg = rargs.pop(0)
Greg Wardeba20e62004-07-31 16:15:44 +00001507 stop = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001508 i = 1
1509 for ch in arg[1:]:
1510 opt = "-" + ch
1511 option = self._short_opt.get(opt)
1512 i += 1 # we have consumed a character
1513
1514 if not option:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001515 raise BadOptionError(opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001516 if option.takes_value():
1517 # Any characters left in arg? Pretend they're the
1518 # next arg, and stop consuming characters of arg.
1519 if i < len(arg):
1520 rargs.insert(0, arg[i:])
Greg Wardeba20e62004-07-31 16:15:44 +00001521 stop = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001522
1523 nargs = option.nargs
1524 if len(rargs) < nargs:
Éric Araujo6a1454f2011-03-20 19:59:25 +01001525 self.error(ngettext(
1526 "%(option)s option requires %(number)d argument",
1527 "%(option)s option requires %(number)d arguments",
1528 nargs) % {"option": opt, "number": nargs})
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001529 elif nargs == 1:
1530 value = rargs.pop(0)
1531 else:
1532 value = tuple(rargs[0:nargs])
1533 del rargs[0:nargs]
1534
1535 else: # option doesn't take a value
1536 value = None
1537
1538 option.process(opt, value, values, self)
1539
1540 if stop:
1541 break
1542
1543
1544 # -- Feedback methods ----------------------------------------------
1545
Greg Wardeba20e62004-07-31 16:15:44 +00001546 def get_prog_name(self):
1547 if self.prog is None:
1548 return os.path.basename(sys.argv[0])
1549 else:
1550 return self.prog
1551
1552 def expand_prog_name(self, s):
1553 return s.replace("%prog", self.get_prog_name())
1554
1555 def get_description(self):
1556 return self.expand_prog_name(self.description)
1557
Greg Ward48aa84b2004-10-27 02:20:04 +00001558 def exit(self, status=0, msg=None):
1559 if msg:
1560 sys.stderr.write(msg)
1561 sys.exit(status)
1562
Greg Wardeba20e62004-07-31 16:15:44 +00001563 def error(self, msg):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001564 """error(msg : string)
1565
1566 Print a usage message incorporating 'msg' to stderr and exit.
1567 If you override this in a subclass, it should not return -- it
1568 should either exit or raise an exception.
1569 """
1570 self.print_usage(sys.stderr)
Greg Ward48aa84b2004-10-27 02:20:04 +00001571 self.exit(2, "%s: error: %s\n" % (self.get_prog_name(), msg))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001572
Greg Wardeba20e62004-07-31 16:15:44 +00001573 def get_usage(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001574 if self.usage:
1575 return self.formatter.format_usage(
Greg Wardeba20e62004-07-31 16:15:44 +00001576 self.expand_prog_name(self.usage))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001577 else:
1578 return ""
1579
Greg Wardeba20e62004-07-31 16:15:44 +00001580 def print_usage(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001581 """print_usage(file : file = stdout)
1582
1583 Print the usage message for the current program (self.usage) to
Mark Dickinson934896d2009-02-21 20:59:32 +00001584 'file' (default stdout). Any occurrence of the string "%prog" in
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001585 self.usage is replaced with the name of the current program
1586 (basename of sys.argv[0]). Does nothing if self.usage is empty
1587 or not defined.
1588 """
1589 if self.usage:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001590 print(self.get_usage(), file=file)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001591
Greg Wardeba20e62004-07-31 16:15:44 +00001592 def get_version(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001593 if self.version:
Greg Wardeba20e62004-07-31 16:15:44 +00001594 return self.expand_prog_name(self.version)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001595 else:
1596 return ""
1597
Greg Wardeba20e62004-07-31 16:15:44 +00001598 def print_version(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001599 """print_version(file : file = stdout)
1600
1601 Print the version message for this program (self.version) to
Mark Dickinson934896d2009-02-21 20:59:32 +00001602 'file' (default stdout). As with print_usage(), any occurrence
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001603 of "%prog" in self.version is replaced by the current program's
1604 name. Does nothing if self.version is empty or undefined.
1605 """
1606 if self.version:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001607 print(self.get_version(), file=file)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001608
Greg Wardeba20e62004-07-31 16:15:44 +00001609 def format_option_help(self, formatter=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001610 if formatter is None:
1611 formatter = self.formatter
1612 formatter.store_option_strings(self)
1613 result = []
Thomas Wouters477c8d52006-05-27 19:21:47 +00001614 result.append(formatter.format_heading(_("Options")))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001615 formatter.indent()
1616 if self.option_list:
1617 result.append(OptionContainer.format_option_help(self, formatter))
1618 result.append("\n")
1619 for group in self.option_groups:
1620 result.append(group.format_help(formatter))
1621 result.append("\n")
1622 formatter.dedent()
1623 # Drop the last "\n", or the header if no options or option groups:
1624 return "".join(result[:-1])
1625
Thomas Wouters477c8d52006-05-27 19:21:47 +00001626 def format_epilog(self, formatter):
1627 return formatter.format_epilog(self.epilog)
1628
Greg Wardeba20e62004-07-31 16:15:44 +00001629 def format_help(self, formatter=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001630 if formatter is None:
1631 formatter = self.formatter
1632 result = []
1633 if self.usage:
1634 result.append(self.get_usage() + "\n")
1635 if self.description:
1636 result.append(self.format_description(formatter) + "\n")
1637 result.append(self.format_option_help(formatter))
Thomas Wouters477c8d52006-05-27 19:21:47 +00001638 result.append(self.format_epilog(formatter))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001639 return "".join(result)
1640
Greg Wardeba20e62004-07-31 16:15:44 +00001641 def print_help(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001642 """print_help(file : file = stdout)
1643
1644 Print an extended help message, listing all options and any
1645 help text provided with them, to 'file' (default stdout).
1646 """
1647 if file is None:
1648 file = sys.stdout
Guido van Rossum34d19282007-08-09 01:03:29 +00001649 file.write(self.format_help())
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001650
1651# class OptionParser
1652
1653
Greg Wardeba20e62004-07-31 16:15:44 +00001654def _match_abbrev(s, wordmap):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001655 """_match_abbrev(s : string, wordmap : {string : Option}) -> string
1656
1657 Return the string key in 'wordmap' for which 's' is an unambiguous
1658 abbreviation. If 's' is found to be ambiguous or doesn't match any of
1659 'words', raise BadOptionError.
1660 """
1661 # Is there an exact match?
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001662 if s in wordmap:
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001663 return s
1664 else:
1665 # Isolate all words with s as a prefix.
1666 possibilities = [word for word in wordmap.keys()
1667 if word.startswith(s)]
1668 # No exact match, so there had better be just one possibility.
1669 if len(possibilities) == 1:
1670 return possibilities[0]
1671 elif not possibilities:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001672 raise BadOptionError(s)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001673 else:
1674 # More than one possible completion: ambiguous prefix.
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001675 possibilities.sort()
Thomas Wouters477c8d52006-05-27 19:21:47 +00001676 raise AmbiguousOptionError(s, possibilities)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001677
1678
1679# Some day, there might be many Option classes. As of Optik 1.3, the
1680# preferred way to instantiate Options is indirectly, via make_option(),
1681# which will become a factory function when there are many Option
1682# classes.
1683make_option = Option