blob: ab2ce40077e336fcc51648715f778fe046811059 [file] [log] [blame]
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001"""optparse - a powerful, extensible, and easy-to-use option parser.
2
3By Greg Ward <gward@python.net>
4
Greg Ward2492fcf2003-04-21 02:40:34 +00005Originally distributed as Optik; see http://optik.sourceforge.net/ .
Guido van Rossumb9ba4582002-11-14 22:00:19 +00006
Greg Ward4656ed42003-05-08 01:38:52 +00007If you have problems with this module, please do not file bugs,
Greg Ward2492fcf2003-04-21 02:40:34 +00008patches, or feature requests with Python; instead, use Optik's
9SourceForge project page:
10 http://sourceforge.net/projects/optik
11
12For support, use the optik-users@lists.sourceforge.net mailing list
13(http://lists.sourceforge.net/lists/listinfo/optik-users).
Guido van Rossumb9ba4582002-11-14 22:00:19 +000014"""
15
Greg Ward2492fcf2003-04-21 02:40:34 +000016# Python developers: please do not make changes to this file, since
17# it is automatically generated from the Optik source code.
18
Thomas Wouters0e3f5912006-08-11 14:57:12 +000019__version__ = "1.5.3"
Greg Ward2492fcf2003-04-21 02:40:34 +000020
Greg Ward4656ed42003-05-08 01:38:52 +000021__all__ = ['Option',
22 'SUPPRESS_HELP',
23 'SUPPRESS_USAGE',
Greg Ward4656ed42003-05-08 01:38:52 +000024 'Values',
25 'OptionContainer',
26 'OptionGroup',
27 'OptionParser',
28 'HelpFormatter',
29 'IndentedHelpFormatter',
30 'TitledHelpFormatter',
31 'OptParseError',
32 'OptionError',
33 'OptionConflictError',
34 'OptionValueError',
35 'BadOptionError']
Greg Ward2492fcf2003-04-21 02:40:34 +000036
Guido van Rossumb9ba4582002-11-14 22:00:19 +000037__copyright__ = """
Thomas Wouters477c8d52006-05-27 19:21:47 +000038Copyright (c) 2001-2006 Gregory P. Ward. All rights reserved.
39Copyright (c) 2002-2006 Python Software Foundation. All rights reserved.
Guido van Rossumb9ba4582002-11-14 22:00:19 +000040
41Redistribution and use in source and binary forms, with or without
42modification, are permitted provided that the following conditions are
43met:
44
45 * Redistributions of source code must retain the above copyright
46 notice, this list of conditions and the following disclaimer.
47
48 * Redistributions in binary form must reproduce the above copyright
49 notice, this list of conditions and the following disclaimer in the
50 documentation and/or other materials provided with the distribution.
51
52 * Neither the name of the author nor the names of its
53 contributors may be used to endorse or promote products derived from
54 this software without specific prior written permission.
55
56THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
57IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
58TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
59PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR
60CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
61EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
62PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
63PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
64LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
65NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
66SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
67"""
68
69import sys, os
Guido van Rossumb9ba4582002-11-14 22:00:19 +000070import textwrap
Greg Wardeba20e62004-07-31 16:15:44 +000071
72def _repr(self):
73 return "<%s at 0x%x: %s>" % (self.__class__.__name__, id(self), self)
74
75
76# This file was generated from:
Thomas Wouters0e3f5912006-08-11 14:57:12 +000077# Id: option_parser.py 527 2006-07-23 15:21:30Z greg
78# Id: option.py 522 2006-06-11 16:22:03Z gward
79# Id: help.py 527 2006-07-23 15:21:30Z greg
Thomas Wouters477c8d52006-05-27 19:21:47 +000080# Id: errors.py 509 2006-04-20 00:58:24Z gward
81
82try:
83 from gettext import gettext
84except ImportError:
85 def gettext(message):
86 return message
87_ = gettext
88
Guido van Rossumb9ba4582002-11-14 22:00:19 +000089
Guido van Rossumb9ba4582002-11-14 22:00:19 +000090class OptParseError (Exception):
Greg Wardeba20e62004-07-31 16:15:44 +000091 def __init__(self, msg):
Guido van Rossumb9ba4582002-11-14 22:00:19 +000092 self.msg = msg
93
Greg Wardeba20e62004-07-31 16:15:44 +000094 def __str__(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +000095 return self.msg
96
Greg Ward2492fcf2003-04-21 02:40:34 +000097
Guido van Rossumb9ba4582002-11-14 22:00:19 +000098class OptionError (OptParseError):
99 """
100 Raised if an Option instance is created with invalid or
101 inconsistent arguments.
102 """
103
Greg Wardeba20e62004-07-31 16:15:44 +0000104 def __init__(self, msg, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000105 self.msg = msg
106 self.option_id = str(option)
107
Greg Wardeba20e62004-07-31 16:15:44 +0000108 def __str__(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000109 if self.option_id:
110 return "option %s: %s" % (self.option_id, self.msg)
111 else:
112 return self.msg
113
114class OptionConflictError (OptionError):
115 """
116 Raised if conflicting options are added to an OptionParser.
117 """
118
119class OptionValueError (OptParseError):
120 """
121 Raised if an invalid option value is encountered on the command
122 line.
123 """
124
125class BadOptionError (OptParseError):
126 """
Thomas Wouters477c8d52006-05-27 19:21:47 +0000127 Raised if an invalid option is seen on the command line.
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000128 """
Thomas Wouters477c8d52006-05-27 19:21:47 +0000129 def __init__(self, opt_str):
130 self.opt_str = opt_str
131
132 def __str__(self):
133 return _("no such option: %s") % self.opt_str
134
135class AmbiguousOptionError (BadOptionError):
136 """
137 Raised if an ambiguous option is seen on the command line.
138 """
139 def __init__(self, opt_str, possibilities):
140 BadOptionError.__init__(self, opt_str)
141 self.possibilities = possibilities
142
143 def __str__(self):
144 return (_("ambiguous option: %s (%s?)")
145 % (self.opt_str, ", ".join(self.possibilities)))
Greg Ward2492fcf2003-04-21 02:40:34 +0000146
147
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000148class HelpFormatter:
149
150 """
151 Abstract base class for formatting option help. OptionParser
152 instances should use one of the HelpFormatter subclasses for
153 formatting help; by default IndentedHelpFormatter is used.
154
155 Instance attributes:
Greg Wardeba20e62004-07-31 16:15:44 +0000156 parser : OptionParser
157 the controlling OptionParser instance
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000158 indent_increment : int
159 the number of columns to indent per nesting level
160 max_help_position : int
161 the maximum starting column for option help text
162 help_position : int
163 the calculated starting column for option help text;
164 initially the same as the maximum
165 width : int
Greg Wardeba20e62004-07-31 16:15:44 +0000166 total number of columns for output (pass None to constructor for
167 this value to be taken from the $COLUMNS environment variable)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000168 level : int
169 current indentation level
170 current_indent : int
171 current indentation level (in columns)
172 help_width : int
173 number of columns available for option help text (calculated)
Greg Wardeba20e62004-07-31 16:15:44 +0000174 default_tag : str
175 text to replace with each option's default value, "%default"
176 by default. Set to false value to disable default value expansion.
177 option_strings : { Option : str }
178 maps Option instances to the snippet of help text explaining
179 the syntax of that option, e.g. "-h, --help" or
180 "-fFILE, --file=FILE"
181 _short_opt_fmt : str
182 format string controlling how short options with values are
183 printed in help text. Must be either "%s%s" ("-fFILE") or
184 "%s %s" ("-f FILE"), because those are the two syntaxes that
185 Optik supports.
186 _long_opt_fmt : str
187 similar but for long options; must be either "%s %s" ("--file FILE")
188 or "%s=%s" ("--file=FILE").
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000189 """
190
Greg Wardeba20e62004-07-31 16:15:44 +0000191 NO_DEFAULT_VALUE = "none"
192
193 def __init__(self,
194 indent_increment,
195 max_help_position,
196 width,
197 short_first):
198 self.parser = None
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000199 self.indent_increment = indent_increment
200 self.help_position = self.max_help_position = max_help_position
Greg Wardeba20e62004-07-31 16:15:44 +0000201 if width is None:
202 try:
203 width = int(os.environ['COLUMNS'])
204 except (KeyError, ValueError):
205 width = 80
206 width -= 2
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000207 self.width = width
208 self.current_indent = 0
209 self.level = 0
Greg Wardeba20e62004-07-31 16:15:44 +0000210 self.help_width = None # computed later
Greg Ward2492fcf2003-04-21 02:40:34 +0000211 self.short_first = short_first
Greg Wardeba20e62004-07-31 16:15:44 +0000212 self.default_tag = "%default"
213 self.option_strings = {}
214 self._short_opt_fmt = "%s %s"
215 self._long_opt_fmt = "%s=%s"
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000216
Greg Wardeba20e62004-07-31 16:15:44 +0000217 def set_parser(self, parser):
218 self.parser = parser
219
220 def set_short_opt_delimiter(self, delim):
221 if delim not in ("", " "):
222 raise ValueError(
223 "invalid metavar delimiter for short options: %r" % delim)
224 self._short_opt_fmt = "%s" + delim + "%s"
225
226 def set_long_opt_delimiter(self, delim):
227 if delim not in ("=", " "):
228 raise ValueError(
229 "invalid metavar delimiter for long options: %r" % delim)
230 self._long_opt_fmt = "%s" + delim + "%s"
231
232 def indent(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000233 self.current_indent += self.indent_increment
234 self.level += 1
235
Greg Wardeba20e62004-07-31 16:15:44 +0000236 def dedent(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000237 self.current_indent -= self.indent_increment
238 assert self.current_indent >= 0, "Indent decreased below 0."
239 self.level -= 1
240
Greg Wardeba20e62004-07-31 16:15:44 +0000241 def format_usage(self, usage):
Collin Winterce36ad82007-08-30 01:19:48 +0000242 raise NotImplementedError("subclasses must implement")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000243
Greg Wardeba20e62004-07-31 16:15:44 +0000244 def format_heading(self, heading):
Collin Winterce36ad82007-08-30 01:19:48 +0000245 raise NotImplementedError("subclasses must implement")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000246
Thomas Wouters477c8d52006-05-27 19:21:47 +0000247 def _format_text(self, text):
248 """
249 Format a paragraph of free-form text for inclusion in the
250 help output at the current indentation level.
251 """
252 text_width = self.width - self.current_indent
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000253 indent = " "*self.current_indent
Thomas Wouters477c8d52006-05-27 19:21:47 +0000254 return textwrap.fill(text,
255 text_width,
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000256 initial_indent=indent,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000257 subsequent_indent=indent)
258
259 def format_description(self, description):
260 if description:
261 return self._format_text(description) + "\n"
262 else:
263 return ""
264
265 def format_epilog(self, epilog):
266 if epilog:
267 return "\n" + self._format_text(epilog) + "\n"
268 else:
269 return ""
270
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000271
Greg Wardeba20e62004-07-31 16:15:44 +0000272 def expand_default(self, option):
273 if self.parser is None or not self.default_tag:
274 return option.help
275
276 default_value = self.parser.defaults.get(option.dest)
277 if default_value is NO_DEFAULT or default_value is None:
278 default_value = self.NO_DEFAULT_VALUE
279
280 return option.help.replace(self.default_tag, str(default_value))
281
282 def format_option(self, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000283 # The help for each option consists of two parts:
284 # * the opt strings and metavars
285 # eg. ("-x", or "-fFILENAME, --file=FILENAME")
286 # * the user-supplied help string
287 # eg. ("turn on expert mode", "read data from FILENAME")
288 #
289 # If possible, we write both of these on the same line:
290 # -x turn on expert mode
291 #
292 # But if the opt string list is too long, we put the help
293 # string on a second line, indented to the same column it would
294 # start in if it fit on the first line.
295 # -fFILENAME, --file=FILENAME
296 # read data from FILENAME
297 result = []
Greg Wardeba20e62004-07-31 16:15:44 +0000298 opts = self.option_strings[option]
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000299 opt_width = self.help_position - self.current_indent - 2
300 if len(opts) > opt_width:
301 opts = "%*s%s\n" % (self.current_indent, "", opts)
302 indent_first = self.help_position
303 else: # start help on same line as opts
304 opts = "%*s%-*s " % (self.current_indent, "", opt_width, opts)
305 indent_first = 0
306 result.append(opts)
307 if option.help:
Greg Wardeba20e62004-07-31 16:15:44 +0000308 help_text = self.expand_default(option)
309 help_lines = textwrap.wrap(help_text, self.help_width)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000310 result.append("%*s%s\n" % (indent_first, "", help_lines[0]))
311 result.extend(["%*s%s\n" % (self.help_position, "", line)
312 for line in help_lines[1:]])
313 elif opts[-1] != "\n":
314 result.append("\n")
315 return "".join(result)
316
Greg Wardeba20e62004-07-31 16:15:44 +0000317 def store_option_strings(self, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000318 self.indent()
319 max_len = 0
320 for opt in parser.option_list:
321 strings = self.format_option_strings(opt)
Greg Wardeba20e62004-07-31 16:15:44 +0000322 self.option_strings[opt] = strings
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000323 max_len = max(max_len, len(strings) + self.current_indent)
324 self.indent()
325 for group in parser.option_groups:
326 for opt in group.option_list:
327 strings = self.format_option_strings(opt)
Greg Wardeba20e62004-07-31 16:15:44 +0000328 self.option_strings[opt] = strings
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000329 max_len = max(max_len, len(strings) + self.current_indent)
330 self.dedent()
331 self.dedent()
332 self.help_position = min(max_len + 2, self.max_help_position)
Greg Wardeba20e62004-07-31 16:15:44 +0000333 self.help_width = self.width - self.help_position
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000334
Greg Wardeba20e62004-07-31 16:15:44 +0000335 def format_option_strings(self, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000336 """Return a comma-separated list of option strings & metavariables."""
Greg Ward2492fcf2003-04-21 02:40:34 +0000337 if option.takes_value():
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000338 metavar = option.metavar or option.dest.upper()
Greg Wardeba20e62004-07-31 16:15:44 +0000339 short_opts = [self._short_opt_fmt % (sopt, metavar)
340 for sopt in option._short_opts]
341 long_opts = [self._long_opt_fmt % (lopt, metavar)
342 for lopt in option._long_opts]
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000343 else:
Greg Ward2492fcf2003-04-21 02:40:34 +0000344 short_opts = option._short_opts
345 long_opts = option._long_opts
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000346
Greg Ward2492fcf2003-04-21 02:40:34 +0000347 if self.short_first:
348 opts = short_opts + long_opts
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000349 else:
Greg Ward2492fcf2003-04-21 02:40:34 +0000350 opts = long_opts + short_opts
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000351
Greg Ward2492fcf2003-04-21 02:40:34 +0000352 return ", ".join(opts)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000353
354class IndentedHelpFormatter (HelpFormatter):
355 """Format help with indented section bodies.
356 """
357
Greg Wardeba20e62004-07-31 16:15:44 +0000358 def __init__(self,
359 indent_increment=2,
360 max_help_position=24,
361 width=None,
362 short_first=1):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000363 HelpFormatter.__init__(
364 self, indent_increment, max_help_position, width, short_first)
365
Greg Wardeba20e62004-07-31 16:15:44 +0000366 def format_usage(self, usage):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000367 return _("Usage: %s\n") % usage
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000368
Greg Wardeba20e62004-07-31 16:15:44 +0000369 def format_heading(self, heading):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000370 return "%*s%s:\n" % (self.current_indent, "", heading)
371
372
373class TitledHelpFormatter (HelpFormatter):
374 """Format help with underlined section headers.
375 """
376
Greg Wardeba20e62004-07-31 16:15:44 +0000377 def __init__(self,
378 indent_increment=0,
379 max_help_position=24,
380 width=None,
381 short_first=0):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000382 HelpFormatter.__init__ (
383 self, indent_increment, max_help_position, width, short_first)
384
Greg Wardeba20e62004-07-31 16:15:44 +0000385 def format_usage(self, usage):
386 return "%s %s\n" % (self.format_heading(_("Usage")), usage)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000387
Greg Wardeba20e62004-07-31 16:15:44 +0000388 def format_heading(self, heading):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000389 return "%s\n%s\n" % (heading, "=-"[self.level] * len(heading))
Greg Ward2492fcf2003-04-21 02:40:34 +0000390
391
Thomas Wouters477c8d52006-05-27 19:21:47 +0000392def _parse_num(val, type):
393 if val[:2].lower() == "0x": # hexadecimal
394 radix = 16
395 elif val[:2].lower() == "0b": # binary
396 radix = 2
397 val = val[2:] or "0" # have to remove "0b" prefix
398 elif val[:1] == "0": # octal
399 radix = 8
400 else: # decimal
401 radix = 10
402
403 return type(val, radix)
404
405def _parse_int(val):
406 return _parse_num(val, int)
407
408def _parse_long(val):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000409 return _parse_num(val, int)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000410
411_builtin_cvt = { "int" : (_parse_int, _("integer")),
412 "long" : (_parse_long, _("long integer")),
Greg Wardeba20e62004-07-31 16:15:44 +0000413 "float" : (float, _("floating-point")),
414 "complex" : (complex, _("complex")) }
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000415
Greg Wardeba20e62004-07-31 16:15:44 +0000416def check_builtin(option, opt, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000417 (cvt, what) = _builtin_cvt[option.type]
418 try:
419 return cvt(value)
420 except ValueError:
421 raise OptionValueError(
Greg Wardeba20e62004-07-31 16:15:44 +0000422 _("option %s: invalid %s value: %r") % (opt, what, value))
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000423
424def check_choice(option, opt, value):
425 if value in option.choices:
426 return value
427 else:
428 choices = ", ".join(map(repr, option.choices))
429 raise OptionValueError(
Greg Wardeba20e62004-07-31 16:15:44 +0000430 _("option %s: invalid choice: %r (choose from %s)")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000431 % (opt, value, choices))
432
433# Not supplying a default is different from a default of None,
434# so we need an explicit "not supplied" value.
Greg Wardeba20e62004-07-31 16:15:44 +0000435NO_DEFAULT = ("NO", "DEFAULT")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000436
437
438class Option:
439 """
440 Instance attributes:
441 _short_opts : [string]
442 _long_opts : [string]
443
444 action : string
445 type : string
446 dest : string
447 default : any
448 nargs : int
449 const : any
450 choices : [string]
451 callback : function
452 callback_args : (any*)
453 callback_kwargs : { string : any }
454 help : string
455 metavar : string
456 """
457
458 # The list of instance attributes that may be set through
459 # keyword args to the constructor.
460 ATTRS = ['action',
461 'type',
462 'dest',
463 'default',
464 'nargs',
465 'const',
466 'choices',
467 'callback',
468 'callback_args',
469 'callback_kwargs',
470 'help',
471 'metavar']
472
473 # The set of actions allowed by option parsers. Explicitly listed
474 # here so the constructor can validate its arguments.
475 ACTIONS = ("store",
476 "store_const",
477 "store_true",
478 "store_false",
479 "append",
Thomas Wouters477c8d52006-05-27 19:21:47 +0000480 "append_const",
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000481 "count",
482 "callback",
483 "help",
484 "version")
485
486 # The set of actions that involve storing a value somewhere;
487 # also listed just for constructor argument validation. (If
488 # the action is one of these, there must be a destination.)
489 STORE_ACTIONS = ("store",
490 "store_const",
491 "store_true",
492 "store_false",
493 "append",
Thomas Wouters477c8d52006-05-27 19:21:47 +0000494 "append_const",
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000495 "count")
496
497 # The set of actions for which it makes sense to supply a value
Greg Ward48aa84b2004-10-27 02:20:04 +0000498 # type, ie. which may consume an argument from the command line.
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000499 TYPED_ACTIONS = ("store",
500 "append",
501 "callback")
502
Greg Ward48aa84b2004-10-27 02:20:04 +0000503 # The set of actions which *require* a value type, ie. that
504 # always consume an argument from the command line.
505 ALWAYS_TYPED_ACTIONS = ("store",
506 "append")
507
Thomas Wouters477c8d52006-05-27 19:21:47 +0000508 # The set of actions which take a 'const' attribute.
509 CONST_ACTIONS = ("store_const",
510 "append_const")
511
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000512 # The set of known types for option parsers. Again, listed here for
513 # constructor argument validation.
514 TYPES = ("string", "int", "long", "float", "complex", "choice")
515
516 # Dictionary of argument checking functions, which convert and
517 # validate option arguments according to the option type.
518 #
519 # Signature of checking functions is:
520 # check(option : Option, opt : string, value : string) -> any
521 # where
522 # option is the Option instance calling the checker
523 # opt is the actual option seen on the command-line
524 # (eg. "-a", "--file")
525 # value is the option argument seen on the command-line
526 #
527 # The return value should be in the appropriate Python type
528 # for option.type -- eg. an integer if option.type == "int".
529 #
530 # If no checker is defined for a type, arguments will be
531 # unchecked and remain strings.
532 TYPE_CHECKER = { "int" : check_builtin,
533 "long" : check_builtin,
534 "float" : check_builtin,
Greg Wardeba20e62004-07-31 16:15:44 +0000535 "complex": check_builtin,
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000536 "choice" : check_choice,
537 }
538
539
540 # CHECK_METHODS is a list of unbound method objects; they are called
541 # by the constructor, in order, after all attributes are
542 # initialized. The list is created and filled in later, after all
543 # the methods are actually defined. (I just put it here because I
544 # like to define and document all class attributes in the same
545 # place.) Subclasses that add another _check_*() method should
546 # define their own CHECK_METHODS list that adds their check method
547 # to those from this class.
548 CHECK_METHODS = None
549
550
551 # -- Constructor/initialization methods ----------------------------
552
Greg Wardeba20e62004-07-31 16:15:44 +0000553 def __init__(self, *opts, **attrs):
Greg Ward2492fcf2003-04-21 02:40:34 +0000554 # Set _short_opts, _long_opts attrs from 'opts' tuple.
555 # Have to be set now, in case no option strings are supplied.
556 self._short_opts = []
557 self._long_opts = []
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000558 opts = self._check_opt_strings(opts)
559 self._set_opt_strings(opts)
560
561 # Set all other attrs (action, type, etc.) from 'attrs' dict
562 self._set_attrs(attrs)
563
564 # Check all the attributes we just set. There are lots of
565 # complicated interdependencies, but luckily they can be farmed
566 # out to the _check_*() methods listed in CHECK_METHODS -- which
567 # could be handy for subclasses! The one thing these all share
568 # is that they raise OptionError if they discover a problem.
569 for checker in self.CHECK_METHODS:
570 checker(self)
571
Greg Wardeba20e62004-07-31 16:15:44 +0000572 def _check_opt_strings(self, opts):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000573 # Filter out None because early versions of Optik had exactly
574 # one short option and one long option, either of which
575 # could be None.
Guido van Rossumc1f779c2007-07-03 08:25:58 +0000576 opts = [opt for opt in opts if opt]
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000577 if not opts:
Greg Ward2492fcf2003-04-21 02:40:34 +0000578 raise TypeError("at least one option string must be supplied")
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000579 return opts
580
Greg Wardeba20e62004-07-31 16:15:44 +0000581 def _set_opt_strings(self, opts):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000582 for opt in opts:
583 if len(opt) < 2:
584 raise OptionError(
585 "invalid option string %r: "
586 "must be at least two characters long" % opt, self)
587 elif len(opt) == 2:
588 if not (opt[0] == "-" and opt[1] != "-"):
589 raise OptionError(
590 "invalid short option string %r: "
591 "must be of the form -x, (x any non-dash char)" % opt,
592 self)
593 self._short_opts.append(opt)
594 else:
595 if not (opt[0:2] == "--" and opt[2] != "-"):
596 raise OptionError(
597 "invalid long option string %r: "
598 "must start with --, followed by non-dash" % opt,
599 self)
600 self._long_opts.append(opt)
601
Greg Wardeba20e62004-07-31 16:15:44 +0000602 def _set_attrs(self, attrs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000603 for attr in self.ATTRS:
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000604 if attr in attrs:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000605 setattr(self, attr, attrs[attr])
606 del attrs[attr]
607 else:
608 if attr == 'default':
609 setattr(self, attr, NO_DEFAULT)
610 else:
611 setattr(self, attr, None)
612 if attrs:
Georg Brandlc2d9d7f2007-02-11 23:06:17 +0000613 attrs = sorted(attrs.keys())
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000614 raise OptionError(
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000615 "invalid keyword arguments: %s" % ", ".join(attrs),
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000616 self)
617
618
619 # -- Constructor validation methods --------------------------------
620
Greg Wardeba20e62004-07-31 16:15:44 +0000621 def _check_action(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000622 if self.action is None:
623 self.action = "store"
624 elif self.action not in self.ACTIONS:
625 raise OptionError("invalid action: %r" % self.action, self)
626
Greg Wardeba20e62004-07-31 16:15:44 +0000627 def _check_type(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000628 if self.type is None:
Greg Ward48aa84b2004-10-27 02:20:04 +0000629 if self.action in self.ALWAYS_TYPED_ACTIONS:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000630 if self.choices is not None:
631 # The "choices" attribute implies "choice" type.
632 self.type = "choice"
633 else:
634 # No type given? "string" is the most sensible default.
635 self.type = "string"
636 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000637 # Allow type objects or builtin type conversion functions
638 # (int, str, etc.) as an alternative to their names. (The
Georg Brandl1a3284e2007-12-02 09:40:06 +0000639 # complicated check of builtins is only necessary for
Thomas Wouters477c8d52006-05-27 19:21:47 +0000640 # Python 2.1 and earlier, and is short-circuited by the
641 # first check on modern Pythons.)
Georg Brandl1a3284e2007-12-02 09:40:06 +0000642 import builtins
Guido van Rossum13257902007-06-07 23:15:56 +0000643 if ( isinstance(self.type, type) or
Thomas Wouters477c8d52006-05-27 19:21:47 +0000644 (hasattr(self.type, "__name__") and
Georg Brandl1a3284e2007-12-02 09:40:06 +0000645 getattr(builtins, self.type.__name__, None) is self.type) ):
Greg Wardeba20e62004-07-31 16:15:44 +0000646 self.type = self.type.__name__
Thomas Wouters477c8d52006-05-27 19:21:47 +0000647
Greg Wardeba20e62004-07-31 16:15:44 +0000648 if self.type == "str":
649 self.type = "string"
650
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000651 if self.type not in self.TYPES:
652 raise OptionError("invalid option type: %r" % self.type, self)
653 if self.action not in self.TYPED_ACTIONS:
654 raise OptionError(
655 "must not supply a type for action %r" % self.action, self)
656
657 def _check_choice(self):
658 if self.type == "choice":
659 if self.choices is None:
660 raise OptionError(
661 "must supply a list of choices for type 'choice'", self)
Guido van Rossum13257902007-06-07 23:15:56 +0000662 elif not isinstance(self.choices, (tuple, list)):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000663 raise OptionError(
664 "choices must be a list of strings ('%s' supplied)"
665 % str(type(self.choices)).split("'")[1], self)
666 elif self.choices is not None:
667 raise OptionError(
668 "must not supply choices for type %r" % self.type, self)
669
Greg Wardeba20e62004-07-31 16:15:44 +0000670 def _check_dest(self):
671 # No destination given, and we need one for this action. The
672 # self.type check is for callbacks that take a value.
673 takes_value = (self.action in self.STORE_ACTIONS or
674 self.type is not None)
675 if self.dest is None and takes_value:
676
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000677 # Glean a destination from the first long option string,
678 # or from the first short option string if no long options.
679 if self._long_opts:
680 # eg. "--foo-bar" -> "foo_bar"
681 self.dest = self._long_opts[0][2:].replace('-', '_')
682 else:
683 self.dest = self._short_opts[0][1]
684
Greg Wardeba20e62004-07-31 16:15:44 +0000685 def _check_const(self):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000686 if self.action not in self.CONST_ACTIONS and self.const is not None:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000687 raise OptionError(
688 "'const' must not be supplied for action %r" % self.action,
689 self)
690
Greg Wardeba20e62004-07-31 16:15:44 +0000691 def _check_nargs(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000692 if self.action in self.TYPED_ACTIONS:
693 if self.nargs is None:
694 self.nargs = 1
695 elif self.nargs is not None:
696 raise OptionError(
697 "'nargs' must not be supplied for action %r" % self.action,
698 self)
699
Greg Wardeba20e62004-07-31 16:15:44 +0000700 def _check_callback(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000701 if self.action == "callback":
Guido van Rossumd59da4b2007-05-22 18:11:13 +0000702 if not hasattr(self.callback, '__call__'):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000703 raise OptionError(
704 "callback not callable: %r" % self.callback, self)
705 if (self.callback_args is not None and
Guido van Rossum13257902007-06-07 23:15:56 +0000706 not isinstance(self.callback_args, tuple)):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000707 raise OptionError(
708 "callback_args, if supplied, must be a tuple: not %r"
709 % self.callback_args, self)
710 if (self.callback_kwargs is not None and
Guido van Rossum13257902007-06-07 23:15:56 +0000711 not isinstance(self.callback_kwargs, dict)):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000712 raise OptionError(
713 "callback_kwargs, if supplied, must be a dict: not %r"
714 % self.callback_kwargs, self)
715 else:
716 if self.callback is not None:
717 raise OptionError(
718 "callback supplied (%r) for non-callback option"
719 % self.callback, self)
720 if self.callback_args is not None:
721 raise OptionError(
722 "callback_args supplied for non-callback option", self)
723 if self.callback_kwargs is not None:
724 raise OptionError(
725 "callback_kwargs supplied for non-callback option", self)
726
727
728 CHECK_METHODS = [_check_action,
729 _check_type,
730 _check_choice,
731 _check_dest,
732 _check_const,
733 _check_nargs,
734 _check_callback]
735
736
737 # -- Miscellaneous methods -----------------------------------------
738
Greg Wardeba20e62004-07-31 16:15:44 +0000739 def __str__(self):
Greg Ward2492fcf2003-04-21 02:40:34 +0000740 return "/".join(self._short_opts + self._long_opts)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000741
Greg Wardeba20e62004-07-31 16:15:44 +0000742 __repr__ = _repr
743
744 def takes_value(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000745 return self.type is not None
746
Greg Wardeba20e62004-07-31 16:15:44 +0000747 def get_opt_string(self):
748 if self._long_opts:
749 return self._long_opts[0]
750 else:
751 return self._short_opts[0]
752
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000753
754 # -- Processing methods --------------------------------------------
755
Greg Wardeba20e62004-07-31 16:15:44 +0000756 def check_value(self, opt, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000757 checker = self.TYPE_CHECKER.get(self.type)
758 if checker is None:
759 return value
760 else:
761 return checker(self, opt, value)
762
Greg Wardeba20e62004-07-31 16:15:44 +0000763 def convert_value(self, opt, value):
764 if value is not None:
765 if self.nargs == 1:
766 return self.check_value(opt, value)
767 else:
768 return tuple([self.check_value(opt, v) for v in value])
769
770 def process(self, opt, value, values, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000771
772 # First, convert the value(s) to the right type. Howl if any
773 # value(s) are bogus.
Greg Wardeba20e62004-07-31 16:15:44 +0000774 value = self.convert_value(opt, value)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000775
776 # And then take whatever action is expected of us.
777 # This is a separate method to make life easier for
778 # subclasses to add new actions.
779 return self.take_action(
780 self.action, self.dest, opt, value, values, parser)
781
Greg Wardeba20e62004-07-31 16:15:44 +0000782 def take_action(self, action, dest, opt, value, values, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000783 if action == "store":
784 setattr(values, dest, value)
785 elif action == "store_const":
786 setattr(values, dest, self.const)
787 elif action == "store_true":
Greg Ward2492fcf2003-04-21 02:40:34 +0000788 setattr(values, dest, True)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000789 elif action == "store_false":
Greg Ward2492fcf2003-04-21 02:40:34 +0000790 setattr(values, dest, False)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000791 elif action == "append":
792 values.ensure_value(dest, []).append(value)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000793 elif action == "append_const":
794 values.ensure_value(dest, []).append(self.const)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000795 elif action == "count":
796 setattr(values, dest, values.ensure_value(dest, 0) + 1)
797 elif action == "callback":
798 args = self.callback_args or ()
799 kwargs = self.callback_kwargs or {}
800 self.callback(self, opt, value, parser, *args, **kwargs)
801 elif action == "help":
802 parser.print_help()
Greg Ward48aa84b2004-10-27 02:20:04 +0000803 parser.exit()
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000804 elif action == "version":
805 parser.print_version()
Greg Ward48aa84b2004-10-27 02:20:04 +0000806 parser.exit()
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000807 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000808 raise RuntimeError("unknown action %r" % self.action)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000809
810 return 1
811
812# class Option
Greg Ward2492fcf2003-04-21 02:40:34 +0000813
814
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000815SUPPRESS_HELP = "SUPPRESS"+"HELP"
816SUPPRESS_USAGE = "SUPPRESS"+"USAGE"
817
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000818class Values:
819
Greg Wardeba20e62004-07-31 16:15:44 +0000820 def __init__(self, defaults=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000821 if defaults:
822 for (attr, val) in defaults.items():
823 setattr(self, attr, val)
824
Greg Wardeba20e62004-07-31 16:15:44 +0000825 def __str__(self):
826 return str(self.__dict__)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000827
Greg Wardeba20e62004-07-31 16:15:44 +0000828 __repr__ = _repr
829
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000830 def __eq__(self, other):
Greg Wardeba20e62004-07-31 16:15:44 +0000831 if isinstance(other, Values):
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000832 return self.__dict__ == other.__dict__
Guido van Rossum13257902007-06-07 23:15:56 +0000833 elif isinstance(other, dict):
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000834 return self.__dict__ == other
Greg Wardeba20e62004-07-31 16:15:44 +0000835 else:
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000836 return NotImplemented
Greg Wardeba20e62004-07-31 16:15:44 +0000837
838 def _update_careful(self, dict):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000839 """
840 Update the option values from an arbitrary dictionary, but only
841 use keys from dict that already have a corresponding attribute
842 in self. Any keys in dict without a corresponding attribute
843 are silently ignored.
844 """
845 for attr in dir(self):
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000846 if attr in dict:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000847 dval = dict[attr]
848 if dval is not None:
849 setattr(self, attr, dval)
850
Greg Wardeba20e62004-07-31 16:15:44 +0000851 def _update_loose(self, dict):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000852 """
853 Update the option values from an arbitrary dictionary,
854 using all keys from the dictionary regardless of whether
855 they have a corresponding attribute in self or not.
856 """
857 self.__dict__.update(dict)
858
Greg Wardeba20e62004-07-31 16:15:44 +0000859 def _update(self, dict, mode):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000860 if mode == "careful":
861 self._update_careful(dict)
862 elif mode == "loose":
863 self._update_loose(dict)
864 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000865 raise ValueError("invalid update mode: %r" % mode)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000866
Greg Wardeba20e62004-07-31 16:15:44 +0000867 def read_module(self, modname, mode="careful"):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000868 __import__(modname)
869 mod = sys.modules[modname]
870 self._update(vars(mod), mode)
871
Greg Wardeba20e62004-07-31 16:15:44 +0000872 def read_file(self, filename, mode="careful"):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000873 vars = {}
Neal Norwitz01688022007-08-12 00:43:29 +0000874 exec(open(filename).read(), vars)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000875 self._update(vars, mode)
876
Greg Wardeba20e62004-07-31 16:15:44 +0000877 def ensure_value(self, attr, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000878 if not hasattr(self, attr) or getattr(self, attr) is None:
879 setattr(self, attr, value)
880 return getattr(self, attr)
881
882
883class OptionContainer:
884
885 """
886 Abstract base class.
887
888 Class attributes:
889 standard_option_list : [Option]
890 list of standard options that will be accepted by all instances
891 of this parser class (intended to be overridden by subclasses).
892
893 Instance attributes:
894 option_list : [Option]
895 the list of Option objects contained by this OptionContainer
896 _short_opt : { string : Option }
897 dictionary mapping short option strings, eg. "-f" or "-X",
898 to the Option instances that implement them. If an Option
899 has multiple short option strings, it will appears in this
900 dictionary multiple times. [1]
901 _long_opt : { string : Option }
902 dictionary mapping long option strings, eg. "--file" or
903 "--exclude", to the Option instances that implement them.
904 Again, a given Option can occur multiple times in this
905 dictionary. [1]
906 defaults : { string : any }
907 dictionary mapping option destination names to default
908 values for each destination [1]
909
910 [1] These mappings are common to (shared by) all components of the
911 controlling OptionParser, where they are initially created.
912
913 """
914
Greg Wardeba20e62004-07-31 16:15:44 +0000915 def __init__(self, option_class, conflict_handler, description):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000916 # Initialize the option list and related data structures.
917 # This method must be provided by subclasses, and it must
918 # initialize at least the following instance attributes:
919 # option_list, _short_opt, _long_opt, defaults.
920 self._create_option_list()
921
922 self.option_class = option_class
923 self.set_conflict_handler(conflict_handler)
924 self.set_description(description)
925
Greg Wardeba20e62004-07-31 16:15:44 +0000926 def _create_option_mappings(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000927 # For use by OptionParser constructor -- create the master
928 # option mappings used by this OptionParser and all
929 # OptionGroups that it owns.
930 self._short_opt = {} # single letter -> Option instance
931 self._long_opt = {} # long option -> Option instance
932 self.defaults = {} # maps option dest -> default value
933
934
Greg Wardeba20e62004-07-31 16:15:44 +0000935 def _share_option_mappings(self, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000936 # For use by OptionGroup constructor -- use shared option
937 # mappings from the OptionParser that owns this OptionGroup.
938 self._short_opt = parser._short_opt
939 self._long_opt = parser._long_opt
940 self.defaults = parser.defaults
941
Greg Wardeba20e62004-07-31 16:15:44 +0000942 def set_conflict_handler(self, handler):
Greg Ward48aa84b2004-10-27 02:20:04 +0000943 if handler not in ("error", "resolve"):
Collin Winterce36ad82007-08-30 01:19:48 +0000944 raise ValueError("invalid conflict_resolution value %r" % handler)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000945 self.conflict_handler = handler
946
Greg Wardeba20e62004-07-31 16:15:44 +0000947 def set_description(self, description):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000948 self.description = description
949
Greg Wardeba20e62004-07-31 16:15:44 +0000950 def get_description(self):
951 return self.description
952
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000953
Thomas Wouters477c8d52006-05-27 19:21:47 +0000954 def destroy(self):
955 """see OptionParser.destroy()."""
956 del self._short_opt
957 del self._long_opt
958 del self.defaults
959
960
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000961 # -- Option-adding methods -----------------------------------------
962
Greg Wardeba20e62004-07-31 16:15:44 +0000963 def _check_conflict(self, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000964 conflict_opts = []
965 for opt in option._short_opts:
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000966 if opt in self._short_opt:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000967 conflict_opts.append((opt, self._short_opt[opt]))
968 for opt in option._long_opts:
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000969 if opt in self._long_opt:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000970 conflict_opts.append((opt, self._long_opt[opt]))
971
972 if conflict_opts:
973 handler = self.conflict_handler
Greg Ward48aa84b2004-10-27 02:20:04 +0000974 if handler == "error":
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000975 raise OptionConflictError(
976 "conflicting option string(s): %s"
977 % ", ".join([co[0] for co in conflict_opts]),
978 option)
Greg Ward48aa84b2004-10-27 02:20:04 +0000979 elif handler == "resolve":
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000980 for (opt, c_option) in conflict_opts:
981 if opt.startswith("--"):
982 c_option._long_opts.remove(opt)
983 del self._long_opt[opt]
984 else:
985 c_option._short_opts.remove(opt)
986 del self._short_opt[opt]
987 if not (c_option._short_opts or c_option._long_opts):
988 c_option.container.option_list.remove(c_option)
989
Greg Wardeba20e62004-07-31 16:15:44 +0000990 def add_option(self, *args, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000991 """add_option(Option)
992 add_option(opt_str, ..., kwarg=val, ...)
993 """
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000994 if isinstance(args[0], str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000995 option = self.option_class(*args, **kwargs)
996 elif len(args) == 1 and not kwargs:
997 option = args[0]
998 if not isinstance(option, Option):
Collin Winterce36ad82007-08-30 01:19:48 +0000999 raise TypeError("not an Option instance: %r" % option)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001000 else:
Collin Winterce36ad82007-08-30 01:19:48 +00001001 raise TypeError("invalid arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001002
1003 self._check_conflict(option)
1004
1005 self.option_list.append(option)
1006 option.container = self
1007 for opt in option._short_opts:
1008 self._short_opt[opt] = option
1009 for opt in option._long_opts:
1010 self._long_opt[opt] = option
1011
1012 if option.dest is not None: # option has a dest, we need a default
1013 if option.default is not NO_DEFAULT:
1014 self.defaults[option.dest] = option.default
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001015 elif option.dest not in self.defaults:
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001016 self.defaults[option.dest] = None
1017
1018 return option
1019
Greg Wardeba20e62004-07-31 16:15:44 +00001020 def add_options(self, option_list):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001021 for option in option_list:
1022 self.add_option(option)
1023
1024 # -- Option query/removal methods ----------------------------------
1025
Greg Wardeba20e62004-07-31 16:15:44 +00001026 def get_option(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001027 return (self._short_opt.get(opt_str) or
1028 self._long_opt.get(opt_str))
1029
Greg Wardeba20e62004-07-31 16:15:44 +00001030 def has_option(self, opt_str):
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001031 return (opt_str in self._short_opt or
Guido van Rossum93662412006-08-19 16:09:41 +00001032 opt_str in self._long_opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001033
Greg Wardeba20e62004-07-31 16:15:44 +00001034 def remove_option(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001035 option = self._short_opt.get(opt_str)
1036 if option is None:
1037 option = self._long_opt.get(opt_str)
1038 if option is None:
1039 raise ValueError("no such option %r" % opt_str)
1040
1041 for opt in option._short_opts:
1042 del self._short_opt[opt]
1043 for opt in option._long_opts:
1044 del self._long_opt[opt]
1045 option.container.option_list.remove(option)
1046
1047
1048 # -- Help-formatting methods ---------------------------------------
1049
Greg Wardeba20e62004-07-31 16:15:44 +00001050 def format_option_help(self, formatter):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001051 if not self.option_list:
1052 return ""
1053 result = []
1054 for option in self.option_list:
1055 if not option.help is SUPPRESS_HELP:
1056 result.append(formatter.format_option(option))
1057 return "".join(result)
1058
Greg Wardeba20e62004-07-31 16:15:44 +00001059 def format_description(self, formatter):
1060 return formatter.format_description(self.get_description())
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001061
Greg Wardeba20e62004-07-31 16:15:44 +00001062 def format_help(self, formatter):
1063 result = []
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001064 if self.description:
Greg Wardeba20e62004-07-31 16:15:44 +00001065 result.append(self.format_description(formatter))
1066 if self.option_list:
1067 result.append(self.format_option_help(formatter))
1068 return "\n".join(result)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001069
1070
1071class OptionGroup (OptionContainer):
1072
Greg Wardeba20e62004-07-31 16:15:44 +00001073 def __init__(self, parser, title, description=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001074 self.parser = parser
1075 OptionContainer.__init__(
1076 self, parser.option_class, parser.conflict_handler, description)
1077 self.title = title
1078
Greg Wardeba20e62004-07-31 16:15:44 +00001079 def _create_option_list(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001080 self.option_list = []
1081 self._share_option_mappings(self.parser)
1082
Greg Wardeba20e62004-07-31 16:15:44 +00001083 def set_title(self, title):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001084 self.title = title
1085
Thomas Wouters477c8d52006-05-27 19:21:47 +00001086 def destroy(self):
1087 """see OptionParser.destroy()."""
1088 OptionContainer.destroy(self)
1089 del self.option_list
1090
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001091 # -- Help-formatting methods ---------------------------------------
1092
Greg Wardeba20e62004-07-31 16:15:44 +00001093 def format_help(self, formatter):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001094 result = formatter.format_heading(self.title)
1095 formatter.indent()
1096 result += OptionContainer.format_help(self, formatter)
1097 formatter.dedent()
1098 return result
1099
1100
1101class OptionParser (OptionContainer):
1102
1103 """
1104 Class attributes:
1105 standard_option_list : [Option]
1106 list of standard options that will be accepted by all instances
1107 of this parser class (intended to be overridden by subclasses).
1108
1109 Instance attributes:
1110 usage : string
1111 a usage string for your program. Before it is displayed
1112 to the user, "%prog" will be expanded to the name of
Greg Ward2492fcf2003-04-21 02:40:34 +00001113 your program (self.prog or os.path.basename(sys.argv[0])).
1114 prog : string
1115 the name of the current program (to override
1116 os.path.basename(sys.argv[0])).
Thomas Wouters477c8d52006-05-27 19:21:47 +00001117 epilog : string
1118 paragraph of help text to print after option help
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001119
Greg Wardeba20e62004-07-31 16:15:44 +00001120 option_groups : [OptionGroup]
1121 list of option groups in this parser (option groups are
1122 irrelevant for parsing the command-line, but very useful
1123 for generating help)
1124
1125 allow_interspersed_args : bool = true
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001126 if true, positional arguments may be interspersed with options.
1127 Assuming -a and -b each take a single argument, the command-line
1128 -ablah foo bar -bboo baz
1129 will be interpreted the same as
1130 -ablah -bboo -- foo bar baz
1131 If this flag were false, that command line would be interpreted as
1132 -ablah -- foo bar -bboo baz
1133 -- ie. we stop processing options as soon as we see the first
1134 non-option argument. (This is the tradition followed by
1135 Python's getopt module, Perl's Getopt::Std, and other argument-
1136 parsing libraries, but it is generally annoying to users.)
1137
Greg Wardeba20e62004-07-31 16:15:44 +00001138 process_default_values : bool = true
1139 if true, option default values are processed similarly to option
1140 values from the command line: that is, they are passed to the
1141 type-checking function for the option's type (as long as the
1142 default value is a string). (This really only matters if you
1143 have defined custom types; see SF bug #955889.) Set it to false
1144 to restore the behaviour of Optik 1.4.1 and earlier.
1145
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001146 rargs : [string]
1147 the argument list currently being parsed. Only set when
1148 parse_args() is active, and continually trimmed down as
1149 we consume arguments. Mainly there for the benefit of
1150 callback options.
1151 largs : [string]
1152 the list of leftover arguments that we have skipped while
1153 parsing options. If allow_interspersed_args is false, this
1154 list is always empty.
1155 values : Values
1156 the set of option values currently being accumulated. Only
1157 set when parse_args() is active. Also mainly for callbacks.
1158
1159 Because of the 'rargs', 'largs', and 'values' attributes,
1160 OptionParser is not thread-safe. If, for some perverse reason, you
1161 need to parse command-line arguments simultaneously in different
1162 threads, use different OptionParser instances.
1163
1164 """
1165
1166 standard_option_list = []
1167
Greg Wardeba20e62004-07-31 16:15:44 +00001168 def __init__(self,
1169 usage=None,
1170 option_list=None,
1171 option_class=Option,
1172 version=None,
1173 conflict_handler="error",
1174 description=None,
1175 formatter=None,
1176 add_help_option=True,
Thomas Wouters477c8d52006-05-27 19:21:47 +00001177 prog=None,
1178 epilog=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001179 OptionContainer.__init__(
1180 self, option_class, conflict_handler, description)
1181 self.set_usage(usage)
Greg Ward2492fcf2003-04-21 02:40:34 +00001182 self.prog = prog
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001183 self.version = version
Greg Wardeba20e62004-07-31 16:15:44 +00001184 self.allow_interspersed_args = True
1185 self.process_default_values = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001186 if formatter is None:
1187 formatter = IndentedHelpFormatter()
1188 self.formatter = formatter
Greg Wardeba20e62004-07-31 16:15:44 +00001189 self.formatter.set_parser(self)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001190 self.epilog = epilog
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001191
1192 # Populate the option list; initial sources are the
1193 # standard_option_list class attribute, the 'option_list'
Greg Wardeba20e62004-07-31 16:15:44 +00001194 # argument, and (if applicable) the _add_version_option() and
1195 # _add_help_option() methods.
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001196 self._populate_option_list(option_list,
1197 add_help=add_help_option)
1198
1199 self._init_parsing_state()
1200
Thomas Wouters477c8d52006-05-27 19:21:47 +00001201
1202 def destroy(self):
1203 """
1204 Declare that you are done with this OptionParser. This cleans up
1205 reference cycles so the OptionParser (and all objects referenced by
1206 it) can be garbage-collected promptly. After calling destroy(), the
1207 OptionParser is unusable.
1208 """
1209 OptionContainer.destroy(self)
1210 for group in self.option_groups:
1211 group.destroy()
1212 del self.option_list
1213 del self.option_groups
1214 del self.formatter
1215
1216
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001217 # -- Private methods -----------------------------------------------
1218 # (used by our or OptionContainer's constructor)
1219
Greg Wardeba20e62004-07-31 16:15:44 +00001220 def _create_option_list(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001221 self.option_list = []
1222 self.option_groups = []
1223 self._create_option_mappings()
1224
Greg Wardeba20e62004-07-31 16:15:44 +00001225 def _add_help_option(self):
1226 self.add_option("-h", "--help",
1227 action="help",
1228 help=_("show this help message and exit"))
1229
1230 def _add_version_option(self):
1231 self.add_option("--version",
1232 action="version",
1233 help=_("show program's version number and exit"))
1234
1235 def _populate_option_list(self, option_list, add_help=True):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001236 if self.standard_option_list:
1237 self.add_options(self.standard_option_list)
1238 if option_list:
1239 self.add_options(option_list)
1240 if self.version:
Greg Wardeba20e62004-07-31 16:15:44 +00001241 self._add_version_option()
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001242 if add_help:
Greg Wardeba20e62004-07-31 16:15:44 +00001243 self._add_help_option()
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001244
Greg Wardeba20e62004-07-31 16:15:44 +00001245 def _init_parsing_state(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001246 # These are set in parse_args() for the convenience of callbacks.
1247 self.rargs = None
1248 self.largs = None
1249 self.values = None
1250
1251
1252 # -- Simple modifier methods ---------------------------------------
1253
Greg Wardeba20e62004-07-31 16:15:44 +00001254 def set_usage(self, usage):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001255 if usage is None:
Greg Wardeba20e62004-07-31 16:15:44 +00001256 self.usage = _("%prog [options]")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001257 elif usage is SUPPRESS_USAGE:
1258 self.usage = None
Greg Wardeba20e62004-07-31 16:15:44 +00001259 # For backwards compatibility with Optik 1.3 and earlier.
Thomas Wouters477c8d52006-05-27 19:21:47 +00001260 elif usage.lower().startswith("usage: "):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001261 self.usage = usage[7:]
1262 else:
1263 self.usage = usage
1264
Greg Wardeba20e62004-07-31 16:15:44 +00001265 def enable_interspersed_args(self):
1266 self.allow_interspersed_args = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001267
Greg Wardeba20e62004-07-31 16:15:44 +00001268 def disable_interspersed_args(self):
1269 self.allow_interspersed_args = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001270
Greg Wardeba20e62004-07-31 16:15:44 +00001271 def set_process_default_values(self, process):
1272 self.process_default_values = process
1273
1274 def set_default(self, dest, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001275 self.defaults[dest] = value
1276
Greg Wardeba20e62004-07-31 16:15:44 +00001277 def set_defaults(self, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001278 self.defaults.update(kwargs)
1279
Greg Wardeba20e62004-07-31 16:15:44 +00001280 def _get_all_options(self):
1281 options = self.option_list[:]
1282 for group in self.option_groups:
1283 options.extend(group.option_list)
1284 return options
1285
1286 def get_default_values(self):
1287 if not self.process_default_values:
1288 # Old, pre-Optik 1.5 behaviour.
1289 return Values(self.defaults)
1290
1291 defaults = self.defaults.copy()
1292 for option in self._get_all_options():
1293 default = defaults.get(option.dest)
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001294 if isinstance(default, str):
Greg Wardeba20e62004-07-31 16:15:44 +00001295 opt_str = option.get_opt_string()
1296 defaults[option.dest] = option.check_value(opt_str, default)
1297
1298 return Values(defaults)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001299
1300
1301 # -- OptionGroup methods -------------------------------------------
1302
Greg Wardeba20e62004-07-31 16:15:44 +00001303 def add_option_group(self, *args, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001304 # XXX lots of overlap with OptionContainer.add_option()
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001305 if isinstance(args[0], str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001306 group = OptionGroup(self, *args, **kwargs)
1307 elif len(args) == 1 and not kwargs:
1308 group = args[0]
1309 if not isinstance(group, OptionGroup):
Collin Winterce36ad82007-08-30 01:19:48 +00001310 raise TypeError("not an OptionGroup instance: %r" % group)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001311 if group.parser is not self:
Collin Winterce36ad82007-08-30 01:19:48 +00001312 raise ValueError("invalid OptionGroup (wrong parser)")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001313 else:
Collin Winterce36ad82007-08-30 01:19:48 +00001314 raise TypeError("invalid arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001315
1316 self.option_groups.append(group)
1317 return group
1318
Greg Wardeba20e62004-07-31 16:15:44 +00001319 def get_option_group(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001320 option = (self._short_opt.get(opt_str) or
1321 self._long_opt.get(opt_str))
1322 if option and option.container is not self:
1323 return option.container
1324 return None
1325
1326
1327 # -- Option-parsing methods ----------------------------------------
1328
Greg Wardeba20e62004-07-31 16:15:44 +00001329 def _get_args(self, args):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001330 if args is None:
1331 return sys.argv[1:]
1332 else:
1333 return args[:] # don't modify caller's list
1334
Greg Wardeba20e62004-07-31 16:15:44 +00001335 def parse_args(self, args=None, values=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001336 """
1337 parse_args(args : [string] = sys.argv[1:],
1338 values : Values = None)
1339 -> (values : Values, args : [string])
1340
1341 Parse the command-line options found in 'args' (default:
1342 sys.argv[1:]). Any errors result in a call to 'error()', which
1343 by default prints the usage message to stderr and calls
1344 sys.exit() with an error message. On success returns a pair
1345 (values, args) where 'values' is an Values instance (with all
1346 your option values) and 'args' is the list of arguments left
1347 over after parsing options.
1348 """
1349 rargs = self._get_args(args)
1350 if values is None:
1351 values = self.get_default_values()
1352
1353 # Store the halves of the argument list as attributes for the
1354 # convenience of callbacks:
1355 # rargs
1356 # the rest of the command-line (the "r" stands for
1357 # "remaining" or "right-hand")
1358 # largs
1359 # the leftover arguments -- ie. what's left after removing
1360 # options and their arguments (the "l" stands for "leftover"
1361 # or "left-hand")
1362 self.rargs = rargs
1363 self.largs = largs = []
1364 self.values = values
1365
1366 try:
1367 stop = self._process_args(largs, rargs, values)
Guido van Rossumb940e112007-01-10 16:19:56 +00001368 except (BadOptionError, OptionValueError) as err:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001369 self.error(str(err))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001370
1371 args = largs + rargs
1372 return self.check_values(values, args)
1373
Greg Wardeba20e62004-07-31 16:15:44 +00001374 def check_values(self, values, args):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001375 """
1376 check_values(values : Values, args : [string])
1377 -> (values : Values, args : [string])
1378
1379 Check that the supplied option values and leftover arguments are
1380 valid. Returns the option values and leftover arguments
1381 (possibly adjusted, possibly completely new -- whatever you
1382 like). Default implementation just returns the passed-in
1383 values; subclasses may override as desired.
1384 """
1385 return (values, args)
1386
Greg Wardeba20e62004-07-31 16:15:44 +00001387 def _process_args(self, largs, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001388 """_process_args(largs : [string],
1389 rargs : [string],
1390 values : Values)
1391
1392 Process command-line arguments and populate 'values', consuming
1393 options and arguments from 'rargs'. If 'allow_interspersed_args' is
1394 false, stop at the first non-option argument. If true, accumulate any
1395 interspersed non-option arguments in 'largs'.
1396 """
1397 while rargs:
1398 arg = rargs[0]
1399 # We handle bare "--" explicitly, and bare "-" is handled by the
1400 # standard arg handler since the short arg case ensures that the
1401 # len of the opt string is greater than 1.
1402 if arg == "--":
1403 del rargs[0]
1404 return
1405 elif arg[0:2] == "--":
1406 # process a single long option (possibly with value(s))
1407 self._process_long_opt(rargs, values)
1408 elif arg[:1] == "-" and len(arg) > 1:
1409 # process a cluster of short options (possibly with
1410 # value(s) for the last one only)
1411 self._process_short_opts(rargs, values)
1412 elif self.allow_interspersed_args:
1413 largs.append(arg)
1414 del rargs[0]
1415 else:
1416 return # stop now, leave this arg in rargs
1417
1418 # Say this is the original argument list:
1419 # [arg0, arg1, ..., arg(i-1), arg(i), arg(i+1), ..., arg(N-1)]
1420 # ^
1421 # (we are about to process arg(i)).
1422 #
1423 # Then rargs is [arg(i), ..., arg(N-1)] and largs is a *subset* of
1424 # [arg0, ..., arg(i-1)] (any options and their arguments will have
1425 # been removed from largs).
1426 #
1427 # The while loop will usually consume 1 or more arguments per pass.
1428 # If it consumes 1 (eg. arg is an option that takes no arguments),
1429 # then after _process_arg() is done the situation is:
1430 #
1431 # largs = subset of [arg0, ..., arg(i)]
1432 # rargs = [arg(i+1), ..., arg(N-1)]
1433 #
1434 # If allow_interspersed_args is false, largs will always be
1435 # *empty* -- still a subset of [arg0, ..., arg(i-1)], but
1436 # not a very interesting subset!
1437
Greg Wardeba20e62004-07-31 16:15:44 +00001438 def _match_long_opt(self, opt):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001439 """_match_long_opt(opt : string) -> string
1440
1441 Determine which long option string 'opt' matches, ie. which one
1442 it is an unambiguous abbrevation for. Raises BadOptionError if
1443 'opt' doesn't unambiguously match any long option string.
1444 """
1445 return _match_abbrev(opt, self._long_opt)
1446
Greg Wardeba20e62004-07-31 16:15:44 +00001447 def _process_long_opt(self, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001448 arg = rargs.pop(0)
1449
1450 # Value explicitly attached to arg? Pretend it's the next
1451 # argument.
1452 if "=" in arg:
1453 (opt, next_arg) = arg.split("=", 1)
1454 rargs.insert(0, next_arg)
Greg Wardeba20e62004-07-31 16:15:44 +00001455 had_explicit_value = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001456 else:
1457 opt = arg
Greg Wardeba20e62004-07-31 16:15:44 +00001458 had_explicit_value = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001459
1460 opt = self._match_long_opt(opt)
1461 option = self._long_opt[opt]
1462 if option.takes_value():
1463 nargs = option.nargs
1464 if len(rargs) < nargs:
1465 if nargs == 1:
Greg Wardeba20e62004-07-31 16:15:44 +00001466 self.error(_("%s option requires an argument") % opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001467 else:
Greg Wardeba20e62004-07-31 16:15:44 +00001468 self.error(_("%s option requires %d arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001469 % (opt, nargs))
1470 elif nargs == 1:
1471 value = rargs.pop(0)
1472 else:
1473 value = tuple(rargs[0:nargs])
1474 del rargs[0:nargs]
1475
1476 elif had_explicit_value:
Greg Wardeba20e62004-07-31 16:15:44 +00001477 self.error(_("%s option does not take a value") % opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001478
1479 else:
1480 value = None
1481
1482 option.process(opt, value, values, self)
1483
Greg Wardeba20e62004-07-31 16:15:44 +00001484 def _process_short_opts(self, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001485 arg = rargs.pop(0)
Greg Wardeba20e62004-07-31 16:15:44 +00001486 stop = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001487 i = 1
1488 for ch in arg[1:]:
1489 opt = "-" + ch
1490 option = self._short_opt.get(opt)
1491 i += 1 # we have consumed a character
1492
1493 if not option:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001494 raise BadOptionError(opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001495 if option.takes_value():
1496 # Any characters left in arg? Pretend they're the
1497 # next arg, and stop consuming characters of arg.
1498 if i < len(arg):
1499 rargs.insert(0, arg[i:])
Greg Wardeba20e62004-07-31 16:15:44 +00001500 stop = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001501
1502 nargs = option.nargs
1503 if len(rargs) < nargs:
1504 if nargs == 1:
Greg Wardeba20e62004-07-31 16:15:44 +00001505 self.error(_("%s option requires an argument") % opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001506 else:
Greg Wardeba20e62004-07-31 16:15:44 +00001507 self.error(_("%s option requires %d arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001508 % (opt, nargs))
1509 elif nargs == 1:
1510 value = rargs.pop(0)
1511 else:
1512 value = tuple(rargs[0:nargs])
1513 del rargs[0:nargs]
1514
1515 else: # option doesn't take a value
1516 value = None
1517
1518 option.process(opt, value, values, self)
1519
1520 if stop:
1521 break
1522
1523
1524 # -- Feedback methods ----------------------------------------------
1525
Greg Wardeba20e62004-07-31 16:15:44 +00001526 def get_prog_name(self):
1527 if self.prog is None:
1528 return os.path.basename(sys.argv[0])
1529 else:
1530 return self.prog
1531
1532 def expand_prog_name(self, s):
1533 return s.replace("%prog", self.get_prog_name())
1534
1535 def get_description(self):
1536 return self.expand_prog_name(self.description)
1537
Greg Ward48aa84b2004-10-27 02:20:04 +00001538 def exit(self, status=0, msg=None):
1539 if msg:
1540 sys.stderr.write(msg)
1541 sys.exit(status)
1542
Greg Wardeba20e62004-07-31 16:15:44 +00001543 def error(self, msg):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001544 """error(msg : string)
1545
1546 Print a usage message incorporating 'msg' to stderr and exit.
1547 If you override this in a subclass, it should not return -- it
1548 should either exit or raise an exception.
1549 """
1550 self.print_usage(sys.stderr)
Greg Ward48aa84b2004-10-27 02:20:04 +00001551 self.exit(2, "%s: error: %s\n" % (self.get_prog_name(), msg))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001552
Greg Wardeba20e62004-07-31 16:15:44 +00001553 def get_usage(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001554 if self.usage:
1555 return self.formatter.format_usage(
Greg Wardeba20e62004-07-31 16:15:44 +00001556 self.expand_prog_name(self.usage))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001557 else:
1558 return ""
1559
Greg Wardeba20e62004-07-31 16:15:44 +00001560 def print_usage(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001561 """print_usage(file : file = stdout)
1562
1563 Print the usage message for the current program (self.usage) to
1564 'file' (default stdout). Any occurence of the string "%prog" in
1565 self.usage is replaced with the name of the current program
1566 (basename of sys.argv[0]). Does nothing if self.usage is empty
1567 or not defined.
1568 """
1569 if self.usage:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001570 print(self.get_usage(), file=file)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001571
Greg Wardeba20e62004-07-31 16:15:44 +00001572 def get_version(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001573 if self.version:
Greg Wardeba20e62004-07-31 16:15:44 +00001574 return self.expand_prog_name(self.version)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001575 else:
1576 return ""
1577
Greg Wardeba20e62004-07-31 16:15:44 +00001578 def print_version(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001579 """print_version(file : file = stdout)
1580
1581 Print the version message for this program (self.version) to
1582 'file' (default stdout). As with print_usage(), any occurence
1583 of "%prog" in self.version is replaced by the current program's
1584 name. Does nothing if self.version is empty or undefined.
1585 """
1586 if self.version:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001587 print(self.get_version(), file=file)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001588
Greg Wardeba20e62004-07-31 16:15:44 +00001589 def format_option_help(self, formatter=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001590 if formatter is None:
1591 formatter = self.formatter
1592 formatter.store_option_strings(self)
1593 result = []
Thomas Wouters477c8d52006-05-27 19:21:47 +00001594 result.append(formatter.format_heading(_("Options")))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001595 formatter.indent()
1596 if self.option_list:
1597 result.append(OptionContainer.format_option_help(self, formatter))
1598 result.append("\n")
1599 for group in self.option_groups:
1600 result.append(group.format_help(formatter))
1601 result.append("\n")
1602 formatter.dedent()
1603 # Drop the last "\n", or the header if no options or option groups:
1604 return "".join(result[:-1])
1605
Thomas Wouters477c8d52006-05-27 19:21:47 +00001606 def format_epilog(self, formatter):
1607 return formatter.format_epilog(self.epilog)
1608
Greg Wardeba20e62004-07-31 16:15:44 +00001609 def format_help(self, formatter=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001610 if formatter is None:
1611 formatter = self.formatter
1612 result = []
1613 if self.usage:
1614 result.append(self.get_usage() + "\n")
1615 if self.description:
1616 result.append(self.format_description(formatter) + "\n")
1617 result.append(self.format_option_help(formatter))
Thomas Wouters477c8d52006-05-27 19:21:47 +00001618 result.append(self.format_epilog(formatter))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001619 return "".join(result)
1620
Greg Wardeba20e62004-07-31 16:15:44 +00001621 def print_help(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001622 """print_help(file : file = stdout)
1623
1624 Print an extended help message, listing all options and any
1625 help text provided with them, to 'file' (default stdout).
1626 """
1627 if file is None:
1628 file = sys.stdout
Guido van Rossum34d19282007-08-09 01:03:29 +00001629 file.write(self.format_help())
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001630
1631# class OptionParser
1632
1633
Greg Wardeba20e62004-07-31 16:15:44 +00001634def _match_abbrev(s, wordmap):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001635 """_match_abbrev(s : string, wordmap : {string : Option}) -> string
1636
1637 Return the string key in 'wordmap' for which 's' is an unambiguous
1638 abbreviation. If 's' is found to be ambiguous or doesn't match any of
1639 'words', raise BadOptionError.
1640 """
1641 # Is there an exact match?
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001642 if s in wordmap:
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001643 return s
1644 else:
1645 # Isolate all words with s as a prefix.
1646 possibilities = [word for word in wordmap.keys()
1647 if word.startswith(s)]
1648 # No exact match, so there had better be just one possibility.
1649 if len(possibilities) == 1:
1650 return possibilities[0]
1651 elif not possibilities:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001652 raise BadOptionError(s)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001653 else:
1654 # More than one possible completion: ambiguous prefix.
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001655 possibilities.sort()
Thomas Wouters477c8d52006-05-27 19:21:47 +00001656 raise AmbiguousOptionError(s, possibilities)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001657
1658
1659# Some day, there might be many Option classes. As of Optik 1.3, the
1660# preferred way to instantiate Options is indirectly, via make_option(),
1661# which will become a factory function when there are many Option
1662# classes.
1663make_option = Option