← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/option.py

  • Committer: Robert Collins
  • Date: 2006-05-18 23:48:54 UTC
  • mto: (1721.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 1722.
  • Revision ID: robertc@robertcollins.net-20060518234854-a4b689eef47e5ba0
Smoke tests for tree.list_files and bzr ignored when a versioned file matches an ignore rule.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/option.py
1
 
# Copyright (C) 2004, 2005, 2006 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2004, 2005, 2006 by Canonical Ltd
 
2
 
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19
19
 
20
20
import re
21
21
 
22
 
from bzrlib.lazy_import import lazy_import
23
 
lazy_import(globals(), """
24
 
import optparse
25
 
 
26
 
from bzrlib import (
27
 
    errors,
28
 
    log,
29
 
    registry,
30
 
    revisionspec,
31
 
    symbol_versioning,
32
 
    )
33
 
""")
34
 
from bzrlib.trace import warning
 
22
import bzrlib.commands
 
23
from bzrlib.trace import warning, mutter
 
24
from bzrlib.revisionspec import RevisionSpec
 
25
from bzrlib.errors import BzrCommandError
35
26
 
36
27
 
37
28
def _parse_revision_str(revstr):
41
32
    each revision specifier supplied.
42
33
 
43
34
    >>> _parse_revision_str('234')
44
 
    [<RevisionSpec_revno 234>]
 
35
    [<RevisionSpec_int 234>]
45
36
    >>> _parse_revision_str('234..567')
46
 
    [<RevisionSpec_revno 234>, <RevisionSpec_revno 567>]
 
37
    [<RevisionSpec_int 234>, <RevisionSpec_int 567>]
47
38
    >>> _parse_revision_str('..')
48
39
    [<RevisionSpec None>, <RevisionSpec None>]
49
40
    >>> _parse_revision_str('..234')
50
 
    [<RevisionSpec None>, <RevisionSpec_revno 234>]
 
41
    [<RevisionSpec None>, <RevisionSpec_int 234>]
51
42
    >>> _parse_revision_str('234..')
52
 
    [<RevisionSpec_revno 234>, <RevisionSpec None>]
 
43
    [<RevisionSpec_int 234>, <RevisionSpec None>]
53
44
    >>> _parse_revision_str('234..456..789') # Maybe this should be an error
54
 
    [<RevisionSpec_revno 234>, <RevisionSpec_revno 456>, <RevisionSpec_revno 789>]
 
45
    [<RevisionSpec_int 234>, <RevisionSpec_int 456>, <RevisionSpec_int 789>]
55
46
    >>> _parse_revision_str('234....789') #Error ?
56
 
    [<RevisionSpec_revno 234>, <RevisionSpec None>, <RevisionSpec_revno 789>]
 
47
    [<RevisionSpec_int 234>, <RevisionSpec None>, <RevisionSpec_int 789>]
57
48
    >>> _parse_revision_str('revid:test@other.com-234234')
58
49
    [<RevisionSpec_revid revid:test@other.com-234234>]
59
50
    >>> _parse_revision_str('revid:test@other.com-234234..revid:test@other.com-234235')
60
51
    [<RevisionSpec_revid revid:test@other.com-234234>, <RevisionSpec_revid revid:test@other.com-234235>]
61
52
    >>> _parse_revision_str('revid:test@other.com-234234..23')
62
 
    [<RevisionSpec_revid revid:test@other.com-234234>, <RevisionSpec_revno 23>]
 
53
    [<RevisionSpec_revid revid:test@other.com-234234>, <RevisionSpec_int 23>]
63
54
    >>> _parse_revision_str('date:2005-04-12')
64
55
    [<RevisionSpec_date date:2005-04-12>]
65
56
    >>> _parse_revision_str('date:2005-04-12 12:24:33')
69
60
    >>> _parse_revision_str('date:2005-04-12,12:24:33')
70
61
    [<RevisionSpec_date date:2005-04-12,12:24:33>]
71
62
    >>> _parse_revision_str('-5..23')
72
 
    [<RevisionSpec_revno -5>, <RevisionSpec_revno 23>]
 
63
    [<RevisionSpec_int -5>, <RevisionSpec_int 23>]
73
64
    >>> _parse_revision_str('-5')
74
 
    [<RevisionSpec_revno -5>]
 
65
    [<RevisionSpec_int -5>]
75
66
    >>> _parse_revision_str('123a')
76
67
    Traceback (most recent call last):
77
68
      ...
78
 
    NoSuchRevisionSpec: No namespace registered for string: '123a'
 
69
    BzrError: No namespace registered for string: '123a'
79
70
    >>> _parse_revision_str('abc')
80
71
    Traceback (most recent call last):
81
72
      ...
82
 
    NoSuchRevisionSpec: No namespace registered for string: 'abc'
 
73
    BzrError: No namespace registered for string: 'abc'
83
74
    >>> _parse_revision_str('branch:../branch2')
84
75
    [<RevisionSpec_branch branch:../branch2>]
85
76
    >>> _parse_revision_str('branch:../../branch2')
86
77
    [<RevisionSpec_branch branch:../../branch2>]
87
78
    >>> _parse_revision_str('branch:../../branch2..23')
88
 
    [<RevisionSpec_branch branch:../../branch2>, <RevisionSpec_revno 23>]
 
79
    [<RevisionSpec_branch branch:../../branch2>, <RevisionSpec_int 23>]
89
80
    """
90
81
    # TODO: Maybe move this into revisionspec.py
 
82
    old_format_re = re.compile('\d*:\d*')
 
83
    m = old_format_re.match(revstr)
91
84
    revs = []
92
 
    # split on the first .. that is not followed by a / ?
93
 
    sep = re.compile("\\.\\.(?!/)")
94
 
    for x in sep.split(revstr):
95
 
        revs.append(revisionspec.RevisionSpec.from_string(x or None))
 
85
    if m:
 
86
        warning('Colon separator for revision numbers is deprecated.'
 
87
                ' Use .. instead')
 
88
        for rev in revstr.split(':'):
 
89
            if rev:
 
90
                revs.append(RevisionSpec(int(rev)))
 
91
            else:
 
92
                revs.append(RevisionSpec(None))
 
93
    else:
 
94
        sep = re.compile("\\.\\.(?!/)")
 
95
        for x in sep.split(revstr):
 
96
            revs.append(RevisionSpec(x or None))
96
97
    return revs
97
98
 
98
99
 
110
111
        type_list = '\n'.join(lines)
111
112
        msg = "No known merge type %s. Supported types are:\n%s" %\
112
113
            (typestring, type_list)
113
 
        raise errors.BzrCommandError(msg)
114
 
 
 
114
        raise BzrCommandError(msg)
115
115
 
116
116
class Option(object):
117
 
    """Description of a command line option
118
 
    
119
 
    :ivar _short_name: If this option has a single-letter name, this is it.
120
 
    Otherwise None.
121
 
    """
122
 
 
 
117
    """Description of a command line option"""
123
118
    # TODO: Some way to show in help a description of the option argument
124
119
 
125
120
    OPTIONS = {}
 
121
    SHORT_OPTIONS = {}
126
122
 
127
 
    def __init__(self, name, help='', type=None, argname=None,
128
 
                 short_name=None):
 
123
    def __init__(self, name, help='', type=None, argname=None):
129
124
        """Make a new command option.
130
125
 
131
126
        name -- regular name of the command, used in the double-dash
139
134
 
140
135
        argname -- name of option argument, if any
141
136
        """
 
137
        # TODO: perhaps a subclass that automatically does 
 
138
        # --option, --no-option for reversable booleans
142
139
        self.name = name
143
140
        self.help = help
144
141
        self.type = type
145
 
        self._short_name = short_name
146
142
        if type is None:
147
143
            assert argname is None
148
144
        elif argname is None:
150
146
        self.argname = argname
151
147
 
152
148
    def short_name(self):
153
 
        if self._short_name:
154
 
            return self._short_name
155
 
        else:
156
 
            # remove this when SHORT_OPTIONS is removed
157
 
            # XXX: This is accessing a DeprecatedDict, so we call the super 
158
 
            # method to avoid warnings
159
 
            for (k, v) in dict.iteritems(Option.SHORT_OPTIONS):
160
 
                if v == self:
161
 
                    return k
162
 
 
163
 
    def set_short_name(self, short_name):
164
 
        self._short_name = short_name
165
 
 
166
 
    def get_negation_name(self):
167
 
        if self.name.startswith('no-'):
168
 
            return self.name[3:]
169
 
        else:
170
 
            return 'no-' + self.name
171
 
 
172
 
    def add_option(self, parser, short_name):
173
 
        """Add this option to an Optparse parser"""
174
 
        option_strings = ['--%s' % self.name]
175
 
        if short_name is not None:
176
 
            option_strings.append('-%s' % short_name)
177
 
        optargfn = self.type
178
 
        if optargfn is None:
179
 
            parser.add_option(action='store_true', dest=self.name, 
180
 
                              help=self.help,
181
 
                              default=OptionParser.DEFAULT_VALUE,
182
 
                              *option_strings)
183
 
            negation_strings = ['--%s' % self.get_negation_name()]
184
 
            parser.add_option(action='store_false', dest=self.name, 
185
 
                              help=optparse.SUPPRESS_HELP, *negation_strings)
186
 
        else:
187
 
            parser.add_option(action='callback', 
188
 
                              callback=self._optparse_callback, 
189
 
                              type='string', metavar=self.argname.upper(),
190
 
                              help=self.help,
191
 
                              default=OptionParser.DEFAULT_VALUE, 
192
 
                              *option_strings)
193
 
 
194
 
    def _optparse_callback(self, option, opt, value, parser):
195
 
        setattr(parser.values, self.name, self.type(value))
196
 
 
197
 
    def iter_switches(self):
198
 
        """Iterate through the list of switches provided by the option
199
 
        
200
 
        :return: an iterator of (name, short_name, argname, help)
201
 
        """
202
 
        argname =  self.argname
203
 
        if argname is not None:
204
 
            argname = argname.upper()
205
 
        yield self.name, self.short_name(), argname, self.help
206
 
 
207
 
 
208
 
class ListOption(Option):
209
 
    """Option used to provide a list of values.
210
 
 
211
 
    On the command line, arguments are specified by a repeated use of the
212
 
    option. '-' is a special argument that resets the list. For example,
213
 
      --foo=a --foo=b
214
 
    sets the value of the 'foo' option to ['a', 'b'], and
215
 
      --foo=a --foo=b --foo=- --foo=c
216
 
    sets the value of the 'foo' option to ['c'].
217
 
    """
218
 
 
219
 
    def add_option(self, parser, short_name):
220
 
        """Add this option to an Optparse parser."""
221
 
        option_strings = ['--%s' % self.name]
222
 
        if short_name is not None:
223
 
            option_strings.append('-%s' % short_name)
224
 
        parser.add_option(action='callback',
225
 
                          callback=self._optparse_callback,
226
 
                          type='string', metavar=self.argname.upper(),
227
 
                          help=self.help, default=[],
228
 
                          *option_strings)
229
 
 
230
 
    def _optparse_callback(self, option, opt, value, parser):
231
 
        values = getattr(parser.values, self.name)
232
 
        if value == '-':
233
 
            del values[:]
234
 
        else:
235
 
            values.append(self.type(value))
236
 
 
237
 
 
238
 
class RegistryOption(Option):
239
 
    """Option based on a registry
240
 
 
241
 
    The values for the options correspond to entries in the registry.  Input
242
 
    must be a registry key.  After validation, it is converted into an object
243
 
    using Registry.get or a caller-provided converter.
244
 
    """
245
 
 
246
 
    def validate_value(self, value):
247
 
        """Validate a value name"""
248
 
        if value not in self.registry:
249
 
            raise errors.BadOptionValue(self.name, value)
250
 
 
251
 
    def convert(self, value):
252
 
        """Convert a value name into an output type"""
253
 
        self.validate_value(value)
254
 
        if self.converter is None:
255
 
            return self.registry.get(value)
256
 
        else:
257
 
            return self.converter(value)
258
 
 
259
 
    def __init__(self, name, help, registry, converter=None,
260
 
        value_switches=False, title=None, enum_switch=True):
261
 
        """
262
 
        Constructor.
263
 
 
264
 
        :param name: The option name.
265
 
        :param help: Help for the option.
266
 
        :param registry: A Registry containing the values
267
 
        :param converter: Callable to invoke with the value name to produce
268
 
            the value.  If not supplied, self.registry.get is used.
269
 
        :param value_switches: If true, each possible value is assigned its
270
 
            own switch.  For example, instead of '--format knit',
271
 
            '--knit' can be used interchangeably.
272
 
        :param enum_switch: If true, a switch is provided with the option name,
273
 
            which takes a value.
274
 
        """
275
 
        Option.__init__(self, name, help, type=self.convert)
276
 
        self.registry = registry
277
 
        self.name = name
278
 
        self.converter = converter
279
 
        self.value_switches = value_switches
280
 
        self.enum_switch = enum_switch
281
 
        self.title = title
282
 
        if self.title is None:
283
 
            self.title = name
284
 
 
285
 
    @staticmethod
286
 
    def from_kwargs(name_, help=None, title=None, value_switches=False,
287
 
                    enum_switch=True, **kwargs):
288
 
        """Convenience method to generate string-map registry options
289
 
 
290
 
        name, help, value_switches and enum_switch are passed to the
291
 
        RegistryOption constructor.  Any other keyword arguments are treated
292
 
        as values for the option, and they value is treated as the help.
293
 
        """
294
 
        reg = registry.Registry()
295
 
        for name, help in kwargs.iteritems():
296
 
            name = name.replace('_', '-')
297
 
            reg.register(name, name, help=help)
298
 
        return RegistryOption(name_, help, reg, title=title,
299
 
            value_switches=value_switches, enum_switch=enum_switch)
300
 
 
301
 
    def add_option(self, parser, short_name):
302
 
        """Add this option to an Optparse parser"""
303
 
        if self.value_switches:
304
 
            parser = parser.add_option_group(self.title)
305
 
        if self.enum_switch:
306
 
            Option.add_option(self, parser, short_name)
307
 
        if self.value_switches:
308
 
            for key in self.registry.keys():
309
 
                option_strings = ['--%s' % key]
310
 
                if getattr(self.registry.get_info(key), 'hidden', False):
311
 
                    help = optparse.SUPPRESS_HELP
312
 
                else:
313
 
                    help = self.registry.get_help(key)
314
 
                parser.add_option(action='callback',
315
 
                              callback=self._optparse_value_callback(key),
316
 
                                  help=help,
317
 
                                  *option_strings)
318
 
 
319
 
    def _optparse_value_callback(self, cb_value):
320
 
        def cb(option, opt, value, parser):
321
 
            setattr(parser.values, self.name, self.type(cb_value))
322
 
        return cb
323
 
 
324
 
    def iter_switches(self):
325
 
        """Iterate through the list of switches provided by the option
326
 
 
327
 
        :return: an iterator of (name, short_name, argname, help)
328
 
        """
329
 
        for value in Option.iter_switches(self):
330
 
            yield value
331
 
        if self.value_switches:
332
 
            for key in sorted(self.registry.keys()):
333
 
                yield key, None, None, self.registry.get_help(key)
334
 
 
335
 
 
336
 
class OptionParser(optparse.OptionParser):
337
 
    """OptionParser that raises exceptions instead of exiting"""
338
 
 
339
 
    DEFAULT_VALUE = object()
340
 
 
341
 
    def error(self, message):
342
 
        raise errors.BzrCommandError(message)
343
 
 
344
 
 
345
 
def get_optparser(options):
346
 
    """Generate an optparse parser for bzrlib-style options"""
347
 
 
348
 
    parser = OptionParser()
349
 
    parser.remove_option('--help')
350
 
    for option in options.itervalues():
351
 
        option.add_option(parser, option.short_name())
352
 
    return parser
 
149
        """Return the single character option for this command, if any.
 
150
 
 
151
        Short options are globally registered.
 
152
        """
 
153
        for short, option in Option.SHORT_OPTIONS.iteritems():
 
154
            if option is self:
 
155
                return short
353
156
 
354
157
 
355
158
def _global_option(name, **kwargs):
356
159
    """Register o as a global option."""
357
160
    Option.OPTIONS[name] = Option(name, **kwargs)
358
161
 
359
 
 
360
 
def _global_registry_option(name, help, registry, **kwargs):
361
 
    Option.OPTIONS[name] = RegistryOption(name, help, registry, **kwargs)
362
 
 
363
 
 
364
 
class MergeTypeRegistry(registry.Registry):
365
 
 
366
 
    pass
367
 
 
368
 
 
369
 
_merge_type_registry = MergeTypeRegistry()
370
 
_merge_type_registry.register_lazy('merge3', 'bzrlib.merge', 'Merge3Merger',
371
 
                                   "Native diff3-style merge")
372
 
_merge_type_registry.register_lazy('diff3', 'bzrlib.merge', 'Diff3Merger',
373
 
                                   "Merge using external diff3")
374
 
_merge_type_registry.register_lazy('weave', 'bzrlib.merge', 'WeaveMerger',
375
 
                                   "Weave-based merge")
376
 
 
377
162
_global_option('all')
378
163
_global_option('overwrite', help='Ignore differences between branches and '
379
164
               'overwrite unconditionally')
381
166
_global_option('bound')
382
167
_global_option('diff-options', type=str)
383
168
_global_option('help',
384
 
               help='show help message',
385
 
               short_name='h')
386
 
_global_option('file', type=unicode, short_name='F')
 
169
               help='show help message')
 
170
_global_option('file', type=unicode)
387
171
_global_option('force')
388
172
_global_option('format', type=unicode)
389
173
_global_option('forward')
390
 
_global_option('message', type=unicode,
391
 
               short_name='m')
 
174
_global_option('message', type=unicode)
392
175
_global_option('no-recurse')
 
176
_global_option('prefix', type=str, 
 
177
               help='Set prefixes to added to old and new filenames, as '
 
178
                    'two values separated by a colon.')
393
179
_global_option('profile',
394
180
               help='show performance profiling information')
395
 
_global_option('revision',
396
 
               type=_parse_revision_str,
397
 
               short_name='r',
398
 
               help='See \'help revisionspec\' for details')
 
181
_global_option('revision', type=_parse_revision_str)
399
182
_global_option('show-ids', 
400
183
               help='show internal object ids')
401
184
_global_option('timezone', 
403
186
               help='display timezone as local, original, or utc')
404
187
_global_option('unbound')
405
188
_global_option('verbose',
406
 
               help='display more information',
407
 
               short_name='v')
 
189
               help='display more information')
408
190
_global_option('version')
409
191
_global_option('email')
410
192
_global_option('update')
411
 
_global_registry_option('log-format', "Use this log format",
412
 
                        log.log_formatter_registry, value_switches=True,
413
 
                        title='Log format')
414
 
_global_option('long', help='Use detailed log format. Same as --log-format long',
415
 
               short_name='l')
 
193
_global_option('log-format', type=str, help="Use this log format")
 
194
_global_option('long', help='Use detailed log format. Same as --log-format long')
416
195
_global_option('short', help='Use moderately short log format. Same as --log-format short')
417
196
_global_option('line', help='Use log format with one line per revision. Same as --log-format line')
418
197
_global_option('root', type=str)
419
198
_global_option('no-backup')
420
 
_global_registry_option('merge-type', 'Select a particular merge algorithm',
421
 
                        _merge_type_registry, value_switches=True,
422
 
                        title='Merge algorithm')
 
199
_global_option('merge-type', type=_parse_merge_type)
423
200
_global_option('pattern', type=str)
424
 
_global_option('quiet', short_name='q')
 
201
_global_option('quiet')
425
202
_global_option('remember', help='Remember the specified location as a'
426
203
               ' default.')
427
204
_global_option('reprocess', help='Reprocess to reduce spurious conflicts')
428
205
_global_option('kind', type=str)
429
206
_global_option('dry-run',
430
207
               help="show what would be done, but don't actually do anything")
431
 
_global_option('name-from-revision', help='The path name in the old tree.')
432
 
 
433
 
 
434
 
# prior to 0.14 these were always globally registered; the old dict is
435
 
# available for plugins that use it but it should not be used.
436
 
Option.SHORT_OPTIONS = symbol_versioning.DeprecatedDict(
437
 
    symbol_versioning.zero_fourteen,
438
 
    'SHORT_OPTIONS',
439
 
    {
440
 
        'F': Option.OPTIONS['file'],
441
 
        'h': Option.OPTIONS['help'],
442
 
        'm': Option.OPTIONS['message'],
443
 
        'r': Option.OPTIONS['revision'],
444
 
        'v': Option.OPTIONS['verbose'],
445
 
        'l': Option.OPTIONS['long'],
446
 
        'q': Option.OPTIONS['quiet'],
447
 
    },
448
 
    'Set the short option name when constructing the Option.',
449
 
    )
 
208
 
 
209
 
 
210
def _global_short(short_name, long_name):
 
211
    assert short_name not in Option.SHORT_OPTIONS
 
212
    Option.SHORT_OPTIONS[short_name] = Option.OPTIONS[long_name]
 
213
    
 
214
 
 
215
Option.SHORT_OPTIONS['F'] = Option.OPTIONS['file']
 
216
Option.SHORT_OPTIONS['h'] = Option.OPTIONS['help']
 
217
Option.SHORT_OPTIONS['m'] = Option.OPTIONS['message']
 
218
Option.SHORT_OPTIONS['r'] = Option.OPTIONS['revision']
 
219
Option.SHORT_OPTIONS['v'] = Option.OPTIONS['verbose']
 
220
Option.SHORT_OPTIONS['l'] = Option.OPTIONS['long']
 
221
Option.SHORT_OPTIONS['q'] = Option.OPTIONS['quiet']
 
222
Option.SHORT_OPTIONS['p'] = Option.OPTIONS['prefix']