home *** CD-ROM | disk | FTP | other *** search
- # Source Generated with Decompyle++
- # File: in.pyc (Python 2.6)
-
- """Parser for command line options.
-
- This module helps scripts to parse the command line arguments in
- sys.argv. It supports the same conventions as the Unix getopt()
- function (including the special meanings of arguments of the form `-'
- and `--'). Long options similar to those supported by GNU software
- may be used as well via an optional third argument. This module
- provides two functions and an exception:
-
- getopt() -- Parse command line options
- gnu_getopt() -- Like getopt(), but allow option and non-option arguments
- to be intermixed.
- GetoptError -- exception (class) raised with 'opt' attribute, which is the
- option involved with the exception.
- """
- __all__ = [
- 'GetoptError',
- 'error',
- 'getopt',
- 'gnu_getopt']
- import os
-
- class GetoptError(Exception):
- opt = ''
- msg = ''
-
- def __init__(self, msg, opt = ''):
- self.msg = msg
- self.opt = opt
- Exception.__init__(self, msg, opt)
-
-
- def __str__(self):
- return self.msg
-
-
- error = GetoptError
-
- def getopt(args, shortopts, longopts = []):
- '''getopt(args, options[, long_options]) -> opts, args
-
- Parses command line options and parameter list. args is the
- argument list to be parsed, without the leading reference to the
- running program. Typically, this means "sys.argv[1:]". shortopts
- is the string of option letters that the script wants to
- recognize, with options that require an argument followed by a
- colon (i.e., the same format that Unix getopt() uses). If
- specified, longopts is a list of strings with the names of the
- long options which should be supported. The leading \'--\'
- characters should not be included in the option name. Options
- which require an argument should be followed by an equal sign
- (\'=\').
-
- The return value consists of two elements: the first is a list of
- (option, value) pairs; the second is the list of program arguments
- left after the option list was stripped (this is a trailing slice
- of the first argument). Each option-and-value pair returned has
- the option as its first element, prefixed with a hyphen (e.g.,
- \'-x\'), and the option argument as its second element, or an empty
- string if the option has no argument. The options occur in the
- list in the same order in which they were found, thus allowing
- multiple occurrences. Long and short options may be mixed.
-
- '''
- opts = []
- if type(longopts) == type(''):
- longopts = [
- longopts]
- else:
- longopts = list(longopts)
- while args and args[0].startswith('-') and args[0] != '-':
- if args[0] == '--':
- args = args[1:]
- break
-
- if args[0].startswith('--'):
- (opts, args) = do_longs(opts, args[0][2:], longopts, args[1:])
- continue
- (opts, args) = do_shorts(opts, args[0][1:], shortopts, args[1:])
- return (opts, args)
-
-
- def gnu_getopt(args, shortopts, longopts = []):
- """getopt(args, options[, long_options]) -> opts, args
-
- This function works like getopt(), except that GNU style scanning
- mode is used by default. This means that option and non-option
- arguments may be intermixed. The getopt() function stops
- processing options as soon as a non-option argument is
- encountered.
-
- If the first character of the option string is `+', or if the
- environment variable POSIXLY_CORRECT is set, then option
- processing stops as soon as a non-option argument is encountered.
-
- """
- opts = []
- prog_args = []
- if isinstance(longopts, str):
- longopts = [
- longopts]
- else:
- longopts = list(longopts)
- if shortopts.startswith('+'):
- shortopts = shortopts[1:]
- all_options_first = True
- elif os.environ.get('POSIXLY_CORRECT'):
- all_options_first = True
- else:
- all_options_first = False
- while args:
- if args[0] == '--':
- prog_args += args[1:]
- break
-
- if args[0][:2] == '--':
- (opts, args) = do_longs(opts, args[0][2:], longopts, args[1:])
- continue
- if args[0][:1] == '-':
- (opts, args) = do_shorts(opts, args[0][1:], shortopts, args[1:])
- continue
- if all_options_first:
- prog_args += args
- break
- continue
- prog_args.append(args[0])
- args = args[1:]
- return (opts, prog_args)
-
-
- def do_longs(opts, opt, longopts, args):
-
- try:
- i = opt.index('=')
- except ValueError:
- optarg = None
-
- opt = opt[:i]
- optarg = opt[i + 1:]
- (has_arg, opt) = long_has_args(opt, longopts)
- if has_arg:
- if optarg is None:
- if not args:
- raise GetoptError('option --%s requires argument' % opt, opt)
- args
- optarg = args[0]
- args = args[1:]
-
- elif optarg:
- raise GetoptError('option --%s must not have an argument' % opt, opt)
-
- if not optarg:
- pass
- opts.append(('--' + opt, ''))
- return (opts, args)
-
-
- def long_has_args(opt, longopts):
- possibilities = _[1]
- if not possibilities:
- raise GetoptError('option --%s not recognized' % opt, opt)
- possibilities
- if opt in possibilities:
- return (False, opt)
- if opt + '=' in possibilities:
- return (True, opt)
- if len(possibilities) > 1:
- raise GetoptError('option --%s not a unique prefix' % opt, opt)
- len(possibilities) > 1
- if not len(possibilities) == 1:
- raise AssertionError
- unique_match = possibilities[0]
- has_arg = unique_match.endswith('=')
- return (has_arg, unique_match)
-
-
- def do_shorts(opts, optstring, shortopts, args):
- while optstring != '':
- opt = optstring[0]
- optstring = optstring[1:]
- if short_has_arg(opt, shortopts):
- if optstring == '':
- if not args:
- raise GetoptError('option -%s requires argument' % opt, opt)
- args
- optstring = args[0]
- args = args[1:]
-
- optarg = optstring
- optstring = ''
- else:
- optarg = ''
- opts.append(('-' + opt, optarg))
- return (opts, args)
-
-
- def short_has_arg(opt, shortopts):
- for i in range(len(shortopts)):
- if shortopts[i] == shortopts[i]:
- pass
- elif shortopts[i] != ':':
- return shortopts.startswith(':', i + 1)
-
- raise GetoptError('option -%s not recognized' % opt, opt)
-
- if __name__ == '__main__':
- import sys
- print getopt(sys.argv[1:], 'a:b', [
- 'alpha=',
- 'beta'])
-
-