blob: e648a281e4ca7a173d81b3e492ea4892d251d324 [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):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000242 raise NotImplementedError, "subclasses must implement"
243
Greg Wardeba20e62004-07-31 16:15:44 +0000244 def format_heading(self, heading):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000245 raise NotImplementedError, "subclasses must implement"
246
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
639 # complicated check of __builtin__ is only necessary for
640 # Python 2.1 and earlier, and is short-circuited by the
641 # first check on modern Pythons.)
642 import __builtin__
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
645 getattr(__builtin__, 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:
808 raise RuntimeError, "unknown action %r" % self.action
809
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
Thomas Wouters477c8d52006-05-27 19:21:47 +0000818def isbasestring(x):
Guido van Rossumc2b87a62007-05-04 17:44:47 +0000819 return isinstance(x, basestring)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000820
821class Values:
822
Greg Wardeba20e62004-07-31 16:15:44 +0000823 def __init__(self, defaults=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000824 if defaults:
825 for (attr, val) in defaults.items():
826 setattr(self, attr, val)
827
Greg Wardeba20e62004-07-31 16:15:44 +0000828 def __str__(self):
829 return str(self.__dict__)
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000830
Greg Wardeba20e62004-07-31 16:15:44 +0000831 __repr__ = _repr
832
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000833 def __eq__(self, other):
Greg Wardeba20e62004-07-31 16:15:44 +0000834 if isinstance(other, Values):
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000835 return self.__dict__ == other.__dict__
Guido van Rossum13257902007-06-07 23:15:56 +0000836 elif isinstance(other, dict):
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000837 return self.__dict__ == other
Greg Wardeba20e62004-07-31 16:15:44 +0000838 else:
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000839 return NotImplemented
Greg Wardeba20e62004-07-31 16:15:44 +0000840
841 def _update_careful(self, dict):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000842 """
843 Update the option values from an arbitrary dictionary, but only
844 use keys from dict that already have a corresponding attribute
845 in self. Any keys in dict without a corresponding attribute
846 are silently ignored.
847 """
848 for attr in dir(self):
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000849 if attr in dict:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000850 dval = dict[attr]
851 if dval is not None:
852 setattr(self, attr, dval)
853
Greg Wardeba20e62004-07-31 16:15:44 +0000854 def _update_loose(self, dict):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000855 """
856 Update the option values from an arbitrary dictionary,
857 using all keys from the dictionary regardless of whether
858 they have a corresponding attribute in self or not.
859 """
860 self.__dict__.update(dict)
861
Greg Wardeba20e62004-07-31 16:15:44 +0000862 def _update(self, dict, mode):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000863 if mode == "careful":
864 self._update_careful(dict)
865 elif mode == "loose":
866 self._update_loose(dict)
867 else:
868 raise ValueError, "invalid update mode: %r" % mode
869
Greg Wardeba20e62004-07-31 16:15:44 +0000870 def read_module(self, modname, mode="careful"):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000871 __import__(modname)
872 mod = sys.modules[modname]
873 self._update(vars(mod), mode)
874
Greg Wardeba20e62004-07-31 16:15:44 +0000875 def read_file(self, filename, mode="careful"):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000876 vars = {}
877 execfile(filename, vars)
878 self._update(vars, mode)
879
Greg Wardeba20e62004-07-31 16:15:44 +0000880 def ensure_value(self, attr, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000881 if not hasattr(self, attr) or getattr(self, attr) is None:
882 setattr(self, attr, value)
883 return getattr(self, attr)
884
885
886class OptionContainer:
887
888 """
889 Abstract base class.
890
891 Class attributes:
892 standard_option_list : [Option]
893 list of standard options that will be accepted by all instances
894 of this parser class (intended to be overridden by subclasses).
895
896 Instance attributes:
897 option_list : [Option]
898 the list of Option objects contained by this OptionContainer
899 _short_opt : { string : Option }
900 dictionary mapping short option strings, eg. "-f" or "-X",
901 to the Option instances that implement them. If an Option
902 has multiple short option strings, it will appears in this
903 dictionary multiple times. [1]
904 _long_opt : { string : Option }
905 dictionary mapping long option strings, eg. "--file" or
906 "--exclude", to the Option instances that implement them.
907 Again, a given Option can occur multiple times in this
908 dictionary. [1]
909 defaults : { string : any }
910 dictionary mapping option destination names to default
911 values for each destination [1]
912
913 [1] These mappings are common to (shared by) all components of the
914 controlling OptionParser, where they are initially created.
915
916 """
917
Greg Wardeba20e62004-07-31 16:15:44 +0000918 def __init__(self, option_class, conflict_handler, description):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000919 # Initialize the option list and related data structures.
920 # This method must be provided by subclasses, and it must
921 # initialize at least the following instance attributes:
922 # option_list, _short_opt, _long_opt, defaults.
923 self._create_option_list()
924
925 self.option_class = option_class
926 self.set_conflict_handler(conflict_handler)
927 self.set_description(description)
928
Greg Wardeba20e62004-07-31 16:15:44 +0000929 def _create_option_mappings(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000930 # For use by OptionParser constructor -- create the master
931 # option mappings used by this OptionParser and all
932 # OptionGroups that it owns.
933 self._short_opt = {} # single letter -> Option instance
934 self._long_opt = {} # long option -> Option instance
935 self.defaults = {} # maps option dest -> default value
936
937
Greg Wardeba20e62004-07-31 16:15:44 +0000938 def _share_option_mappings(self, parser):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000939 # For use by OptionGroup constructor -- use shared option
940 # mappings from the OptionParser that owns this OptionGroup.
941 self._short_opt = parser._short_opt
942 self._long_opt = parser._long_opt
943 self.defaults = parser.defaults
944
Greg Wardeba20e62004-07-31 16:15:44 +0000945 def set_conflict_handler(self, handler):
Greg Ward48aa84b2004-10-27 02:20:04 +0000946 if handler not in ("error", "resolve"):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000947 raise ValueError, "invalid conflict_resolution value %r" % handler
948 self.conflict_handler = handler
949
Greg Wardeba20e62004-07-31 16:15:44 +0000950 def set_description(self, description):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000951 self.description = description
952
Greg Wardeba20e62004-07-31 16:15:44 +0000953 def get_description(self):
954 return self.description
955
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000956
Thomas Wouters477c8d52006-05-27 19:21:47 +0000957 def destroy(self):
958 """see OptionParser.destroy()."""
959 del self._short_opt
960 del self._long_opt
961 del self.defaults
962
963
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000964 # -- Option-adding methods -----------------------------------------
965
Greg Wardeba20e62004-07-31 16:15:44 +0000966 def _check_conflict(self, option):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000967 conflict_opts = []
968 for opt in option._short_opts:
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000969 if opt in self._short_opt:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000970 conflict_opts.append((opt, self._short_opt[opt]))
971 for opt in option._long_opts:
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000972 if opt in self._long_opt:
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000973 conflict_opts.append((opt, self._long_opt[opt]))
974
975 if conflict_opts:
976 handler = self.conflict_handler
Greg Ward48aa84b2004-10-27 02:20:04 +0000977 if handler == "error":
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000978 raise OptionConflictError(
979 "conflicting option string(s): %s"
980 % ", ".join([co[0] for co in conflict_opts]),
981 option)
Greg Ward48aa84b2004-10-27 02:20:04 +0000982 elif handler == "resolve":
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000983 for (opt, c_option) in conflict_opts:
984 if opt.startswith("--"):
985 c_option._long_opts.remove(opt)
986 del self._long_opt[opt]
987 else:
988 c_option._short_opts.remove(opt)
989 del self._short_opt[opt]
990 if not (c_option._short_opts or c_option._long_opts):
991 c_option.container.option_list.remove(c_option)
992
Greg Wardeba20e62004-07-31 16:15:44 +0000993 def add_option(self, *args, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000994 """add_option(Option)
995 add_option(opt_str, ..., kwarg=val, ...)
996 """
Guido van Rossumc2b87a62007-05-04 17:44:47 +0000997 if isbasestring(args[0]):
Guido van Rossumb9ba4582002-11-14 22:00:19 +0000998 option = self.option_class(*args, **kwargs)
999 elif len(args) == 1 and not kwargs:
1000 option = args[0]
1001 if not isinstance(option, Option):
1002 raise TypeError, "not an Option instance: %r" % option
1003 else:
1004 raise TypeError, "invalid arguments"
1005
1006 self._check_conflict(option)
1007
1008 self.option_list.append(option)
1009 option.container = self
1010 for opt in option._short_opts:
1011 self._short_opt[opt] = option
1012 for opt in option._long_opts:
1013 self._long_opt[opt] = option
1014
1015 if option.dest is not None: # option has a dest, we need a default
1016 if option.default is not NO_DEFAULT:
1017 self.defaults[option.dest] = option.default
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001018 elif option.dest not in self.defaults:
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001019 self.defaults[option.dest] = None
1020
1021 return option
1022
Greg Wardeba20e62004-07-31 16:15:44 +00001023 def add_options(self, option_list):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001024 for option in option_list:
1025 self.add_option(option)
1026
1027 # -- Option query/removal methods ----------------------------------
1028
Greg Wardeba20e62004-07-31 16:15:44 +00001029 def get_option(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001030 return (self._short_opt.get(opt_str) or
1031 self._long_opt.get(opt_str))
1032
Greg Wardeba20e62004-07-31 16:15:44 +00001033 def has_option(self, opt_str):
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001034 return (opt_str in self._short_opt or
Guido van Rossum93662412006-08-19 16:09:41 +00001035 opt_str in self._long_opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001036
Greg Wardeba20e62004-07-31 16:15:44 +00001037 def remove_option(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001038 option = self._short_opt.get(opt_str)
1039 if option is None:
1040 option = self._long_opt.get(opt_str)
1041 if option is None:
1042 raise ValueError("no such option %r" % opt_str)
1043
1044 for opt in option._short_opts:
1045 del self._short_opt[opt]
1046 for opt in option._long_opts:
1047 del self._long_opt[opt]
1048 option.container.option_list.remove(option)
1049
1050
1051 # -- Help-formatting methods ---------------------------------------
1052
Greg Wardeba20e62004-07-31 16:15:44 +00001053 def format_option_help(self, formatter):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001054 if not self.option_list:
1055 return ""
1056 result = []
1057 for option in self.option_list:
1058 if not option.help is SUPPRESS_HELP:
1059 result.append(formatter.format_option(option))
1060 return "".join(result)
1061
Greg Wardeba20e62004-07-31 16:15:44 +00001062 def format_description(self, formatter):
1063 return formatter.format_description(self.get_description())
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001064
Greg Wardeba20e62004-07-31 16:15:44 +00001065 def format_help(self, formatter):
1066 result = []
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001067 if self.description:
Greg Wardeba20e62004-07-31 16:15:44 +00001068 result.append(self.format_description(formatter))
1069 if self.option_list:
1070 result.append(self.format_option_help(formatter))
1071 return "\n".join(result)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001072
1073
1074class OptionGroup (OptionContainer):
1075
Greg Wardeba20e62004-07-31 16:15:44 +00001076 def __init__(self, parser, title, description=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001077 self.parser = parser
1078 OptionContainer.__init__(
1079 self, parser.option_class, parser.conflict_handler, description)
1080 self.title = title
1081
Greg Wardeba20e62004-07-31 16:15:44 +00001082 def _create_option_list(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001083 self.option_list = []
1084 self._share_option_mappings(self.parser)
1085
Greg Wardeba20e62004-07-31 16:15:44 +00001086 def set_title(self, title):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001087 self.title = title
1088
Thomas Wouters477c8d52006-05-27 19:21:47 +00001089 def destroy(self):
1090 """see OptionParser.destroy()."""
1091 OptionContainer.destroy(self)
1092 del self.option_list
1093
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001094 # -- Help-formatting methods ---------------------------------------
1095
Greg Wardeba20e62004-07-31 16:15:44 +00001096 def format_help(self, formatter):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001097 result = formatter.format_heading(self.title)
1098 formatter.indent()
1099 result += OptionContainer.format_help(self, formatter)
1100 formatter.dedent()
1101 return result
1102
1103
1104class OptionParser (OptionContainer):
1105
1106 """
1107 Class attributes:
1108 standard_option_list : [Option]
1109 list of standard options that will be accepted by all instances
1110 of this parser class (intended to be overridden by subclasses).
1111
1112 Instance attributes:
1113 usage : string
1114 a usage string for your program. Before it is displayed
1115 to the user, "%prog" will be expanded to the name of
Greg Ward2492fcf2003-04-21 02:40:34 +00001116 your program (self.prog or os.path.basename(sys.argv[0])).
1117 prog : string
1118 the name of the current program (to override
1119 os.path.basename(sys.argv[0])).
Thomas Wouters477c8d52006-05-27 19:21:47 +00001120 epilog : string
1121 paragraph of help text to print after option help
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001122
Greg Wardeba20e62004-07-31 16:15:44 +00001123 option_groups : [OptionGroup]
1124 list of option groups in this parser (option groups are
1125 irrelevant for parsing the command-line, but very useful
1126 for generating help)
1127
1128 allow_interspersed_args : bool = true
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001129 if true, positional arguments may be interspersed with options.
1130 Assuming -a and -b each take a single argument, the command-line
1131 -ablah foo bar -bboo baz
1132 will be interpreted the same as
1133 -ablah -bboo -- foo bar baz
1134 If this flag were false, that command line would be interpreted as
1135 -ablah -- foo bar -bboo baz
1136 -- ie. we stop processing options as soon as we see the first
1137 non-option argument. (This is the tradition followed by
1138 Python's getopt module, Perl's Getopt::Std, and other argument-
1139 parsing libraries, but it is generally annoying to users.)
1140
Greg Wardeba20e62004-07-31 16:15:44 +00001141 process_default_values : bool = true
1142 if true, option default values are processed similarly to option
1143 values from the command line: that is, they are passed to the
1144 type-checking function for the option's type (as long as the
1145 default value is a string). (This really only matters if you
1146 have defined custom types; see SF bug #955889.) Set it to false
1147 to restore the behaviour of Optik 1.4.1 and earlier.
1148
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001149 rargs : [string]
1150 the argument list currently being parsed. Only set when
1151 parse_args() is active, and continually trimmed down as
1152 we consume arguments. Mainly there for the benefit of
1153 callback options.
1154 largs : [string]
1155 the list of leftover arguments that we have skipped while
1156 parsing options. If allow_interspersed_args is false, this
1157 list is always empty.
1158 values : Values
1159 the set of option values currently being accumulated. Only
1160 set when parse_args() is active. Also mainly for callbacks.
1161
1162 Because of the 'rargs', 'largs', and 'values' attributes,
1163 OptionParser is not thread-safe. If, for some perverse reason, you
1164 need to parse command-line arguments simultaneously in different
1165 threads, use different OptionParser instances.
1166
1167 """
1168
1169 standard_option_list = []
1170
Greg Wardeba20e62004-07-31 16:15:44 +00001171 def __init__(self,
1172 usage=None,
1173 option_list=None,
1174 option_class=Option,
1175 version=None,
1176 conflict_handler="error",
1177 description=None,
1178 formatter=None,
1179 add_help_option=True,
Thomas Wouters477c8d52006-05-27 19:21:47 +00001180 prog=None,
1181 epilog=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001182 OptionContainer.__init__(
1183 self, option_class, conflict_handler, description)
1184 self.set_usage(usage)
Greg Ward2492fcf2003-04-21 02:40:34 +00001185 self.prog = prog
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001186 self.version = version
Greg Wardeba20e62004-07-31 16:15:44 +00001187 self.allow_interspersed_args = True
1188 self.process_default_values = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001189 if formatter is None:
1190 formatter = IndentedHelpFormatter()
1191 self.formatter = formatter
Greg Wardeba20e62004-07-31 16:15:44 +00001192 self.formatter.set_parser(self)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001193 self.epilog = epilog
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001194
1195 # Populate the option list; initial sources are the
1196 # standard_option_list class attribute, the 'option_list'
Greg Wardeba20e62004-07-31 16:15:44 +00001197 # argument, and (if applicable) the _add_version_option() and
1198 # _add_help_option() methods.
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001199 self._populate_option_list(option_list,
1200 add_help=add_help_option)
1201
1202 self._init_parsing_state()
1203
Thomas Wouters477c8d52006-05-27 19:21:47 +00001204
1205 def destroy(self):
1206 """
1207 Declare that you are done with this OptionParser. This cleans up
1208 reference cycles so the OptionParser (and all objects referenced by
1209 it) can be garbage-collected promptly. After calling destroy(), the
1210 OptionParser is unusable.
1211 """
1212 OptionContainer.destroy(self)
1213 for group in self.option_groups:
1214 group.destroy()
1215 del self.option_list
1216 del self.option_groups
1217 del self.formatter
1218
1219
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001220 # -- Private methods -----------------------------------------------
1221 # (used by our or OptionContainer's constructor)
1222
Greg Wardeba20e62004-07-31 16:15:44 +00001223 def _create_option_list(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001224 self.option_list = []
1225 self.option_groups = []
1226 self._create_option_mappings()
1227
Greg Wardeba20e62004-07-31 16:15:44 +00001228 def _add_help_option(self):
1229 self.add_option("-h", "--help",
1230 action="help",
1231 help=_("show this help message and exit"))
1232
1233 def _add_version_option(self):
1234 self.add_option("--version",
1235 action="version",
1236 help=_("show program's version number and exit"))
1237
1238 def _populate_option_list(self, option_list, add_help=True):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001239 if self.standard_option_list:
1240 self.add_options(self.standard_option_list)
1241 if option_list:
1242 self.add_options(option_list)
1243 if self.version:
Greg Wardeba20e62004-07-31 16:15:44 +00001244 self._add_version_option()
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001245 if add_help:
Greg Wardeba20e62004-07-31 16:15:44 +00001246 self._add_help_option()
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001247
Greg Wardeba20e62004-07-31 16:15:44 +00001248 def _init_parsing_state(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001249 # These are set in parse_args() for the convenience of callbacks.
1250 self.rargs = None
1251 self.largs = None
1252 self.values = None
1253
1254
1255 # -- Simple modifier methods ---------------------------------------
1256
Greg Wardeba20e62004-07-31 16:15:44 +00001257 def set_usage(self, usage):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001258 if usage is None:
Greg Wardeba20e62004-07-31 16:15:44 +00001259 self.usage = _("%prog [options]")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001260 elif usage is SUPPRESS_USAGE:
1261 self.usage = None
Greg Wardeba20e62004-07-31 16:15:44 +00001262 # For backwards compatibility with Optik 1.3 and earlier.
Thomas Wouters477c8d52006-05-27 19:21:47 +00001263 elif usage.lower().startswith("usage: "):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001264 self.usage = usage[7:]
1265 else:
1266 self.usage = usage
1267
Greg Wardeba20e62004-07-31 16:15:44 +00001268 def enable_interspersed_args(self):
1269 self.allow_interspersed_args = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001270
Greg Wardeba20e62004-07-31 16:15:44 +00001271 def disable_interspersed_args(self):
1272 self.allow_interspersed_args = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001273
Greg Wardeba20e62004-07-31 16:15:44 +00001274 def set_process_default_values(self, process):
1275 self.process_default_values = process
1276
1277 def set_default(self, dest, value):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001278 self.defaults[dest] = value
1279
Greg Wardeba20e62004-07-31 16:15:44 +00001280 def set_defaults(self, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001281 self.defaults.update(kwargs)
1282
Greg Wardeba20e62004-07-31 16:15:44 +00001283 def _get_all_options(self):
1284 options = self.option_list[:]
1285 for group in self.option_groups:
1286 options.extend(group.option_list)
1287 return options
1288
1289 def get_default_values(self):
1290 if not self.process_default_values:
1291 # Old, pre-Optik 1.5 behaviour.
1292 return Values(self.defaults)
1293
1294 defaults = self.defaults.copy()
1295 for option in self._get_all_options():
1296 default = defaults.get(option.dest)
Thomas Wouters477c8d52006-05-27 19:21:47 +00001297 if isbasestring(default):
Greg Wardeba20e62004-07-31 16:15:44 +00001298 opt_str = option.get_opt_string()
1299 defaults[option.dest] = option.check_value(opt_str, default)
1300
1301 return Values(defaults)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001302
1303
1304 # -- OptionGroup methods -------------------------------------------
1305
Greg Wardeba20e62004-07-31 16:15:44 +00001306 def add_option_group(self, *args, **kwargs):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001307 # XXX lots of overlap with OptionContainer.add_option()
Guido van Rossumc2b87a62007-05-04 17:44:47 +00001308 if isbasestring(args[0]):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001309 group = OptionGroup(self, *args, **kwargs)
1310 elif len(args) == 1 and not kwargs:
1311 group = args[0]
1312 if not isinstance(group, OptionGroup):
1313 raise TypeError, "not an OptionGroup instance: %r" % group
1314 if group.parser is not self:
1315 raise ValueError, "invalid OptionGroup (wrong parser)"
1316 else:
1317 raise TypeError, "invalid arguments"
1318
1319 self.option_groups.append(group)
1320 return group
1321
Greg Wardeba20e62004-07-31 16:15:44 +00001322 def get_option_group(self, opt_str):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001323 option = (self._short_opt.get(opt_str) or
1324 self._long_opt.get(opt_str))
1325 if option and option.container is not self:
1326 return option.container
1327 return None
1328
1329
1330 # -- Option-parsing methods ----------------------------------------
1331
Greg Wardeba20e62004-07-31 16:15:44 +00001332 def _get_args(self, args):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001333 if args is None:
1334 return sys.argv[1:]
1335 else:
1336 return args[:] # don't modify caller's list
1337
Greg Wardeba20e62004-07-31 16:15:44 +00001338 def parse_args(self, args=None, values=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001339 """
1340 parse_args(args : [string] = sys.argv[1:],
1341 values : Values = None)
1342 -> (values : Values, args : [string])
1343
1344 Parse the command-line options found in 'args' (default:
1345 sys.argv[1:]). Any errors result in a call to 'error()', which
1346 by default prints the usage message to stderr and calls
1347 sys.exit() with an error message. On success returns a pair
1348 (values, args) where 'values' is an Values instance (with all
1349 your option values) and 'args' is the list of arguments left
1350 over after parsing options.
1351 """
1352 rargs = self._get_args(args)
1353 if values is None:
1354 values = self.get_default_values()
1355
1356 # Store the halves of the argument list as attributes for the
1357 # convenience of callbacks:
1358 # rargs
1359 # the rest of the command-line (the "r" stands for
1360 # "remaining" or "right-hand")
1361 # largs
1362 # the leftover arguments -- ie. what's left after removing
1363 # options and their arguments (the "l" stands for "leftover"
1364 # or "left-hand")
1365 self.rargs = rargs
1366 self.largs = largs = []
1367 self.values = values
1368
1369 try:
1370 stop = self._process_args(largs, rargs, values)
Guido van Rossumb940e112007-01-10 16:19:56 +00001371 except (BadOptionError, OptionValueError) as err:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001372 self.error(str(err))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001373
1374 args = largs + rargs
1375 return self.check_values(values, args)
1376
Greg Wardeba20e62004-07-31 16:15:44 +00001377 def check_values(self, values, args):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001378 """
1379 check_values(values : Values, args : [string])
1380 -> (values : Values, args : [string])
1381
1382 Check that the supplied option values and leftover arguments are
1383 valid. Returns the option values and leftover arguments
1384 (possibly adjusted, possibly completely new -- whatever you
1385 like). Default implementation just returns the passed-in
1386 values; subclasses may override as desired.
1387 """
1388 return (values, args)
1389
Greg Wardeba20e62004-07-31 16:15:44 +00001390 def _process_args(self, largs, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001391 """_process_args(largs : [string],
1392 rargs : [string],
1393 values : Values)
1394
1395 Process command-line arguments and populate 'values', consuming
1396 options and arguments from 'rargs'. If 'allow_interspersed_args' is
1397 false, stop at the first non-option argument. If true, accumulate any
1398 interspersed non-option arguments in 'largs'.
1399 """
1400 while rargs:
1401 arg = rargs[0]
1402 # We handle bare "--" explicitly, and bare "-" is handled by the
1403 # standard arg handler since the short arg case ensures that the
1404 # len of the opt string is greater than 1.
1405 if arg == "--":
1406 del rargs[0]
1407 return
1408 elif arg[0:2] == "--":
1409 # process a single long option (possibly with value(s))
1410 self._process_long_opt(rargs, values)
1411 elif arg[:1] == "-" and len(arg) > 1:
1412 # process a cluster of short options (possibly with
1413 # value(s) for the last one only)
1414 self._process_short_opts(rargs, values)
1415 elif self.allow_interspersed_args:
1416 largs.append(arg)
1417 del rargs[0]
1418 else:
1419 return # stop now, leave this arg in rargs
1420
1421 # Say this is the original argument list:
1422 # [arg0, arg1, ..., arg(i-1), arg(i), arg(i+1), ..., arg(N-1)]
1423 # ^
1424 # (we are about to process arg(i)).
1425 #
1426 # Then rargs is [arg(i), ..., arg(N-1)] and largs is a *subset* of
1427 # [arg0, ..., arg(i-1)] (any options and their arguments will have
1428 # been removed from largs).
1429 #
1430 # The while loop will usually consume 1 or more arguments per pass.
1431 # If it consumes 1 (eg. arg is an option that takes no arguments),
1432 # then after _process_arg() is done the situation is:
1433 #
1434 # largs = subset of [arg0, ..., arg(i)]
1435 # rargs = [arg(i+1), ..., arg(N-1)]
1436 #
1437 # If allow_interspersed_args is false, largs will always be
1438 # *empty* -- still a subset of [arg0, ..., arg(i-1)], but
1439 # not a very interesting subset!
1440
Greg Wardeba20e62004-07-31 16:15:44 +00001441 def _match_long_opt(self, opt):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001442 """_match_long_opt(opt : string) -> string
1443
1444 Determine which long option string 'opt' matches, ie. which one
1445 it is an unambiguous abbrevation for. Raises BadOptionError if
1446 'opt' doesn't unambiguously match any long option string.
1447 """
1448 return _match_abbrev(opt, self._long_opt)
1449
Greg Wardeba20e62004-07-31 16:15:44 +00001450 def _process_long_opt(self, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001451 arg = rargs.pop(0)
1452
1453 # Value explicitly attached to arg? Pretend it's the next
1454 # argument.
1455 if "=" in arg:
1456 (opt, next_arg) = arg.split("=", 1)
1457 rargs.insert(0, next_arg)
Greg Wardeba20e62004-07-31 16:15:44 +00001458 had_explicit_value = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001459 else:
1460 opt = arg
Greg Wardeba20e62004-07-31 16:15:44 +00001461 had_explicit_value = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001462
1463 opt = self._match_long_opt(opt)
1464 option = self._long_opt[opt]
1465 if option.takes_value():
1466 nargs = option.nargs
1467 if len(rargs) < nargs:
1468 if nargs == 1:
Greg Wardeba20e62004-07-31 16:15:44 +00001469 self.error(_("%s option requires an argument") % opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001470 else:
Greg Wardeba20e62004-07-31 16:15:44 +00001471 self.error(_("%s option requires %d arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001472 % (opt, nargs))
1473 elif nargs == 1:
1474 value = rargs.pop(0)
1475 else:
1476 value = tuple(rargs[0:nargs])
1477 del rargs[0:nargs]
1478
1479 elif had_explicit_value:
Greg Wardeba20e62004-07-31 16:15:44 +00001480 self.error(_("%s option does not take a value") % opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001481
1482 else:
1483 value = None
1484
1485 option.process(opt, value, values, self)
1486
Greg Wardeba20e62004-07-31 16:15:44 +00001487 def _process_short_opts(self, rargs, values):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001488 arg = rargs.pop(0)
Greg Wardeba20e62004-07-31 16:15:44 +00001489 stop = False
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001490 i = 1
1491 for ch in arg[1:]:
1492 opt = "-" + ch
1493 option = self._short_opt.get(opt)
1494 i += 1 # we have consumed a character
1495
1496 if not option:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001497 raise BadOptionError(opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001498 if option.takes_value():
1499 # Any characters left in arg? Pretend they're the
1500 # next arg, and stop consuming characters of arg.
1501 if i < len(arg):
1502 rargs.insert(0, arg[i:])
Greg Wardeba20e62004-07-31 16:15:44 +00001503 stop = True
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001504
1505 nargs = option.nargs
1506 if len(rargs) < nargs:
1507 if nargs == 1:
Greg Wardeba20e62004-07-31 16:15:44 +00001508 self.error(_("%s option requires an argument") % opt)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001509 else:
Greg Wardeba20e62004-07-31 16:15:44 +00001510 self.error(_("%s option requires %d arguments")
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001511 % (opt, nargs))
1512 elif nargs == 1:
1513 value = rargs.pop(0)
1514 else:
1515 value = tuple(rargs[0:nargs])
1516 del rargs[0:nargs]
1517
1518 else: # option doesn't take a value
1519 value = None
1520
1521 option.process(opt, value, values, self)
1522
1523 if stop:
1524 break
1525
1526
1527 # -- Feedback methods ----------------------------------------------
1528
Greg Wardeba20e62004-07-31 16:15:44 +00001529 def get_prog_name(self):
1530 if self.prog is None:
1531 return os.path.basename(sys.argv[0])
1532 else:
1533 return self.prog
1534
1535 def expand_prog_name(self, s):
1536 return s.replace("%prog", self.get_prog_name())
1537
1538 def get_description(self):
1539 return self.expand_prog_name(self.description)
1540
Greg Ward48aa84b2004-10-27 02:20:04 +00001541 def exit(self, status=0, msg=None):
1542 if msg:
1543 sys.stderr.write(msg)
1544 sys.exit(status)
1545
Greg Wardeba20e62004-07-31 16:15:44 +00001546 def error(self, msg):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001547 """error(msg : string)
1548
1549 Print a usage message incorporating 'msg' to stderr and exit.
1550 If you override this in a subclass, it should not return -- it
1551 should either exit or raise an exception.
1552 """
1553 self.print_usage(sys.stderr)
Greg Ward48aa84b2004-10-27 02:20:04 +00001554 self.exit(2, "%s: error: %s\n" % (self.get_prog_name(), msg))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001555
Greg Wardeba20e62004-07-31 16:15:44 +00001556 def get_usage(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001557 if self.usage:
1558 return self.formatter.format_usage(
Greg Wardeba20e62004-07-31 16:15:44 +00001559 self.expand_prog_name(self.usage))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001560 else:
1561 return ""
1562
Greg Wardeba20e62004-07-31 16:15:44 +00001563 def print_usage(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001564 """print_usage(file : file = stdout)
1565
1566 Print the usage message for the current program (self.usage) to
1567 'file' (default stdout). Any occurence of the string "%prog" in
1568 self.usage is replaced with the name of the current program
1569 (basename of sys.argv[0]). Does nothing if self.usage is empty
1570 or not defined.
1571 """
1572 if self.usage:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001573 print(self.get_usage(), file=file)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001574
Greg Wardeba20e62004-07-31 16:15:44 +00001575 def get_version(self):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001576 if self.version:
Greg Wardeba20e62004-07-31 16:15:44 +00001577 return self.expand_prog_name(self.version)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001578 else:
1579 return ""
1580
Greg Wardeba20e62004-07-31 16:15:44 +00001581 def print_version(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001582 """print_version(file : file = stdout)
1583
1584 Print the version message for this program (self.version) to
1585 'file' (default stdout). As with print_usage(), any occurence
1586 of "%prog" in self.version is replaced by the current program's
1587 name. Does nothing if self.version is empty or undefined.
1588 """
1589 if self.version:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001590 print(self.get_version(), file=file)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001591
Greg Wardeba20e62004-07-31 16:15:44 +00001592 def format_option_help(self, formatter=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001593 if formatter is None:
1594 formatter = self.formatter
1595 formatter.store_option_strings(self)
1596 result = []
Thomas Wouters477c8d52006-05-27 19:21:47 +00001597 result.append(formatter.format_heading(_("Options")))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001598 formatter.indent()
1599 if self.option_list:
1600 result.append(OptionContainer.format_option_help(self, formatter))
1601 result.append("\n")
1602 for group in self.option_groups:
1603 result.append(group.format_help(formatter))
1604 result.append("\n")
1605 formatter.dedent()
1606 # Drop the last "\n", or the header if no options or option groups:
1607 return "".join(result[:-1])
1608
Thomas Wouters477c8d52006-05-27 19:21:47 +00001609 def format_epilog(self, formatter):
1610 return formatter.format_epilog(self.epilog)
1611
Greg Wardeba20e62004-07-31 16:15:44 +00001612 def format_help(self, formatter=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001613 if formatter is None:
1614 formatter = self.formatter
1615 result = []
1616 if self.usage:
1617 result.append(self.get_usage() + "\n")
1618 if self.description:
1619 result.append(self.format_description(formatter) + "\n")
1620 result.append(self.format_option_help(formatter))
Thomas Wouters477c8d52006-05-27 19:21:47 +00001621 result.append(self.format_epilog(formatter))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001622 return "".join(result)
1623
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001624 # used by test suite
1625 def _get_encoding(self, file):
1626 encoding = getattr(file, "encoding", None)
1627 if not encoding:
1628 encoding = sys.getdefaultencoding()
1629 return encoding
1630
Greg Wardeba20e62004-07-31 16:15:44 +00001631 def print_help(self, file=None):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001632 """print_help(file : file = stdout)
1633
1634 Print an extended help message, listing all options and any
1635 help text provided with them, to 'file' (default stdout).
1636 """
1637 if file is None:
1638 file = sys.stdout
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001639 encoding = self._get_encoding(file)
1640 file.write(self.format_help().encode(encoding, "replace"))
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001641
1642# class OptionParser
1643
1644
Greg Wardeba20e62004-07-31 16:15:44 +00001645def _match_abbrev(s, wordmap):
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001646 """_match_abbrev(s : string, wordmap : {string : Option}) -> string
1647
1648 Return the string key in 'wordmap' for which 's' is an unambiguous
1649 abbreviation. If 's' is found to be ambiguous or doesn't match any of
1650 'words', raise BadOptionError.
1651 """
1652 # Is there an exact match?
Guido van Rossume2b70bc2006-08-18 22:13:04 +00001653 if s in wordmap:
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001654 return s
1655 else:
1656 # Isolate all words with s as a prefix.
1657 possibilities = [word for word in wordmap.keys()
1658 if word.startswith(s)]
1659 # No exact match, so there had better be just one possibility.
1660 if len(possibilities) == 1:
1661 return possibilities[0]
1662 elif not possibilities:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001663 raise BadOptionError(s)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001664 else:
1665 # More than one possible completion: ambiguous prefix.
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001666 possibilities.sort()
Thomas Wouters477c8d52006-05-27 19:21:47 +00001667 raise AmbiguousOptionError(s, possibilities)
Guido van Rossumb9ba4582002-11-14 22:00:19 +00001668
1669
1670# Some day, there might be many Option classes. As of Optik 1.3, the
1671# preferred way to instantiate Options is indirectly, via make_option(),
1672# which will become a factory function when there are many Option
1673# classes.
1674make_option = Option