1# Copyright (c) 2005 The Regents of The University of Michigan 2# All rights reserved. 3# 4# Redistribution and use in source and binary forms, with or without 5# modification, are permitted provided that the following conditions are 6# met: redistributions of source code must retain the above copyright 7# notice, this list of conditions and the following disclaimer; 8# redistributions in binary form must reproduce the above copyright --- 14 unchanged lines hidden (view full) --- 23# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 25# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26# 27# Authors: Nathan Binkert 28 29import optparse 30import sys |
31 32from optparse import * 33 34class nodefault(object): pass 35 36class splitter(object): 37 def __init__(self, split): 38 self.split = split 39 def __call__(self, option, opt_str, value, parser): 40 values = value.split(self.split) 41 dest = getattr(parser.values, option.dest) 42 if dest is None: 43 setattr(parser.values, option.dest, values) 44 else: 45 dest.extend(values) 46 |
47class OptionParser(dict): |
48 def __init__(self, *args, **kwargs): 49 kwargs.setdefault('formatter', optparse.TitledHelpFormatter()) 50 self._optparse = optparse.OptionParser(*args, **kwargs) 51 self._optparse.disable_interspersed_args() 52 53 self._allopts = {} |
54 55 # current option group 56 self._group = self._optparse 57 |
58 def set_defaults(self, *args, **kwargs): 59 return self._optparse.set_defaults(*args, **kwargs) 60 |
61 def set_group(self, *args, **kwargs): 62 '''set the current option group''' 63 if not args and not kwargs: 64 self._group = self._optparse 65 else: 66 self._group = self._optparse.add_option_group(*args, **kwargs) 67 68 def add_option(self, *args, **kwargs): --- 4 unchanged lines hidden (view full) --- 73 74 if kwargs.get('action', None) == 'append' and 'split' in kwargs: 75 split = kwargs.pop('split') 76 kwargs['default'] = [] 77 kwargs['type'] = 'string' 78 kwargs['action'] = 'callback' 79 kwargs['callback'] = splitter(split) 80 |
81 option = self._group.add_option(*args, **kwargs) 82 dest = option.dest 83 if dest not in self._allopts: 84 self._allopts[dest] = option 85 |
86 return option 87 88 def bool_option(self, name, default, help): 89 '''add a boolean option called --name and --no-name. 90 Display help depending on which is the default''' 91 92 tname = '--%s' % name 93 fname = '--no-%s' % name --- 9 unchanged lines hidden (view full) --- 103 help=thelp) 104 fopt = self.add_option(fname, action="store_false", dest=dest, 105 help=fhelp) 106 107 return topt,fopt 108 109 def __getattr__(self, attr): 110 if attr.startswith('_'): |
111 return super(OptionParser, self).__getattribute__(attr) |
112 |
113 if attr in self: 114 return self[attr] |
115 |
116 return super(OptionParser, self).__getattribute__(attr) |
117 118 def __setattr__(self, attr, value): 119 if attr.startswith('_'): |
120 super(OptionParser, self).__setattr__(attr, value) 121 elif attr in self._allopts: 122 defaults = { attr : value } 123 self.set_defaults(**defaults) 124 if attr in self: 125 self[attr] = value 126 else: 127 super(OptionParser, self).__setattr__(attr, value) |
128 |
129 def parse_args(self): 130 opts,args = self._optparse.parse_args() 131 132 for key,val in opts.__dict__.iteritems(): |
133 if val is not None or key not in self: 134 self[key] = val |
135 136 return args 137 138 def usage(self, exitcode=None): 139 self._optparse.print_help() 140 if exitcode is not None: 141 sys.exit(exitcode) 142 |