~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revisionspec.py

  • Committer: Martin Pool
  • Date: 2009-01-13 03:06:36 UTC
  • mfrom: (3932.2.3 1.11)
  • mto: This revision was merged to the branch mainline in revision 3937.
  • Revision ID: mbp@sourcefrog.net-20090113030636-dqx4t8yaaqgdvam5
MergeĀ 1.11rc1

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
2
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
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
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
from __future__ import absolute_import
18
 
 
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
import re
19
19
 
20
20
from bzrlib.lazy_import import lazy_import
21
21
lazy_import(globals(), """
22
22
import bisect
23
23
import datetime
 
24
""")
24
25
 
25
26
from bzrlib import (
26
 
    branch as _mod_branch,
 
27
    errors,
27
28
    osutils,
28
29
    revision,
29
30
    symbol_versioning,
30
 
    workingtree,
31
 
    )
32
 
from bzrlib.i18n import gettext
33
 
""")
34
 
 
35
 
from bzrlib import (
36
 
    errors,
37
 
    lazy_regex,
38
 
    registry,
39
31
    trace,
40
32
    )
41
33
 
42
34
 
 
35
_marker = []
 
36
 
 
37
 
43
38
class RevisionInfo(object):
44
39
    """The results of applying a revision specification to a branch."""
45
40
 
57
52
    or treat the result as a tuple.
58
53
    """
59
54
 
60
 
    def __init__(self, branch, revno=None, rev_id=None):
 
55
    def __init__(self, branch, revno, rev_id=_marker):
61
56
        self.branch = branch
62
 
        self._has_revno = (revno is not None)
63
 
        self._revno = revno
64
 
        self.rev_id = rev_id
65
 
        if self.rev_id is None and self._revno is not None:
 
57
        self.revno = revno
 
58
        if rev_id is _marker:
66
59
            # allow caller to be lazy
67
 
            self.rev_id = branch.get_rev_id(self._revno)
68
 
 
69
 
    @property
70
 
    def revno(self):
71
 
        if not self._has_revno and self.rev_id is not None:
72
 
            try:
73
 
                self._revno = self.branch.revision_id_to_revno(self.rev_id)
74
 
            except errors.NoSuchRevision:
75
 
                self._revno = None
76
 
            self._has_revno = True
77
 
        return self._revno
 
60
            if self.revno is None:
 
61
                self.rev_id = None
 
62
            else:
 
63
                self.rev_id = branch.get_rev_id(self.revno)
 
64
        else:
 
65
            self.rev_id = rev_id
78
66
 
79
67
    def __nonzero__(self):
 
68
        # first the easy ones...
80
69
        if self.rev_id is None:
81
70
            return False
 
71
        if self.revno is not None:
 
72
            return True
82
73
        # TODO: otherwise, it should depend on how I was built -
83
74
        # if it's in_history(branch), then check revision_history(),
84
75
        # if it's in_store(branch), do the check below
107
98
            self.revno, self.rev_id, self.branch)
108
99
 
109
100
    @staticmethod
110
 
    def from_revision_id(branch, revision_id, revs=symbol_versioning.DEPRECATED_PARAMETER):
 
101
    def from_revision_id(branch, revision_id, revs):
111
102
        """Construct a RevisionInfo given just the id.
112
103
 
113
104
        Use this if you don't know or care what the revno is.
114
105
        """
115
 
        if symbol_versioning.deprecated_passed(revs):
116
 
            symbol_versioning.warn(
117
 
                'RevisionInfo.from_revision_id(revs) was deprecated in 2.5.',
118
 
                DeprecationWarning,
119
 
                stacklevel=2)
120
 
        return RevisionInfo(branch, revno=None, rev_id=revision_id)
 
106
        if revision_id == revision.NULL_REVISION:
 
107
            return RevisionInfo(branch, 0, revision_id)
 
108
        try:
 
109
            revno = revs.index(revision_id) + 1
 
110
        except ValueError:
 
111
            revno = None
 
112
        return RevisionInfo(branch, revno, revision_id)
 
113
 
 
114
 
 
115
# classes in this list should have a "prefix" attribute, against which
 
