← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/option.py

  • Committer: John Arbash Meinel
  • Date: 2006-08-17 13:53:36 UTC
  • mfrom: (1939 +trunk)
  • mto: This revision was merged to the branch mainline in revision 1940.
  • Revision ID: john@arbash-meinel.com-20060817135336-100de82962355d3b
[merge] bzr.dev 1939

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/option.py
17
17
# TODO: For things like --diff-prefix, we want a way to customize the display
18
18
# of the option argument.
19
19
 
 
20
import optparse
20
21
import re
21
22
 
22
23
from bzrlib.trace import warning
153
154
            if option is self:
154
155
                return short
155
156
 
 
157
    def get_negation_name(self):
 
158
        if self.name.startswith('no-'):
 
159
            return self.name[3:]
 
160
        else:
 
161
            return 'no-' + self.name
 
162
 
 
163
    def add_option(self, parser, short_name):
 
164
        """Add this option to an Optparse parser"""
 
165
        option_strings = ['--%s' % self.name]
 
166
        if short_name is not None:
 
167
            option_strings.append('-%s' % short_name)
 
168
        optargfn = self.type
 
169
        if optargfn is None:
 
170
            parser.add_option(action='store_true', dest=self.name, 
 
171
                              help=self.help,
 
172
                              default=OptionParser.DEFAULT_VALUE,
 
173
                              *option_strings)
 
174
            negation_strings = ['--%s' % self.get_negation_name()]
 
175
            parser.add_option(action='store_false', dest=self.name, 
 
176
                              help=optparse.SUPPRESS_HELP, *negation_strings)
 
177
        else:
 
178
            parser.add_option(action='callback', 
 
179
                              callback=self._optparse_callback, 
 
180
                              type='string', metavar=self.argname.upper(),
 
181
                              help=self.help,
 
182
                              default=OptionParser.DEFAULT_VALUE, 
 
183
                              *option_strings)
 
184
 
 
185
    def _optparse_callback(self, option, opt, value, parser):
 
186
        setattr(parser.values, self.name, self.type(value))
 
187
 
 
188
    def iter_switches(self):
 
189
        """Iterate through the list of switches provided by the option
 
190
        
 
191
        :return: an iterator of (name, short_name, argname, help)
 
192
        """
 
193
        argname =  self.argname
 
194
        if argname is not None:
 
195
            argname = argname.upper()
 
196
        yield self.name, self.short_name(), argname, self.help
 
197
 
 
198
 
 
199
class OptionParser(optparse.OptionParser):
 
200
    """OptionParser that raises exceptions instead of exiting"""
 
201
 
 
202
    DEFAULT_VALUE = object()
 
203
 
 
204
    def error(self, message):
 
205
        raise BzrCommandError(message)
 
206
 
 
207
 
 
208
def get_optparser(options):
 
209
    """Generate an optparse parser for bzrlib-style options"""
 
210
 
 
211
    parser = OptionParser()
 
212
    parser.remove_option('--help')
 
213
    short_options = dict((k.name, v) for v, k in 
 
214
                         Option.SHORT_OPTIONS.iteritems())
 
215
    for option in options.itervalues():
 
216
        option.add_option(parser, short_options.get(option.name))
 
217
    return parser
 
218
 
156
219
 
157
220
def _global_option(name, **kwargs):
158
221
    """Register o as a global option."""