blob: b09a13cd344a13a4947e78c6a42003b2f5a571b3 [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])).
R David Murrayfc5ed802011-05-04 21:06:57 -04001129 description : string
1130 A paragraph of text giving a brief overview of your program.
1131 optparse reformats this paragraph to fit the current terminal
1132 width and prints it when the user requests help (after usage,
1133 but before the list of options).
Thomas Wouters477c8d52006-05-27 19:21:47 +00001134 epilog : string
1135 paragraph of help text to print after option help
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001136
Greg Wardeba20e62004-07-31 16:15:44 +00001137 option_groups : [OptionGroup]
1138 list of option groups in this parser (option groups are
1139 irrelevant for parsing the command-line, but very useful
1140 for generating help)
1141
1142 allow_interspersed_args : bool = true
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001143 if true, positional arguments may be interspersed with options.
1144 Assuming -a and -b each take a single argument, the command-line
1145 -ablah foo bar -bboo baz
1146 will be interpreted the same as
1147 -ablah -bboo -- foo bar baz
1148 If this flag were false, that command line would be interpreted as
1149 -ablah -- foo bar -bboo baz
1150 -- ie. we stop processing options as soon as we see the first
1151 non-option argument. (This is the tradition followed by
1152 Python's getopt module, Perl's Getopt::Std, and other argument-
1153 parsing libraries, but it is generally annoying to users.)
1154
Greg Wardeba20e62004-07-31 16:15:44 +00001155 process_default_values : bool = true
1156 if true, option default values are processed similarly to option
1157 values from the command line: that is, they are passed to the
1158 type-checking function for the option's type (as long as the
1159 default value is a string). (This really only matters if you
1160 have defined custom types; see SF bug #955889.) Set it to false
1161 to restore the behaviour of Optik 1.4.1 and earlier.
1162
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001163 rargs : [string]
1164 the argument list currently being parsed. Only set when
1165 parse_args() is active, and continually trimmed down as
1166 we consume arguments. Mainly there for the benefit of
1167 callback options.
1168 largs : [string]
1169 the list of leftover arguments that we have skipped while
1170 parsing options. If allow_interspersed_args is false, this
1171 list is always empty.
1172 values : Values
1173 the set of option values currently being accumulated. Only
1174 set when parse_args() is active. Also mainly for callbacks.
1175
1176 Because of the 'rargs', 'largs', and 'values' attributes,
1177 OptionParser is not thread-safe. If, for some perverse reason, you
1178 need to parse command-line arguments simultaneously in different
1179 threads, use different OptionParser instances.
1180
1181 """
1182
1183 standard_option_list = []
1184
Greg Wardeba20e62004-07-31 16:15:44 +00001185 def __init__(self,
1186 usage=None,
1187 option_list=None,
1188 option_class=Option,
1189 version=None,
1190 conflict_handler="error",
1191 description=None,
1192 formatter=None,
1193 add_help_option=True,
Thomas Wouters477c8d52006-05-27 19:21:47 +00001194 prog=None,
1195 epilog=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001196 OptionContainer.__init__(
1197 self, option_class, conflict_handler, description)
1198 self.set_usage(usage)
Greg Ward2492fcf2003-04-21 02:40:34 +00001199 self.prog = prog
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001200 self.version = version
Greg Wardeba20e62004-07-31 16:15:44 +00001201 self.allow_interspersed_args = True
1202 self.process_default_values = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001203 if formatter is None:
1204 formatter = IndentedHelpFormatter()
1205 self.formatter = formatter
Greg Wardeba20e62004-07-31 16:15:44 +00001206 self.formatter.set_parser(self)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001207 self.epilog = epilog
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001208
1209 # Populate the option list; initial sources are the
1210 # standard_option_list class attribute, the 'option_list'
Greg Wardeba20e62004-07-31 16:15:44 +00001211 # argument, and (if applicable) the _add_version_option() and
1212 # _add_help_option() methods.
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001213 self._populate_option_list(option_list,
1214 add_help=add_help_option)
1215
1216 self._init_parsing_state()
1217
Thomas Wouters477c8d52006-05-27 19:21:47 +00001218
1219 def destroy(self):
1220 """
1221 Declare that you are done with this OptionParser. This cleans up
1222 reference cycles so the OptionParser (and all objects referenced by
1223 it) can be garbage-collected promptly. After calling destroy(), the
1224 OptionParser is unusable.
1225 """
1226 OptionContainer.destroy(self)
1227 for group in self.option_groups:
1228 group.destroy()
1229 del self.option_list
1230 del self.option_groups
1231 del self.formatter
1232
1233
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001234 # -- Private methods -----------------------------------------------
1235 # (used by our or OptionContainer's constructor)
1236
Greg Wardeba20e62004-07-31 16:15:44 +00001237 def _create_option_list(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001238 self.option_list = []
1239 self.option_groups = []
1240 self._create_option_mappings()
1241
Greg Wardeba20e62004-07-31 16:15:44 +00001242 def _add_help_option(self):
1243 self.add_option("-h", "--help",
1244 action="help",
1245 help=_("show this help message and exit"))
1246
1247 def _add_version_option(self):
1248 self.add_option("--version",
1249 action="version",
1250 help=_("show program's version number and exit"))
1251
1252 def _populate_option_list(self, option_list, add_help=True):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001253 if self.standard_option_list:
1254 self.add_options(self.standard_option_list)
1255 if option_list:
1256 self.add_options(option_list)
1257 if self.version:
Greg Wardeba20e62004-07-31 16:15:44 +00001258 self._add_version_option()
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001259 if add_help:
Greg Wardeba20e62004-07-31 16:15:44 +00001260 self._add_help_option()
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001261
Greg Wardeba20e62004-07-31 16:15:44 +00001262 def _init_parsing_state(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001263 # These are set in parse_args() for the convenience of callbacks.
1264 self.rargs = None
1265 self.largs = None
1266 self.values = None
1267
1268
1269 # -- Simple modifier methods ---------------------------------------
1270
Greg Wardeba20e62004-07-31 16:15:44 +00001271 def set_usage(self, usage):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001272 if usage is None:
Greg Wardeba20e62004-07-31 16:15:44 +00001273 self.usage = _("%prog [options]")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001274 elif usage is SUPPRESS_USAGE:
1275 self.usage = None
Greg Wardeba20e62004-07-31 16:15:44 +00001276 # For backwards compatibility with Optik 1.3 and earlier.
Thomas Wouters477c8d52006-05-27 19:21:47 +00001277 elif usage.lower().startswith("usage: "):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001278 self.usage = usage[7:]
1279 else:
1280 self.usage = usage
1281
Greg Wardeba20e62004-07-31 16:15:44 +00001282 def enable_interspersed_args(self):
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00001283 """Set parsing to not stop on the first non-option, allowing
1284 interspersing switches with command arguments. This is the
1285 default behavior. See also disable_interspersed_args() and the
1286 class documentation description of the attribute
1287 allow_interspersed_args."""
Greg Wardeba20e62004-07-31 16:15:44 +00001288 self.allow_interspersed_args = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001289
Greg Wardeba20e62004-07-31 16:15:44 +00001290 def disable_interspersed_args(self):
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00001291 """Set parsing to stop on the first non-option. Use this if
1292 you have a command processor which runs another command that
1293 has options of its own and you want to make sure these options
1294 don't get confused.
1295 """
Greg Wardeba20e62004-07-31 16:15:44 +00001296 self.allow_interspersed_args = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001297
Greg Wardeba20e62004-07-31 16:15:44 +00001298 def set_process_default_values(self, process):
1299 self.process_default_values = process
1300
1301 def set_default(self, dest, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001302 self.defaults[dest] = value
1303
Greg Wardeba20e62004-07-31 16:15:44 +00001304 def set_defaults(self, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001305 self.defaults.update(kwargs)
1306
Greg Wardeba20e62004-07-31 16:15:44 +00001307 def _get_all_options(self):
1308 options = self.option_list[:]
1309 for group in self.option_groups:
1310 options.extend(group.option_list)
1311 return options
1312
1313 def get_default_values(self):
1314 if not self.process_default_values:
1315 # Old, pre-Optik 1.5 behaviour.
1316 return Values(self.defaults)
1317
1318 defaults = self.defaults.copy()
1319 for option in self._get_all_options():
1320 default = defaults.get(option.dest)
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001321 if isinstance(default, str):
Greg Wardeba20e62004-07-31 16:15:44 +00001322 opt_str = option.get_opt_string()
1323 defaults[option.dest] = option.check_value(opt_str, default)
1324
1325 return Values(defaults)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001326
1327
1328 # -- OptionGroup methods -------------------------------------------
1329
Greg Wardeba20e62004-07-31 16:15:44 +00001330 def add_option_group(self, *args, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001331 # XXX lots of overlap with OptionContainer.add_option()
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001332 if isinstance(args[0], str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001333 group = OptionGroup(self, *args, **kwargs)
1334 elif len(args) == 1 and not kwargs:
1335 group = args[0]
1336 if not isinstance(group, OptionGroup):
Collin Winterce36ad82007-08-30 01:19:48 +00001337 raise TypeError("not an OptionGroup instance: %r" % group)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001338 if group.parser is not self:
Collin Winterce36ad82007-08-30 01:19:48 +00001339 raise ValueError("invalid OptionGroup (wrong parser)")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001340 else:
Collin Winterce36ad82007-08-30 01:19:48 +00001341 raise TypeError("invalid arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001342
1343 self.option_groups.append(group)
1344 return group
1345
Greg Wardeba20e62004-07-31 16:15:44 +00001346 def get_option_group(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001347 option = (self._short_opt.get(opt_str) or
1348 self._long_opt.get(opt_str))
1349 if option and option.container is not self:
1350 return option.container
1351 return None
1352
1353
1354 # -- Option-parsing methods ----------------------------------------
1355
Greg Wardeba20e62004-07-31 16:15:44 +00001356 def _get_args(self, args):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001357 if args is None:
1358 return sys.argv[1:]
1359 else:
1360 return args[:] # don't modify caller's list
1361
Greg Wardeba20e62004-07-31 16:15:44 +00001362 def parse_args(self, args=None, values=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001363 """
1364 parse_args(args : [string] = sys.argv[1:],
1365 values : Values = None)
1366 -> (values : Values, args : [string])
1367
1368 Parse the command-line options found in 'args' (default:
1369 sys.argv[1:]). Any errors result in a call to 'error()', which
1370 by default prints the usage message to stderr and calls
1371 sys.exit() with an error message. On success returns a pair
1372 (values, args) where 'values' is an Values instance (with all
1373 your option values) and 'args' is the list of arguments left
1374 over after parsing options.
1375 """
1376 rargs = self._get_args(args)
1377 if values is None:
1378 values = self.get_default_values()
1379
1380 # Store the halves of the argument list as attributes for the
1381 # convenience of callbacks:
1382 # rargs
1383 # the rest of the command-line (the "r" stands for
1384 # "remaining" or "right-hand")
1385 # largs
1386 # the leftover arguments -- ie. what's left after removing
1387 # options and their arguments (the "l" stands for "leftover"
1388 # or "left-hand")
1389 self.rargs = rargs
1390 self.largs = largs = []
1391 self.values = values
1392
1393 try:
1394 stop = self._process_args(largs, rargs, values)
Guido van Rossumb940e112007-01-10 16:19:56 +00001395 except (BadOptionError, OptionValueError) as err:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001396 self.error(str(err))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001397
1398 args = largs + rargs
1399 return self.check_values(values, args)
1400
Greg Wardeba20e62004-07-31 16:15:44 +00001401 def check_values(self, values, args):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001402 """
1403 check_values(values : Values, args : [string])
1404 -> (values : Values, args : [string])
1405
1406 Check that the supplied option values and leftover arguments are
1407 valid. Returns the option values and leftover arguments
1408 (possibly adjusted, possibly completely new -- whatever you
1409 like). Default implementation just returns the passed-in
1410 values; subclasses may override as desired.
1411 """
1412 return (values, args)
1413
Greg Wardeba20e62004-07-31 16:15:44 +00001414 def _process_args(self, largs, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001415 """_process_args(largs : [string],
1416 rargs : [string],
1417 values : Values)
1418
1419 Process command-line arguments and populate 'values', consuming
1420 options and arguments from 'rargs'. If 'allow_interspersed_args' is
1421 false, stop at the first non-option argument. If true, accumulate any
1422 interspersed non-option arguments in 'largs'.
1423 """
1424 while rargs:
1425 arg = rargs[0]
1426 # We handle bare "--" explicitly, and bare "-" is handled by the
1427 # standard arg handler since the short arg case ensures that the
1428 # len of the opt string is greater than 1.
1429 if arg == "--":
1430 del rargs[0]
1431 return
1432 elif arg[0:2] == "--":
1433 # process a single long option (possibly with value(s))
1434 self._process_long_opt(rargs, values)
1435 elif arg[:1] == "-" and len(arg) > 1:
1436 # process a cluster of short options (possibly with
1437 # value(s) for the last one only)
1438 self._process_short_opts(rargs, values)
1439 elif self.allow_interspersed_args:
1440 largs.append(arg)
1441 del rargs[0]
1442 else:
1443 return # stop now, leave this arg in rargs
1444
1445 # Say this is the original argument list:
1446 # [arg0, arg1, ..., arg(i-1), arg(i), arg(i+1), ..., arg(N-1)]
1447 # ^
1448 # (we are about to process arg(i)).
1449 #
1450 # Then rargs is [arg(i), ..., arg(N-1)] and largs is a *subset* of
1451 # [arg0, ..., arg(i-1)] (any options and their arguments will have
1452 # been removed from largs).
1453 #
1454 # The while loop will usually consume 1 or more arguments per pass.
1455 # If it consumes 1 (eg. arg is an option that takes no arguments),
1456 # then after _process_arg() is done the situation is:
1457 #
1458 # largs = subset of [arg0, ..., arg(i)]
1459 # rargs = [arg(i+1), ..., arg(N-1)]
1460 #
1461 # If allow_interspersed_args is false, largs will always be
1462 # *empty* -- still a subset of [arg0, ..., arg(i-1)], but
1463 # not a very interesting subset!
1464
Greg Wardeba20e62004-07-31 16:15:44 +00001465 def _match_long_opt(self, opt):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001466 """_match_long_opt(opt : string) -> string
1467
1468 Determine which long option string 'opt' matches, ie. which one
1469 it is an unambiguous abbrevation for. Raises BadOptionError if
1470 'opt' doesn't unambiguously match any long option string.
1471 """
1472 return _match_abbrev(opt, self._long_opt)
1473
Greg Wardeba20e62004-07-31 16:15:44 +00001474 def _process_long_opt(self, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001475 arg = rargs.pop(0)
1476
1477 # Value explicitly attached to arg? Pretend it's the next
1478 # argument.
1479 if "=" in arg:
1480 (opt, next_arg) = arg.split("=", 1)
1481 rargs.insert(0, next_arg)
Greg Wardeba20e62004-07-31 16:15:44 +00001482 had_explicit_value = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001483 else:
1484 opt = arg
Greg Wardeba20e62004-07-31 16:15:44 +00001485 had_explicit_value = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001486
1487 opt = self._match_long_opt(opt)
1488 option = self._long_opt[opt]
1489 if option.takes_value():
1490 nargs = option.nargs
1491 if len(rargs) < nargs:
Éric Araujo6a1454f2011-03-20 19:59:25 +01001492 self.error(ngettext(
1493 "%(option)s option requires %(number)d argument",
1494 "%(option)s option requires %(number)d arguments",
1495 nargs) % {"option": opt, "number": nargs})
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001496 elif nargs == 1:
1497 value = rargs.pop(0)
1498 else:
1499 value = tuple(rargs[0:nargs])
1500 del rargs[0:nargs]
1501
1502 elif had_explicit_value:
Greg Wardeba20e62004-07-31 16:15:44 +00001503 self.error(_("%s option does not take a value") % opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001504
1505 else:
1506 value = None
1507
1508 option.process(opt, value, values, self)
1509
Greg Wardeba20e62004-07-31 16:15:44 +00001510 def _process_short_opts(self, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001511 arg = rargs.pop(0)
Greg Wardeba20e62004-07-31 16:15:44 +00001512 stop = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001513 i = 1
1514 for ch in arg[1:]:
1515 opt = "-" + ch
1516 option = self._short_opt.get(opt)
1517 i += 1 # we have consumed a character
1518
1519 if not option:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001520 raise BadOptionError(opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001521 if option.takes_value():
1522 # Any characters left in arg? Pretend they're the
1523 # next arg, and stop consuming characters of arg.
1524 if i < len(arg):
1525 rargs.insert(0, arg[i:])
Greg Wardeba20e62004-07-31 16:15:44 +00001526 stop = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001527
1528 nargs = option.nargs
1529 if len(rargs) < nargs:
Éric Araujo6a1454f2011-03-20 19:59:25 +01001530 self.error(ngettext(
1531 "%(option)s option requires %(number)d argument",
1532 "%(option)s option requires %(number)d arguments",
1533 nargs) % {"option": opt, "number": nargs})
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001534 elif nargs == 1:
1535 value = rargs.pop(0)
1536 else:
1537 value = tuple(rargs[0:nargs])
1538 del rargs[0:nargs]
1539
1540 else: # option doesn't take a value
1541 value = None
1542
1543 option.process(opt, value, values, self)
1544
1545 if stop:
1546 break
1547
1548
1549 # -- Feedback methods ----------------------------------------------
1550
Greg Wardeba20e62004-07-31 16:15:44 +00001551 def get_prog_name(self):
1552 if self.prog is None:
1553 return os.path.basename(sys.argv[0])
1554 else:
1555 return self.prog
1556
1557 def expand_prog_name(self, s):
1558 return s.replace("%prog", self.get_prog_name())
1559
1560 def get_description(self):
1561 return self.expand_prog_name(self.description)
1562
Greg Ward48aa84b2004-10-27 02:20:04 +00001563 def exit(self, status=0, msg=None):
1564 if msg:
1565 sys.stderr.write(msg)
1566 sys.exit(status)
1567
Greg Wardeba20e62004-07-31 16:15:44 +00001568 def error(self, msg):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001569 """error(msg : string)
1570
1571 Print a usage message incorporating 'msg' to stderr and exit.
1572 If you override this in a subclass, it should not return -- it
1573 should either exit or raise an exception.
1574 """
1575 self.print_usage(sys.stderr)
Greg Ward48aa84b2004-10-27 02:20:04 +00001576 self.exit(2, "%s: error: %s\n" % (self.get_prog_name(), msg))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001577
Greg Wardeba20e62004-07-31 16:15:44 +00001578 def get_usage(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001579 if self.usage:
1580 return self.formatter.format_usage(
Greg Wardeba20e62004-07-31 16:15:44 +00001581 self.expand_prog_name(self.usage))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001582 else:
1583 return ""
1584
Greg Wardeba20e62004-07-31 16:15:44 +00001585 def print_usage(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001586 """print_usage(file : file = stdout)
1587
1588 Print the usage message for the current program (self.usage) to
Mark Dickinson934896d2009-02-21 20:59:32 +00001589 'file' (default stdout). Any occurrence of the string "%prog" in
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001590 self.usage is replaced with the name of the current program
1591 (basename of sys.argv[0]). Does nothing if self.usage is empty
1592 or not defined.
1593 """
1594 if self.usage:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001595 print(self.get_usage(), file=file)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001596
Greg Wardeba20e62004-07-31 16:15:44 +00001597 def get_version(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001598 if self.version:
Greg Wardeba20e62004-07-31 16:15:44 +00001599 return self.expand_prog_name(self.version)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001600 else:
1601 return ""
1602
Greg Wardeba20e62004-07-31 16:15:44 +00001603 def print_version(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001604 """print_version(file : file = stdout)
1605
1606 Print the version message for this program (self.version) to
Mark Dickinson934896d2009-02-21 20:59:32 +00001607 'file' (default stdout). As with print_usage(), any occurrence
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001608 of "%prog" in self.version is replaced by the current program's
1609 name. Does nothing if self.version is empty or undefined.
1610 """
1611 if self.version:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001612 print(self.get_version(), file=file)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001613
Greg Wardeba20e62004-07-31 16:15:44 +00001614 def format_option_help(self, formatter=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001615 if formatter is None:
1616 formatter = self.formatter
1617 formatter.store_option_strings(self)
1618 result = []
Thomas Wouters477c8d52006-05-27 19:21:47 +00001619 result.append(formatter.format_heading(_("Options")))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001620 formatter.indent()
1621 if self.option_list:
1622 result.append(OptionContainer.format_option_help(self, formatter))
1623 result.append("\n")
1624 for group in self.option_groups:
1625 result.append(group.format_help(formatter))
1626 result.append("\n")
1627 formatter.dedent()
1628 # Drop the last "\n", or the header if no options or option groups:
1629 return "".join(result[:-1])
1630
Thomas Wouters477c8d52006-05-27 19:21:47 +00001631 def format_epilog(self, formatter):
1632 return formatter.format_epilog(self.epilog)
1633
Greg Wardeba20e62004-07-31 16:15:44 +00001634 def format_help(self, formatter=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001635 if formatter is None:
1636 formatter = self.formatter
1637 result = []
1638 if self.usage:
1639 result.append(self.get_usage() + "\n")
1640 if self.description:
1641 result.append(self.format_description(formatter) + "\n")
1642 result.append(self.format_option_help(formatter))
Thomas Wouters477c8d52006-05-27 19:21:47 +00001643 result.append(self.format_epilog(formatter))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001644 return "".join(result)
1645
Greg Wardeba20e62004-07-31 16:15:44 +00001646 def print_help(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001647 """print_help(file : file = stdout)
1648
1649 Print an extended help message, listing all options and any
1650 help text provided with them, to 'file' (default stdout).
1651 """
1652 if file is None:
1653 file = sys.stdout
Guido van Rossum34d19282007-08-09 01:03:29 +00001654 file.write(self.format_help())
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001655
1656# class OptionParser
1657
1658
Greg Wardeba20e62004-07-31 16:15:44 +00001659def _match_abbrev(s, wordmap):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001660 """_match_abbrev(s : string, wordmap : {string : Option}) -> string
1661
1662 Return the string key in 'wordmap' for which 's' is an unambiguous
1663 abbreviation. If 's' is found to be ambiguous or doesn't match any of
1664 'words', raise BadOptionError.
1665 """
1666 # Is there an exact match?
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001667 if s in wordmap:
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001668 return s
1669 else:
1670 # Isolate all words with s as a prefix.
1671 possibilities = [word for word in wordmap.keys()
1672 if word.startswith(s)]
1673 # No exact match, so there had better be just one possibility.
1674 if len(possibilities) == 1:
1675 return possibilities[0]
1676 elif not possibilities:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001677 raise BadOptionError(s)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001678 else:
1679 # More than one possible completion: ambiguous prefix.
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001680 possibilities.sort()
Thomas Wouters477c8d52006-05-27 19:21:47 +00001681 raise AmbiguousOptionError(s, possibilities)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001682
1683
1684# Some day, there might be many Option classes. As of Optik 1.3, the
1685# preferred way to instantiate Options is indirectly, via make_option(),
1686# which will become a factory function when there are many Option
1687# classes.
1688make_option = Option