116
# string specs are matched
 
117
SPEC_TYPES = []
 
118
_revno_regex = None
121
119
 
122
120
 
123
121
class RevisionSpec(object):
125
123
 
126
124
    help_txt = """A parsed revision specification.
127
125
 
128
 
    A revision specification is a string, which may be unambiguous about
129
 
    what it represents by giving a prefix like 'date:' or 'revid:' etc,
130
 
    or it may have no prefix, in which case it's tried against several
131
 
    specifier types in sequence to determine what the user meant.
 
126
    A revision specification can be an integer, in which case it is
 
127
    assumed to be a revno (though this will translate negative values
 
128
    into positive ones); or it can be a string, in which case it is
 
129
    parsed for something like 'date:' or 'revid:' etc.
132
130
 
133
131
    Revision specs are an UI element, and they have been moved out
134
132
    of the branch class to leave "back-end" classes unaware of such
140
138
    """
141
139
 
142
140
    prefix = None
143
 
    # wants_revision_history has been deprecated in 2.5.
144
 
    wants_revision_history = False
145
 
    dwim_catchable_exceptions = (errors.InvalidRevisionSpec,)
146
 
    """Exceptions that RevisionSpec_dwim._match_on will catch.
147
 
 
148
 
    If the revspec is part of ``dwim_revspecs``, it may be tried with an
149
 
    invalid revspec and raises some exception. The exceptions mentioned here
150
 
    will not be reported to the user but simply ignored without stopping the
151
 
    dwim processing.
152
 
    """
 
141
    wants_revision_history = True
153
142
 
154
143
    @staticmethod
155
144
    def from_string(spec):
164
153
 
165
154
        if spec is None:
166
155
            return RevisionSpec(None, _internal=True)
167
 
        match = revspec_registry.get_prefix(spec)
168
 
        if match is not None:
169
 
            spectype, specsuffix = match
170
 
            trace.mutter('Returning RevisionSpec %s for %s',
171
 
                         spectype.__name__, spec)
172
 
            return spectype(spec, _internal=True)
 
156
        for spectype in SPEC_TYPES:
 
157
            if spec.startswith(spectype.prefix):
 
158
                trace.mutter('Returning RevisionSpec %s for %s',
 
159
                             spectype.__name__, spec)
 
160
                return spectype(spec, _internal=True)
173
161
        else:
174
 
            # Otherwise treat it as a DWIM, build the RevisionSpec object and
175
 
            # wait for _match_on to be called.
176
 
            return RevisionSpec_dwim(spec, _internal=True)
 
162
            # RevisionSpec_revno is special cased, because it is the only
 
163
            # one that directly handles plain integers
 
164
            # TODO: This should not be special cased rather it should be
 
165
            # a method invocation on spectype.canparse()
 
166
            global _revno_regex
 
167
            if _revno_regex is None:
 
168
                _revno_regex = re.compile(r'^(?:(\d+(\.\d+)*)|-\d+)(:.*)?$')
 
169
            if _revno_regex.match(spec) is not None:
 
170
                return RevisionSpec_revno(spec, _internal=True)
 
171
 
 
172
            raise errors.NoSuchRevisionSpec(spec)
177
173
 
178
174
    def __init__(self, spec, _internal=False):
179
175
        """Create a RevisionSpec referring to the Null revision.
183
179
            called directly. Only from RevisionSpec.from_string()
184
180
        """
185
181
        if not _internal:
 
182
            # XXX: Update this after 0.10 is released
186
183
            symbol_versioning.warn('Creating a RevisionSpec directly has'
187
184
                                   ' been deprecated in version 0.11. Use'
188
185
                                   ' RevisionSpec.from_string()'
212
209
    def in_history(self, branch):
213
210
        if branch:
214
211
            if self.wants_revision_history:
215
 
                symbol_versioning.warn(
216
 
                    "RevisionSpec.wants_revision_history was "
217
 
                    "deprecated in 2.5 (%s)." % self.__class__.__name__,
218
 
                    DeprecationWarning)
219
 
                branch.lock_read()
220
 
                try:
221
 
                    graph = branch.repository.get_graph()
222
 
                    revs = list(graph.iter_lefthand_ancestry(
223
 
                        branch.last_revision(), [revision.NULL_REVISION]))
224
 
                finally:
225
 
                    branch.unlock()
226
 
                revs.reverse()
 
212
                revs = branch.revision_history()
227
213
            else:
228
214
                revs = None
229
215
        else:
281
267
        # this is mostly for helping with testing
282
268
        return '<%s %s>' % (self.__class__.__name__,
283
269
                              self.user_spec)
284
 
 
 
270
    
285
271
    def needs_branch(self):
286
272
        """Whether this revision spec needs a branch.
287
273
 
291
277
 
292
278
    def get_branch(self):
293
279
        """When the revision specifier contains a branch location, return it.
294
 
 
 
280
        
295
281
        Otherwise, return None.
296
282
        """
297
283
        return None
299
285
 
300
286
# private API
301
287
 
302
 
class RevisionSpec_dwim(RevisionSpec):
303
 
    """Provides a DWIMish revision specifier lookup.
304
 
 
305
 
    Note that this does not go in the revspec_registry because by definition
306
 
    there is no prefix to identify it.  It's solely called from
307
 
    RevisionSpec.from_string() because the DWIMification happen when _match_on
308
 
    is called so the string describing the revision is kept here until needed.
309
 
    """
310
 
 
311
 
    help_txt = None
312
 
 
313
 
    _revno_regex = lazy_regex.lazy_compile(r'^(?:(\d+(\.\d+)*)|-\d+)(:.*)?$')
314
 
 
315
 
    # The revspecs to try
316
 
    _possible_revspecs = []
317
 
 
318
 
    def _try_spectype(self, rstype, branch):
319
 
        rs = rstype(self.spec, _internal=True)
320
 
        # Hit in_history to find out if it exists, or we need to try the
321
 
        # next type.
322
 
        return rs.in_history(branch)
323
 
 
324
 
    def _match_on(self, branch, revs):
325
 
        """Run the lookup and see what we can get."""
326
 
 
327
 
        # First, see if it's a revno
328
 
        if self._revno_regex.match(self.spec) is not None:
329
 
            try:
330
 
                return self._try_spectype(RevisionSpec_revno, branch)
331
 
            except RevisionSpec_revno.dwim_catchable_exceptions:
332
 
                pass
333
 
 
334
 
        # Next see what has been registered
335
 
        for objgetter in self._possible_revspecs:
336
 
            rs_class = objgetter.get_obj()
337
 
            try:
338
 
                return self._try_spectype(rs_class, branch)
339
 
            except rs_class.dwim_catchable_exceptions:
340
 
                pass
341
 
 
342
 
        # Try the old (deprecated) dwim list:
343
 
        for rs_class in dwim_revspecs:
344
 
            try:
345
 
                return self._try_spectype(rs_class, branch)
346
 
            except rs_class.dwim_catchable_exceptions:
347
 
                pass
348
 
 
349
 
        # Well, I dunno what it is. Note that we don't try to keep track of the
350
 
        # first of last exception raised during the DWIM tries as none seems
351
 
        # really relevant.
352
 
        raise errors.InvalidRevisionSpec(self.spec, branch)
353
 
 
354
 
    @classmethod
355
 
    def append_possible_revspec(cls, revspec):
356
 
        """Append a possible DWIM revspec.
357
 
 
358
 
        :param revspec: Revision spec to try.
359
 
        """
360
 
        cls._possible_revspecs.append(registry._ObjectGetter(revspec))
361
 
 
362
 
    @classmethod
363
 
    def append_possible_lazy_revspec(cls, module_name, member_name):
364
 
        """Append a possible lazily loaded DWIM revspec.
365
 
 
366
 
        :param module_name: Name of the module with the revspec
367
 
        :param member_name: Name of the revspec within the module
368
 
        """
369
 
        cls._possible_revspecs.append(
370
 
            registry._LazyObjectGetter(module_name, member_name))
371
 
 
372
 
 
373
288
class RevisionSpec_revno(RevisionSpec):
374
289
    """Selects a revision using a number."""
375
290
 
376
291
    help_txt = """Selects a revision using a number.
377
292
 
378
293
    Use an integer to specify a revision in the history of the branch.
379
 
    Optionally a branch can be specified.  A negative number will count
380
 
    from the end of the branch (-1 is the last revision, -2 the previous
381
 
    one). If the negative number is larger than the branch's history, the
382
 
    first revision is returned.
 
294
    Optionally a branch can be specified. The 'revno:' prefix is optional.
 
295
    A negative number will count from the end of the branch (-1 is the
 
296
    last revision, -2 the previous one). If the negative number is larger
 
297
    than the branch's history, the first revision is returned.
383
298
    Examples::
384
299
 
385
300
      revno:1                   -> return the first revision of this branch
392
307
                                   your history is very long.
393
308
    """
394
309
    prefix = 'revno:'
 
310
    wants_revision_history = False
395
311
 
396
312
    def _match_on(self, branch, revs):
397
313
        """Lookup a revision by revision number"""
398
 
        branch, revno, revision_id = self._lookup(branch)
 
314
        branch, revno, revision_id = self._lookup(branch, revs)
399
315
        return RevisionInfo(branch, revno, revision_id)
400
316
 
401
 
    def _lookup(self, branch):
 
317
    def _lookup(self, branch, revs_or_none):
402
318
        loc = self.spec.find(':')
403
319
        if loc == -1:
404
320
            revno_spec = self.spec
418
334
                dotted = False
419
335
            except ValueError:
420
336
                # dotted decimal. This arguably should not be here
421
 
                # but the from_string method is a little primitive
 
337
                # but the from_string method is a little primitive 
422
338
                # right now - RBC 20060928
423
339
                try:
424
340
                    match_revno = tuple((int(number) for number in revno_spec.split('.')))
428
344
                dotted = True
429
345
 
430
346
        if branch_spec:
431
 
            # the user has overriden the branch to look in.
432
 
            branch = _mod_branch.Branch.open(branch_spec)
 
347
            # the user has override the branch to look in.
 
348
            # we need to refresh the revision_history map and
 
349
            # the branch object.
 
350
            from bzrlib.branch import Branch
 
351
            branch = Branch.open(branch_spec)
 
352
            revs_or_none = None
433
353
 
434
354
        if dotted:
 
355
            branch.lock_read()
435
356
            try:
436
 
                revision_id = branch.dotted_revno_to_revision_id(match_revno,
437
 
                    _cache_reverse=True)
438
 
            except errors.NoSuchRevision:
 
357
                revision_id_to_revno = branch.get_revision_id_to_revno_map()
 
358
                revisions = [revision_id for revision_id, revno
 
359
                             in revision_id_to_revno.iteritems()
 
360
                             if revno == match_revno]
 
361
            finally:
 
362
                branch.unlock()
 
363
            if len(revisions) != 1:
439
364
                raise errors.InvalidRevisionSpec(self.user_spec, branch)
440
365
            else:
441
366
                # there is no traditional 'revno' for dotted-decimal revnos.
442
 
                # so for API compatibility we return None.
443
 
                return branch, None, revision_id
 
367
                # so for  API compatability we return None.
 
368
                return branch, None, revisions[0]
444
369
        else:
445
370
            last_revno, last_revision_id = branch.last_revision_info()
446
371
            if revno < 0:
451
376
                else:
452
377
                    revno = last_revno + revno + 1
453
378
            try:
454
 
                revision_id = branch.get_rev_id(revno)
 
379
                revision_id = branch.get_rev_id(revno, revs_or_none)
455
380
            except errors.NoSuchRevision:
456
381
                raise errors.InvalidRevisionSpec(self.user_spec, branch)
457
382
        return branch, revno, revision_id
458
383
 
459
384
    def _as_revision_id(self, context_branch):
460
385
        # We would have the revno here, but we don't really care
461
 
        branch, revno, revision_id = self._lookup(context_branch)
 
386
        branch, revno, revision_id = self._lookup(context_branch, None)
462
387
        return revision_id
463
388
 
464
389
    def needs_branch(self):
470
395
        else:
471
396
            return self.spec[self.spec.find(':')+1:]
472
397
 
473
 
# Old compatibility
 
398
# Old compatibility 
474
399
RevisionSpec_int = RevisionSpec_revno
475
400
 
476
 
 
477
 
class RevisionIDSpec(RevisionSpec):
478
 
 
479
 
    def _match_on(self, branch, revs):
480
 
        revision_id = self.as_revision_id(branch)
481
 
        return RevisionInfo.from_revision_id(branch, revision_id)
482
 
 
483
 
 
484
 
class RevisionSpec_revid(RevisionIDSpec):
 
401
SPEC_TYPES.append(RevisionSpec_revno)
 
402
 
 
403
 
 
404
class RevisionSpec_revid(RevisionSpec):
485
405
    """Selects a revision using the revision id."""
486
406
 
487
407
    help_txt = """Selects a revision using the revision id.
488
408
 
489
409
    Supply a specific revision id, that can be used to specify any
490
 
    revision id in the ancestry of the branch.
 
410
    revision id in the ancestry of the branch. 
491
411
    Including merges, and pending merges.
492
412
    Examples::
493
413
 
496
416
 
497
417
    prefix = 'revid:'
498
418
 
499
 
    def _as_revision_id(self, context_branch):
 
419
    def _match_on(self, branch, revs):
500
420
        # self.spec comes straight from parsing the command line arguments,
501
421
        # so we expect it to be a Unicode string. Switch it to the internal
502
422
        # representation.
 
423
        revision_id = osutils.safe_revision_id(self.spec, warn=False)
 
424
        return RevisionInfo.from_revision_id(branch, revision_id, revs)
 
425
 
 
426
    def _as_revision_id(self, context_branch):
503
427
        return osutils.safe_revision_id(self.spec, warn=False)
504
428
 
 
429
SPEC_TYPES.append(RevisionSpec_revid)
505
430
 
506
431
 
507
432
class RevisionSpec_last(RevisionSpec):
520
445
    prefix = 'last:'
521
446
 
522
447
    def _match_on(self, branch, revs):
523
 
        revno, revision_id = self._revno_and_revision_id(branch)
 
448
        revno, revision_id = self._revno_and_revision_id(branch, revs)
524
449
        return RevisionInfo(branch, revno, revision_id)
525
450
 
526
 
    def _revno_and_revision_id(self, context_branch):
 
451
    def _revno_and_revision_id(self, context_branch, revs_or_none):
527
452
        last_revno, last_revision_id = context_branch.last_revision_info()
528
453
 
529
454
        if self.spec == '':
542
467
 
543
468
        revno = last_revno - offset + 1
544
469
        try:
545
 
            revision_id = context_branch.get_rev_id(revno)
 
470
            revision_id = context_branch.get_rev_id(revno, revs_or_none)
546
471
        except errors.NoSuchRevision:
547
472
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch)
548
473
        return revno, revision_id
550
475
    def _as_revision_id(self, context_branch):
551
476
        # We compute the revno as part of the process, but we don't really care
552
477
        # about it.
553
 
        revno, revision_id = self._revno_and_revision_id(context_branch)
 
478
        revno, revision_id = self._revno_and_revision_id(context_branch, None)
554
479
        return revision_id
555
480
 
 
481
SPEC_TYPES.append(RevisionSpec_last)
556
482
 
557
483
 
558
484
class RevisionSpec_before(RevisionSpec):
578
504
    """
579
505
 
580
506
    prefix = 'before:'
581
 
 
 
507
    
582
508
    def _match_on(self, branch, revs):
583
509
        r = RevisionSpec.from_string(self.spec)._match_on(branch, revs)
584
510
        if r.revno == 0:
588
514
            # We need to use the repository history here
589
515
            rev = branch.repository.get_revision(r.rev_id)
590
516
            if not rev.parent_ids:
 
517
                revno = 0
591
518
                revision_id = revision.NULL_REVISION
592
519
            else:
593
520
                revision_id = rev.parent_ids[0]
594
 
            revno = None
 
521
                try:
 
522
                    revno = revs.index(revision_id) + 1
 
523
                except ValueError:
 
524
                    revno = None
595
525
        else:
596
526
            revno = r.revno - 1
597
527
            try:
602
532
        return RevisionInfo(branch, revno, revision_id)
603
533
 
604
534
    def _as_revision_id(self, context_branch):
605
 
        base_revision_id = RevisionSpec.from_string(self.spec)._as_revision_id(context_branch)
 
535
        base_revspec = RevisionSpec.from_string(self.spec)
 
536
        base_revision_id = base_revspec.as_revision_id(context_branch)
606
537
        if base_revision_id == revision.NULL_REVISION:
607
538
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
608
539
                                         'cannot go before the null: revision')
622
553
                'No parents for revision.')
623
554
        return parents[0]
624
555
 
 
556
SPEC_TYPES.append(RevisionSpec_before)
625
557
 
626
558
 
627
559
class RevisionSpec_tag(RevisionSpec):
633
565
    """
634
566
 
635
567
    prefix = 'tag:'
636
 
    dwim_catchable_exceptions = (errors.NoSuchTag, errors.TagsNotSupported)
637
568
 
638
569
    def _match_on(self, branch, revs):
639
570
        # Can raise tags not supported, NoSuchTag, etc
640
571
        return RevisionInfo.from_revision_id(branch,
641
 
            branch.tags.lookup_tag(self.spec))
 
572
            branch.tags.lookup_tag(self.spec),
 
573
            revs)
642
574
 
643
575
    def _as_revision_id(self, context_branch):
644
576
        return context_branch.tags.lookup_tag(self.spec)
645
577
 
 
578
SPEC_TYPES.append(RevisionSpec_tag)
646
579
 
647
580
 
648
581
class _RevListToTimestamps(object):
649
582
    """This takes a list of revisions, and allows you to bisect by date"""
650
583
 
651
 
    __slots__ = ['branch']
 
584
    __slots__ = ['revs', 'branch']
652
585
 
653
 
    def __init__(self, branch):
 
586
    def __init__(self, revs, branch):
 
587
        self.revs = revs
654
588
        self.branch = branch
655
589
 
656
590
    def __getitem__(self, index):
657
591
        """Get the date of the index'd item"""
658
 
        r = self.branch.repository.get_revision(self.branch.get_rev_id(index))
 
592
        r = self.branch.repository.get_revision(self.revs[index])
659
593
        # TODO: Handle timezone.
660
594
        return datetime.datetime.fromtimestamp(r.timestamp)
661
595
 
662
596
    def __len__(self):
663
 
        return self.branch.revno()
 
597
        return len(self.revs)
664
598
 
665
599
 
666
600
class RevisionSpec_date(RevisionSpec):
682
616
      date:yesterday            -> select the first revision since yesterday
683
617
      date:2006-08-14,17:10:14  -> select the first revision after
684
618
                                   August 14th, 2006 at 5:10pm.
685
 
    """
 
619
    """    
686
620
    prefix = 'date:'
687
 
    _date_regex = lazy_regex.lazy_compile(
 
621
    _date_re = re.compile(
688
622
            r'(?P<date>(?P<year>\d\d\d\d)-(?P<month>\d\d)-(?P<day>\d\d))?'
689
623
            r'(,|T)?\s*'
690
624
            r'(?P<time>(?P<hour>\d\d):(?P<minute>\d\d)(:(?P<second>\d\d))?)?'
708
642
        elif self.spec.lower() == 'tomorrow':
709
643
            dt = today + datetime.timedelta(days=1)
710
644
        else:
711
 
            m = self._date_regex.match(self.spec)
 
645
            m = self._date_re.match(self.spec)
712
646
            if not m or (not m.group('date') and not m.group('time')):
713
647
                raise errors.InvalidRevisionSpec(self.user_spec,
714
648
                                                 branch, 'invalid date')
740
674
                    hour=hour, minute=minute, second=second)
741
675
        branch.lock_read()
742
676
        try:
743
 
            rev = bisect.bisect(_RevListToTimestamps(branch), dt, 1)
 
677
            rev = bisect.bisect(_RevListToTimestamps(revs, branch), dt)
744
678
        finally:
745
679
            branch.unlock()
746
 
        if rev == branch.revno():
 
680
        if rev == len(revs):
747
681
            raise errors.InvalidRevisionSpec(self.user_spec, branch)
748
 
        return RevisionInfo(branch, rev)
 
682
        else:
 
683
            return RevisionInfo(branch, rev + 1)
749
684
 
 
685
SPEC_TYPES.append(RevisionSpec_date)
750
686
 
751
687
 
752
688
class RevisionSpec_ancestor(RevisionSpec):
782
718
    def _find_revision_info(branch, other_location):
783
719
        revision_id = RevisionSpec_ancestor._find_revision_id(branch,
784
720
                                                              other_location)
785
 
        return RevisionInfo(branch, None, revision_id)
 
721
        try:
 
722
            revno = branch.revision_id_to_revno(revision_id)
 
723
        except errors.NoSuchRevision:
 
724
            revno = None
 
725
        return RevisionInfo(branch, revno, revision_id)
786
726
 
787
727
    @staticmethod
788
728
    def _find_revision_id(branch, other_location):
793
733
            revision_a = revision.ensure_null(branch.last_revision())
794
734
            if revision_a == revision.NULL_REVISION:
795
735
                raise errors.NoCommits(branch)
796
 
            if other_location == '':
797
 
                other_location = branch.get_parent()
798
736
            other_branch = Branch.open(other_location)
799
737
            other_branch.lock_read()
800
738
            try:
812
750
            branch.unlock()
813
751
 
814
752
 
 
753
SPEC_TYPES.append(RevisionSpec_ancestor)
815
754
 
816
755
 
817
756
class RevisionSpec_branch(RevisionSpec):
826
765
      branch:/path/to/branch
827
766
    """
828
767
    prefix = 'branch:'
829
 
    dwim_catchable_exceptions = (errors.NotBranchError,)
830
768
 
831
769
    def _match_on(self, branch, revs):
832
770
        from bzrlib.branch import Branch
834
772
        revision_b = other_branch.last_revision()
835
773
        if revision_b in (None, revision.NULL_REVISION):
836
774
            raise errors.NoCommits(other_branch)
837
 
        if branch is None:
838
 
            branch = other_branch
839
 
        else:
840
 
            try:
841
 
                # pull in the remote revisions so we can diff
842
 
                branch.fetch(other_branch, revision_b)
843
 
            except errors.ReadOnlyError:
844
 
                branch = other_branch
845
 
        return RevisionInfo(branch, None, revision_b)
 
775
        # pull in the remote revisions so we can diff
 
776
        branch.fetch(other_branch, revision_b)
 
777
        try:
 
778
            revno = branch.revision_id_to_revno(revision_b)
 
779
        except errors.NoSuchRevision:
 
780
            revno = None
 
781
        return RevisionInfo(branch, revno, revision_b)
846
782
 
847
783
    def _as_revision_id(self, context_branch):
848
784
        from bzrlib.branch import Branch
863
799
            raise errors.NoCommits(other_branch)
864
800
        return other_branch.repository.revision_tree(last_revision)
865
801
 
866
 
    def needs_branch(self):
867
 
        return False
868
 
 
869
 
    def get_branch(self):
870
 
        return self.spec
871
 
 
 
802
SPEC_TYPES.append(RevisionSpec_branch)
872
803
 
873
804
 
874
805
class RevisionSpec_submit(RevisionSpec_ancestor):
900
831
            location_type = 'parent branch'
901
832
        if submit_location is None:
902
833
            raise errors.NoSubmitBranch(branch)
903
 
        trace.note(gettext('Using {0} {1}').format(location_type,
904
 
                                                        submit_location))
 
834
        trace.note('Using %s %s', location_type, submit_location)
905
835
        return submit_location
906
836
 
907
837
    def _match_on(self, branch, revs):
914
844
            self._get_submit_location(context_branch))
915
845
 
916
846
 
917
 
class RevisionSpec_annotate(RevisionIDSpec):
918
 
 
919
 
    prefix = 'annotate:'
920
 
 
921
 
    help_txt = """Select the revision that last modified the specified line.
922
 
 
923
 
    Select the revision that last modified the specified line.  Line is
924
 
    specified as path:number.  Path is a relative path to the file.  Numbers
925
 
    start at 1, and are relative to the current version, not the last-
926
 
    committed version of the file.
927
 
    """
928
 
 
929
 
    def _raise_invalid(self, numstring, context_branch):
930
 
        raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
931
 
            'No such line: %s' % numstring)
932
 
 
933
 
    def _as_revision_id(self, context_branch):
934
 
        path, numstring = self.spec.rsplit(':', 1)
935
 
        try:
936
 
            index = int(numstring) - 1
937
 
        except ValueError:
938
 
            self._raise_invalid(numstring, context_branch)
939
 
        tree, file_path = workingtree.WorkingTree.open_containing(path)
940
 
        tree.lock_read()
941
 
        try:
942
 
            file_id = tree.path2id(file_path)
943
 
            if file_id is None:
944
 
                raise errors.InvalidRevisionSpec(self.user_spec,
945
 
                    context_branch, "File '%s' is not versioned." %
946
 
                    file_path)
947
 
            revision_ids = [r for (r, l) in tree.annotate_iter(file_id)]
948
 
        finally:
949
 
            tree.unlock()
950
 
        try:
951
 
            revision_id = revision_ids[index]
952
 
        except IndexError:
953
 
            self._raise_invalid(numstring, context_branch)
954
 
        if revision_id == revision.CURRENT_REVISION:
955
 
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
956
 
                'Line %s has not been committed.' % numstring)
957
 
        return revision_id
958
 
 
959
 
 
960
 
class RevisionSpec_mainline(RevisionIDSpec):
961
 
 
962
 
    help_txt = """Select mainline revision that merged the specified revision.
963
 
 
964
 
    Select the revision that merged the specified revision into mainline.
965
 
    """
966
 
 
967
 
    prefix = 'mainline:'
968
 
 
969
 
    def _as_revision_id(self, context_branch):
970
 
        revspec = RevisionSpec.from_string(self.spec)
971
 
        if revspec.get_branch() is None:
972
 
            spec_branch = context_branch
973
 
        else:
974
 
            spec_branch = _mod_branch.Branch.open(revspec.get_branch())
975
 
        revision_id = revspec.as_revision_id(spec_branch)
976
 
        graph = context_branch.repository.get_graph()
977
 
        result = graph.find_lefthand_merger(revision_id,
978
 
                                            context_branch.last_revision())
979
 
        if result is None:
980
 
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch)
981
 
        return result
982
 
 
983
 
 
984
 
# The order in which we want to DWIM a revision spec without any prefix.
985
 
# revno is always tried first and isn't listed here, this is used by
986
 
# RevisionSpec_dwim._match_on
987
 
dwim_revspecs = symbol_versioning.deprecated_list(
988
 
    symbol_versioning.deprecated_in((2, 4, 0)), "dwim_revspecs", [])
989
 
 
990
 
RevisionSpec_dwim.append_possible_revspec(RevisionSpec_tag)
991
 
RevisionSpec_dwim.append_possible_revspec(RevisionSpec_revid)
992
 
RevisionSpec_dwim.append_possible_revspec(RevisionSpec_date)
993
 
RevisionSpec_dwim.append_possible_revspec(RevisionSpec_branch)
994
 
 
995
 
revspec_registry = registry.Registry()
996
 
def _register_revspec(revspec):
997
 
    revspec_registry.register(revspec.prefix, revspec)
998
 
 
999
 
_register_revspec(RevisionSpec_revno)
1000
 
_register_revspec(RevisionSpec_revid)
1001
 
_register_revspec(RevisionSpec_last)
1002
 
_register_revspec(RevisionSpec_before)
1003
 
_register_revspec(RevisionSpec_tag)
1004
 
_register_revspec(RevisionSpec_date)
1005
 
_register_revspec(RevisionSpec_ancestor)
1006
 
_register_revspec(RevisionSpec_branch)
1007
 
_register_revspec(RevisionSpec_submit)
1008
 
_register_revspec(RevisionSpec_annotate)
1009
 
_register_revspec(RevisionSpec_mainline)
 
847
SPEC_TYPES.append(RevisionSpec_submit)