~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revisionspec.py

  • Committer: Martin Pool
  • Date: 2009-06-19 06:21:13 UTC
  • mto: This revision was merged to the branch mainline in revision 4558.
  • Revision ID: mbp@sourcefrog.net-20090619062113-019bp0a3bl2y4nkx
Un-soft-deprecate _supports_progress - still useful

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
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
 
 
18
import re
 
19
 
 
20
from bzrlib.lazy_import import lazy_import
 
21
lazy_import(globals(), """
 
22
import bisect
 
23
import datetime
 
24
""")
 
25
 
 
26
from bzrlib import (
 
27
    errors,
 
28
    osutils,
 
29
    registry,
 
30
    revision,
 
31
    symbol_versioning,
 
32
    trace,
 
33
    )
 
34
 
 
35
 
 
36
_marker = []
 
37
 
 
38
 
 
39
class RevisionInfo(object):
 
40
    """The results of applying a revision specification to a branch."""
 
41
 
 
42
    help_txt = """The results of applying a revision specification to a branch.
 
43
 
 
44
    An instance has two useful attributes: revno, and rev_id.
 
45
 
 
46
    They can also be accessed as spec[0] and spec[1] respectively,
 
47
    so that you can write code like:
 
48
    revno, rev_id = RevisionSpec(branch, spec)
 
49
    although this is probably going to be deprecated later.
 
50
 
 
51
    This class exists mostly to be the return value of a RevisionSpec,
 
52
    so that you can access the member you're interested in (number or id)
 
53
    or treat the result as a tuple.
 
54
    """
 
55
 
 
56
    def __init__(self, branch, revno, rev_id=_marker):
 
57
        self.branch = branch
 
58
        self.revno = revno
 
59
        if rev_id is _marker:
 
60
            # allow caller to be lazy
 
61
            if self.revno is None:
 
62
                self.rev_id = None
 
63
            else:
 
64
                self.rev_id = branch.get_rev_id(self.revno)
 
65
        else:
 
66
            self.rev_id = rev_id
 
67
 
 
68
    def __nonzero__(self):
 
69
        # first the easy ones...
 
70
        if self.rev_id is None:
 
71
            return False
 
72
        if self.revno is not None:
 
73
            return True
 
74
        # TODO: otherwise, it should depend on how I was built -
 
75
        # if it's in_history(branch), then check revision_history(),
 
76
        # if it's in_store(branch), do the check below
 
77
        return self.branch.repository.has_revision(self.rev_id)
 
78
 
 
79
    def __len__(self):
 
80
        return 2
 
81
 
 
82
    def __getitem__(self, index):
 
83
        if index == 0: return self.revno
 
84
        if index == 1: return self.rev_id
 
85
        raise IndexError(index)
 
86
 
 
87
    def get(self):
 
88
        return self.branch.repository.get_revision(self.rev_id)
 
89
 
 
90
    def __eq__(self, other):
 
91
        if type(other) not in (tuple, list, type(self)):
 
92
            return False
 
93
        if type(other) is type(self) and self.branch is not other.branch:
 
94
            return False
 
95
        return tuple(self) == tuple(other)
 
96
 
 
97
    def __repr__(self):
 
98
        return '<bzrlib.revisionspec.RevisionInfo object %s, %s for %r>' % (
 
99
            self.revno, self.rev_id, self.branch)
 
100
 
 
101
    @staticmethod
 
102
    def from_revision_id(branch, revision_id, revs):
 
103
        """Construct a RevisionInfo given just the id.
 
104
 
 
105
        Use this if you don't know or care what the revno is.
 
106
        """
 
107
        if revision_id == revision.NULL_REVISION:
 
108
            return RevisionInfo(branch, 0, revision_id)
 
109
        try:
 
110
            revno = revs.index(revision_id) + 1
 
111
        except ValueError:
 
112
            revno = None
 
113
        return RevisionInfo(branch, revno, revision_id)
 
114
 
 
115
 
 
116
# classes in this list should have a "prefix" attribute, against which
 
117
# string specs are matched
 
118
_revno_regex = None
 
119
 
 
120
 
 
121
class RevisionSpec(object):
 
122
    """A parsed revision specification."""
 
123
 
 
124
    help_txt = """A parsed revision specification.
 
125
 
 
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.
 
130
 
 
131
    Revision specs are an UI element, and they have been moved out
 
132
    of the branch class to leave "back-end" classes unaware of such
 
133
    details.  Code that gets a revno or rev_id from other code should
 
134
    not be using revision specs - revnos and revision ids are the
 
135
    accepted ways to refer to revisions internally.
 
136
 
 
137
    (Equivalent to the old Branch method get_revision_info())
 
138
    """
 
139
 
 
140
    prefix = None
 
141
    wants_revision_history = True
 
142
 
 
143
    @staticmethod
 
144
    def from_string(spec):
 
145
        """Parse a revision spec string into a RevisionSpec object.
 
146
 
 
147
        :param spec: A string specified by the user
 
148
        :return: A RevisionSpec object that understands how to parse the
 
149
            supplied notation.
 
150
        """
 
151
        if not isinstance(spec, (type(None), basestring)):
 
152
            raise TypeError('error')
 
153
 
 
154
        if spec is None:
 
155
            return RevisionSpec(None, _internal=True)
 
156
        match = revspec_registry.get_prefix(spec)
 
157
        if match is not None:
 
158
            spectype, specsuffix = match
 
159
            trace.mutter('Returning RevisionSpec %s for %s',
 
160
                         spectype.__name__, spec)
 
161
            return spectype(spec, _internal=True)
 
162
        else:
 
163
            for spectype in SPEC_TYPES:
 
164
                trace.mutter('Returning RevisionSpec %s for %s',
 
165
                             spectype.__name__, spec)
 
166
                if spec.startswith(spectype.prefix):
 
167
                    return spectype(spec, _internal=True)
 
168
            # RevisionSpec_revno is special cased, because it is the only
 
169
            # one that directly handles plain integers
 
170
            # TODO: This should not be special cased rather it should be
 
171
            # a method invocation on spectype.canparse()
 
172
            global _revno_regex
 
173
            if _revno_regex is None:
 
174
                _revno_regex = re.compile(r'^(?:(\d+(\.\d+)*)|-\d+)(:.*)?$')
 
175
            if _revno_regex.match(spec) is not None:
 
176
                return RevisionSpec_revno(spec, _internal=True)
 
177
 
 
178
            raise errors.NoSuchRevisionSpec(spec)
 
179
 
 
180
    def __init__(self, spec, _internal=False):
 
181
        """Create a RevisionSpec referring to the Null revision.
 
182
 
 
183
        :param spec: The original spec supplied by the user
 
184
        :param _internal: Used to ensure that RevisionSpec is not being
 
185
            called directly. Only from RevisionSpec.from_string()
 
186
        """
 
187
        if not _internal:
 
188
            # XXX: Update this after 0.10 is released
 
189
            symbol_versioning.warn('Creating a RevisionSpec directly has'
 
190
                                   ' been deprecated in version 0.11. Use'
 
191
                                   ' RevisionSpec.from_string()'
 
192
                                   ' instead.',
 
193
                                   DeprecationWarning, stacklevel=2)
 
194
        self.user_spec = spec
 
195
        if self.prefix and spec.startswith(self.prefix):
 
196
            spec = spec[len(self.prefix):]
 
197
        self.spec = spec
 
198
 
 
199
    def _match_on(self, branch, revs):
 
200
        trace.mutter('Returning RevisionSpec._match_on: None')
 
201
        return RevisionInfo(branch, None, None)
 
202
 
 
203
    def _match_on_and_check(self, branch, revs):
 
204
        info = self._match_on(branch, revs)
 
205
        if info:
 
206
            return info
 
207
        elif info == (None, None):
 
208
            # special case - nothing supplied
 
209
            return info
 
210
        elif self.prefix:
 
211
            raise errors.InvalidRevisionSpec(self.user_spec, branch)
 
212
        else:
 
213
            raise errors.InvalidRevisionSpec(self.spec, branch)
 
214
 
 
215
    def in_history(self, branch):
 
216
        if branch:
 
217
            if self.wants_revision_history:
 
218
                revs = branch.revision_history()
 
219
            else:
 
220
                revs = None
 
221
        else:
 
222
            # this should never trigger.
 
223
            # TODO: make it a deprecated code path. RBC 20060928
 
224
            revs = None
 
225
        return self._match_on_and_check(branch, revs)
 
226
 
 
227
        # FIXME: in_history is somewhat broken,
 
228
        # it will return non-history revisions in many
 
229
        # circumstances. The expected facility is that
 
230
        # in_history only returns revision-history revs,
 
231
        # in_store returns any rev. RBC 20051010
 
232
    # aliases for now, when we fix the core logic, then they
 
233
    # will do what you expect.
 
234
    in_store = in_history
 
235
    in_branch = in_store
 
236
 
 
237
    def as_revision_id(self, context_branch):
 
238
        """Return just the revision_id for this revisions spec.
 
239
 
 
240
        Some revision specs require a context_branch to be able to determine
 
241
        their value. Not all specs will make use of it.
 
242
        """
 
243
        return self._as_revision_id(context_branch)
 
244
 
 
245
    def _as_revision_id(self, context_branch):
 
246
        """Implementation of as_revision_id()
 
247
 
 
248
        Classes should override this function to provide appropriate
 
249
        functionality. The default is to just call '.in_history().rev_id'
 
250
        """
 
251
        return self.in_history(context_branch).rev_id
 
252
 
 
253
    def as_tree(self, context_branch):
 
254
        """Return the tree object for this revisions spec.
 
255
 
 
256
        Some revision specs require a context_branch to be able to determine
 
257
        the revision id and access the repository. Not all specs will make
 
258
        use of it.
 
259
        """
 
260
        return self._as_tree(context_branch)
 
261
 
 
262
    def _as_tree(self, context_branch):
 
263
        """Implementation of as_tree().
 
264
 
 
265
        Classes should override this function to provide appropriate
 
266
        functionality. The default is to just call '.as_revision_id()'
 
267
        and get the revision tree from context_branch's repository.
 
268
        """
 
269
        revision_id = self.as_revision_id(context_branch)
 
270
        return context_branch.repository.revision_tree(revision_id)
 
271
 
 
272
    def __repr__(self):
 
273
        # this is mostly for helping with testing
 
274
        return '<%s %s>' % (self.__class__.__name__,
 
275
                              self.user_spec)
 
276
 
 
277
    def needs_branch(self):
 
278
        """Whether this revision spec needs a branch.
 
279
 
 
280
        Set this to False the branch argument of _match_on is not used.
 
281
        """
 
282
        return True
 
283
 
 
284
    def get_branch(self):
 
285
        """When the revision specifier contains a branch location, return it.
 
286
 
 
287
        Otherwise, return None.
 
288
        """
 
289
        return None
 
290
 
 
291
 
 
292
# private API
 
293
 
 
294
class RevisionSpec_revno(RevisionSpec):
 
295
    """Selects a revision using a number."""
 
296
 
 
297
    help_txt = """Selects a revision using a number.
 
298
 
 
299
    Use an integer to specify a revision in the history of the branch.
 
300
    Optionally a branch can be specified. The 'revno:' prefix is optional.
 
301
    A negative number will count from the end of the branch (-1 is the
 
302
    last revision, -2 the previous one). If the negative number is larger
 
303
    than the branch's history, the first revision is returned.
 
304
    Examples::
 
305
 
 
306
      revno:1                   -> return the first revision of this branch
 
307
      revno:3:/path/to/branch   -> return the 3rd revision of
 
308
                                   the branch '/path/to/branch'
 
309
      revno:-1                  -> The last revision in a branch.
 
310
      -2:http://other/branch    -> The second to last revision in the
 
311
                                   remote branch.
 
312
      -1000000                  -> Most likely the first revision, unless
 
313
                                   your history is very long.
 
314
    """
 
315
    prefix = 'revno:'
 
316
    wants_revision_history = False
 
317
 
 
318
    def _match_on(self, branch, revs):
 
319
        """Lookup a revision by revision number"""
 
320
        branch, revno, revision_id = self._lookup(branch, revs)
 
321
        return RevisionInfo(branch, revno, revision_id)
 
322
 
 
323
    def _lookup(self, branch, revs_or_none):
 
324
        loc = self.spec.find(':')
 
325
        if loc == -1:
 
326
            revno_spec = self.spec
 
327
            branch_spec = None
 
328
        else:
 
329
            revno_spec = self.spec[:loc]
 
330
            branch_spec = self.spec[loc+1:]
 
331
 
 
332
        if revno_spec == '':
 
333
            if not branch_spec:
 
334
                raise errors.InvalidRevisionSpec(self.user_spec,
 
335
                        branch, 'cannot have an empty revno and no branch')
 
336
            revno = None
 
337
        else:
 
338
            try:
 
339
                revno = int(revno_spec)
 
340
                dotted = False
 
341
            except ValueError:
 
342
                # dotted decimal. This arguably should not be here
 
343
                # but the from_string method is a little primitive
 
344
                # right now - RBC 20060928
 
345
                try:
 
346
                    match_revno = tuple((int(number) for number in revno_spec.split('.')))
 
347
                except ValueError, e:
 
348
                    raise errors.InvalidRevisionSpec(self.user_spec, branch, e)
 
349
 
 
350
                dotted = True
 
351
 
 
352
        if branch_spec:
 
353
            # the user has override the branch to look in.
 
354
            # we need to refresh the revision_history map and
 
355
            # the branch object.
 
356
            from bzrlib.branch import Branch
 
357
            branch = Branch.open(branch_spec)
 
358
            revs_or_none = None
 
359
 
 
360
        if dotted:
 
361
            try:
 
362
                revision_id = branch.dotted_revno_to_revision_id(match_revno,
 
363
                    _cache_reverse=True)
 
364
            except errors.NoSuchRevision:
 
365
                raise errors.InvalidRevisionSpec(self.user_spec, branch)
 
366
            else:
 
367
                # there is no traditional 'revno' for dotted-decimal revnos.
 
368
                # so for  API compatability we return None.
 
369
                return branch, None, revision_id
 
370
        else:
 
371
            last_revno, last_revision_id = branch.last_revision_info()
 
372
            if revno < 0:
 
373
                # if get_rev_id supported negative revnos, there would not be a
 
374
                # need for this special case.
 
375
                if (-revno) >= last_revno:
 
376
                    revno = 1
 
377
                else:
 
378
                    revno = last_revno + revno + 1
 
379
            try:
 
380
                revision_id = branch.get_rev_id(revno, revs_or_none)
 
381
            except errors.NoSuchRevision:
 
382
                raise errors.InvalidRevisionSpec(self.user_spec, branch)
 
383
        return branch, revno, revision_id
 
384
 
 
385
    def _as_revision_id(self, context_branch):
 
386
        # We would have the revno here, but we don't really care
 
387
        branch, revno, revision_id = self._lookup(context_branch, None)
 
388
        return revision_id
 
389
 
 
390
    def needs_branch(self):
 
391
        return self.spec.find(':') == -1
 
392
 
 
393
    def get_branch(self):
 
394
        if self.spec.find(':') == -1:
 
395
            return None
 
396
        else:
 
397
            return self.spec[self.spec.find(':')+1:]
 
398
 
 
399
# Old compatibility
 
400
RevisionSpec_int = RevisionSpec_revno
 
401
 
 
402
 
 
403
 
 
404
class RevisionSpec_revid(RevisionSpec):
 
405
    """Selects a revision using the revision id."""
 
406
 
 
407
    help_txt = """Selects a revision using the revision id.
 
408
 
 
409
    Supply a specific revision id, that can be used to specify any
 
410
    revision id in the ancestry of the branch.
 
411
    Including merges, and pending merges.
 
412
    Examples::
 
413
 
 
414
      revid:aaaa@bbbb-123456789 -> Select revision 'aaaa@bbbb-123456789'
 
415
    """
 
416
 
 
417
    prefix = 'revid:'
 
418
 
 
419
    def _match_on(self, branch, revs):
 
420
        # self.spec comes straight from parsing the command line arguments,
 
421
        # so we expect it to be a Unicode string. Switch it to the internal
 
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):
 
427
        return osutils.safe_revision_id(self.spec, warn=False)
 
428
 
 
429
 
 
430
 
 
431
class RevisionSpec_last(RevisionSpec):
 
432
    """Selects the nth revision from the end."""
 
433
 
 
434
    help_txt = """Selects the nth revision from the end.
 
435
 
 
436
    Supply a positive number to get the nth revision from the end.
 
437
    This is the same as supplying negative numbers to the 'revno:' spec.
 
438
    Examples::
 
439
 
 
440
      last:1        -> return the last revision
 
441
      last:3        -> return the revision 2 before the end.
 
442
    """
 
443
 
 
444
    prefix = 'last:'
 
445
 
 
446
    def _match_on(self, branch, revs):
 
447
        revno, revision_id = self._revno_and_revision_id(branch, revs)
 
448
        return RevisionInfo(branch, revno, revision_id)
 
449
 
 
450
    def _revno_and_revision_id(self, context_branch, revs_or_none):
 
451
        last_revno, last_revision_id = context_branch.last_revision_info()
 
452
 
 
453
        if self.spec == '':
 
454
            if not last_revno:
 
455
                raise errors.NoCommits(context_branch)
 
456
            return last_revno, last_revision_id
 
457
 
 
458
        try:
 
459
            offset = int(self.spec)
 
460
        except ValueError, e:
 
461
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch, e)
 
462
 
 
463
        if offset <= 0:
 
464
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
 
465
                                             'you must supply a positive value')
 
466
 
 
467
        revno = last_revno - offset + 1
 
468
        try:
 
469
            revision_id = context_branch.get_rev_id(revno, revs_or_none)
 
470
        except errors.NoSuchRevision:
 
471
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch)
 
472
        return revno, revision_id
 
473
 
 
474
    def _as_revision_id(self, context_branch):
 
475
        # We compute the revno as part of the process, but we don't really care
 
476
        # about it.
 
477
        revno, revision_id = self._revno_and_revision_id(context_branch, None)
 
478
        return revision_id
 
479
 
 
480
 
 
481
 
 
482
class RevisionSpec_before(RevisionSpec):
 
483
    """Selects the parent of the revision specified."""
 
484
 
 
485
    help_txt = """Selects the parent of the revision specified.
 
486
 
 
487
    Supply any revision spec to return the parent of that revision.  This is
 
488
    mostly useful when inspecting revisions that are not in the revision history
 
489
    of a branch.
 
490
 
 
491
    It is an error to request the parent of the null revision (before:0).
 
492
 
 
493
    Examples::
 
494
 
 
495
      before:1913    -> Return the parent of revno 1913 (revno 1912)
 
496
      before:revid:aaaa@bbbb-1234567890  -> return the parent of revision
 
497
                                            aaaa@bbbb-1234567890
 
498
      bzr diff -r before:1913..1913
 
499
            -> Find the changes between revision 1913 and its parent (1912).
 
500
               (What changes did revision 1913 introduce).
 
501
               This is equivalent to:  bzr diff -c 1913
 
502
    """
 
503
 
 
504
    prefix = 'before:'
 
505
 
 
506
    def _match_on(self, branch, revs):
 
507
        r = RevisionSpec.from_string(self.spec)._match_on(branch, revs)
 
508
        if r.revno == 0:
 
509
            raise errors.InvalidRevisionSpec(self.user_spec, branch,
 
510
                                         'cannot go before the null: revision')
 
511
        if r.revno is None:
 
512
            # We need to use the repository history here
 
513
            rev = branch.repository.get_revision(r.rev_id)
 
514
            if not rev.parent_ids:
 
515
                revno = 0
 
516
                revision_id = revision.NULL_REVISION
 
517
            else:
 
518
                revision_id = rev.parent_ids[0]
 
519
                try:
 
520
                    revno = revs.index(revision_id) + 1
 
521
                except ValueError:
 
522
                    revno = None
 
523
        else:
 
524
            revno = r.revno - 1
 
525
            try:
 
526
                revision_id = branch.get_rev_id(revno, revs)
 
527
            except errors.NoSuchRevision:
 
528
                raise errors.InvalidRevisionSpec(self.user_spec,
 
529
                                                 branch)
 
530
        return RevisionInfo(branch, revno, revision_id)
 
531
 
 
532
    def _as_revision_id(self, context_branch):
 
533
        base_revspec = RevisionSpec.from_string(self.spec)
 
534
        base_revision_id = base_revspec.as_revision_id(context_branch)
 
535
        if base_revision_id == revision.NULL_REVISION:
 
536
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
 
537
                                         'cannot go before the null: revision')
 
538
        context_repo = context_branch.repository
 
539
        context_repo.lock_read()
 
540
        try:
 
541
            parent_map = context_repo.get_parent_map([base_revision_id])
 
542
        finally:
 
543
            context_repo.unlock()
 
544
        if base_revision_id not in parent_map:
 
545
            # Ghost, or unknown revision id
 
546
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
 
547
                'cannot find the matching revision')
 
548
        parents = parent_map[base_revision_id]
 
549
        if len(parents) < 1:
 
550
            raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
 
551
                'No parents for revision.')
 
552
        return parents[0]
 
553
 
 
554
 
 
555
 
 
556
class RevisionSpec_tag(RevisionSpec):
 
557
    """Select a revision identified by tag name"""
 
558
 
 
559
    help_txt = """Selects a revision identified by a tag name.
 
560
 
 
561
    Tags are stored in the branch and created by the 'tag' command.
 
562
    """
 
563
 
 
564
    prefix = 'tag:'
 
565
 
 
566
    def _match_on(self, branch, revs):
 
567
        # Can raise tags not supported, NoSuchTag, etc
 
568
        return RevisionInfo.from_revision_id(branch,
 
569
            branch.tags.lookup_tag(self.spec),
 
570
            revs)
 
571
 
 
572
    def _as_revision_id(self, context_branch):
 
573
        return context_branch.tags.lookup_tag(self.spec)
 
574
 
 
575
 
 
576
 
 
577
class _RevListToTimestamps(object):
 
578
    """This takes a list of revisions, and allows you to bisect by date"""
 
579
 
 
580
    __slots__ = ['revs', 'branch']
 
581
 
 
582
    def __init__(self, revs, branch):
 
583
        self.revs = revs
 
584
        self.branch = branch
 
585
 
 
586
    def __getitem__(self, index):
 
587
        """Get the date of the index'd item"""
 
588
        r = self.branch.repository.get_revision(self.revs[index])
 
589
        # TODO: Handle timezone.
 
590
        return datetime.datetime.fromtimestamp(r.timestamp)
 
591
 
 
592
    def __len__(self):
 
593
        return len(self.revs)
 
594
 
 
595
 
 
596
class RevisionSpec_date(RevisionSpec):
 
597
    """Selects a revision on the basis of a datestamp."""
 
598
 
 
599
    help_txt = """Selects a revision on the basis of a datestamp.
 
600
 
 
601
    Supply a datestamp to select the first revision that matches the date.
 
602
    Date can be 'yesterday', 'today', 'tomorrow' or a YYYY-MM-DD string.
 
603
    Matches the first entry after a given date (either at midnight or
 
604
    at a specified time).
 
605
 
 
606
    One way to display all the changes since yesterday would be::
 
607
 
 
608
        bzr log -r date:yesterday..
 
609
 
 
610
    Examples::
 
611
 
 
612
      date:yesterday            -> select the first revision since yesterday
 
613
      date:2006-08-14,17:10:14  -> select the first revision after
 
614
                                   August 14th, 2006 at 5:10pm.
 
615
    """
 
616
    prefix = 'date:'
 
617
    _date_re = re.compile(
 
618
            r'(?P<date>(?P<year>\d\d\d\d)-(?P<month>\d\d)-(?P<day>\d\d))?'
 
619
            r'(,|T)?\s*'
 
620
            r'(?P<time>(?P<hour>\d\d):(?P<minute>\d\d)(:(?P<second>\d\d))?)?'
 
621
        )
 
622
 
 
623
    def _match_on(self, branch, revs):
 
624
        """Spec for date revisions:
 
625
          date:value
 
626
          value can be 'yesterday', 'today', 'tomorrow' or a YYYY-MM-DD string.
 
627
          matches the first entry after a given date (either at midnight or
 
628
          at a specified time).
 
629
        """
 
630
        #  XXX: This doesn't actually work
 
631
        #  So the proper way of saying 'give me all entries for today' is:
 
632
        #      -r date:yesterday..date:today
 
633
        today = datetime.datetime.fromordinal(datetime.date.today().toordinal())
 
634
        if self.spec.lower() == 'yesterday':
 
635
            dt = today - datetime.timedelta(days=1)
 
636
        elif self.spec.lower() == 'today':
 
637
            dt = today
 
638
        elif self.spec.lower() == 'tomorrow':
 
639
            dt = today + datetime.timedelta(days=1)
 
640
        else:
 
641
            m = self._date_re.match(self.spec)
 
642
            if not m or (not m.group('date') and not m.group('time')):
 
643
                raise errors.InvalidRevisionSpec(self.user_spec,
 
644
                                                 branch, 'invalid date')
 
645
 
 
646
            try:
 
647
                if m.group('date'):
 
648
                    year = int(m.group('year'))
 
649
                    month = int(m.group('month'))
 
650
                    day = int(m.group('day'))
 
651
                else:
 
652
                    year = today.year
 
653
                    month = today.month
 
654
                    day = today.day
 
655
 
 
656
                if m.group('time'):
 
657
                    hour = int(m.group('hour'))
 
658
                    minute = int(m.group('minute'))
 
659
                    if m.group('second'):
 
660
                        second = int(m.group('second'))
 
661
                    else:
 
662
                        second = 0
 
663
                else:
 
664
                    hour, minute, second = 0,0,0
 
665
            except ValueError:
 
666
                raise errors.InvalidRevisionSpec(self.user_spec,
 
667
                                                 branch, 'invalid date')
 
668
 
 
669
            dt = datetime.datetime(year=year, month=month, day=day,
 
670
                    hour=hour, minute=minute, second=second)
 
671
        branch.lock_read()
 
672
        try:
 
673
            rev = bisect.bisect(_RevListToTimestamps(revs, branch), dt)
 
674
        finally:
 
675
            branch.unlock()
 
676
        if rev == len(revs):
 
677
            raise errors.InvalidRevisionSpec(self.user_spec, branch)
 
678
        else:
 
679
            return RevisionInfo(branch, rev + 1)
 
680
 
 
681
 
 
682
 
 
683
class RevisionSpec_ancestor(RevisionSpec):
 
684
    """Selects a common ancestor with a second branch."""
 
685
 
 
686
    help_txt = """Selects a common ancestor with a second branch.
 
687
 
 
688
    Supply the path to a branch to select the common ancestor.
 
689
 
 
690
    The common ancestor is the last revision that existed in both
 
691
    branches. Usually this is the branch point, but it could also be
 
692
    a revision that was merged.
 
693
 
 
694
    This is frequently used with 'diff' to return all of the changes
 
695
    that your branch introduces, while excluding the changes that you
 
696
    have not merged from the remote branch.
 
697
 
 
698
    Examples::
 
699
 
 
700
      ancestor:/path/to/branch
 
701
      $ bzr diff -r ancestor:../../mainline/branch
 
702
    """
 
703
    prefix = 'ancestor:'
 
704
 
 
705
    def _match_on(self, branch, revs):
 
706
        trace.mutter('matching ancestor: on: %s, %s', self.spec, branch)
 
707
        return self._find_revision_info(branch, self.spec)
 
708
 
 
709
    def _as_revision_id(self, context_branch):
 
710
        return self._find_revision_id(context_branch, self.spec)
 
711
 
 
712
    @staticmethod
 
713
    def _find_revision_info(branch, other_location):
 
714
        revision_id = RevisionSpec_ancestor._find_revision_id(branch,
 
715
                                                              other_location)
 
716
        try:
 
717
            revno = branch.revision_id_to_revno(revision_id)
 
718
        except errors.NoSuchRevision:
 
719
            revno = None
 
720
        return RevisionInfo(branch, revno, revision_id)
 
721
 
 
722
    @staticmethod
 
723
    def _find_revision_id(branch, other_location):
 
724
        from bzrlib.branch import Branch
 
725
 
 
726
        branch.lock_read()
 
727
        try:
 
728
            revision_a = revision.ensure_null(branch.last_revision())
 
729
            if revision_a == revision.NULL_REVISION:
 
730
                raise errors.NoCommits(branch)
 
731
            if other_location == '':
 
732
                other_location = branch.get_parent()
 
733
            other_branch = Branch.open(other_location)
 
734
            other_branch.lock_read()
 
735
            try:
 
736
                revision_b = revision.ensure_null(other_branch.last_revision())
 
737
                if revision_b == revision.NULL_REVISION:
 
738
                    raise errors.NoCommits(other_branch)
 
739
                graph = branch.repository.get_graph(other_branch.repository)
 
740
                rev_id = graph.find_unique_lca(revision_a, revision_b)
 
741
            finally:
 
742
                other_branch.unlock()
 
743
            if rev_id == revision.NULL_REVISION:
 
744
                raise errors.NoCommonAncestor(revision_a, revision_b)
 
745
            return rev_id
 
746
        finally:
 
747
            branch.unlock()
 
748
 
 
749
 
 
750
 
 
751
 
 
752
class RevisionSpec_branch(RevisionSpec):
 
753
    """Selects the last revision of a specified branch."""
 
754
 
 
755
    help_txt = """Selects the last revision of a specified branch.
 
756
 
 
757
    Supply the path to a branch to select its last revision.
 
758
 
 
759
    Examples::
 
760
 
 
761
      branch:/path/to/branch
 
762
    """
 
763
    prefix = 'branch:'
 
764
 
 
765
    def _match_on(self, branch, revs):
 
766
        from bzrlib.branch import Branch
 
767
        other_branch = Branch.open(self.spec)
 
768
        revision_b = other_branch.last_revision()
 
769
        if revision_b in (None, revision.NULL_REVISION):
 
770
            raise errors.NoCommits(other_branch)
 
771
        # pull in the remote revisions so we can diff
 
772
        branch.fetch(other_branch, revision_b)
 
773
        try:
 
774
            revno = branch.revision_id_to_revno(revision_b)
 
775
        except errors.NoSuchRevision:
 
776
            revno = None
 
777
        return RevisionInfo(branch, revno, revision_b)
 
778
 
 
779
    def _as_revision_id(self, context_branch):
 
780
        from bzrlib.branch import Branch
 
781
        other_branch = Branch.open(self.spec)
 
782
        last_revision = other_branch.last_revision()
 
783
        last_revision = revision.ensure_null(last_revision)
 
784
        context_branch.fetch(other_branch, last_revision)
 
785
        if last_revision == revision.NULL_REVISION:
 
786
            raise errors.NoCommits(other_branch)
 
787
        return last_revision
 
788
 
 
789
    def _as_tree(self, context_branch):
 
790
        from bzrlib.branch import Branch
 
791
        other_branch = Branch.open(self.spec)
 
792
        last_revision = other_branch.last_revision()
 
793
        last_revision = revision.ensure_null(last_revision)
 
794
        if last_revision == revision.NULL_REVISION:
 
795
            raise errors.NoCommits(other_branch)
 
796
        return other_branch.repository.revision_tree(last_revision)
 
797
 
 
798
 
 
799
 
 
800
class RevisionSpec_submit(RevisionSpec_ancestor):
 
801
    """Selects a common ancestor with a submit branch."""
 
802
 
 
803
    help_txt = """Selects a common ancestor with the submit branch.
 
804
 
 
805
    Diffing against this shows all the changes that were made in this branch,
 
806
    and is a good predictor of what merge will do.  The submit branch is
 
807
    used by the bundle and merge directive commands.  If no submit branch
 
808
    is specified, the parent branch is used instead.
 
809
 
 
810
    The common ancestor is the last revision that existed in both
 
811
    branches. Usually this is the branch point, but it could also be
 
812
    a revision that was merged.
 
813
 
 
814
    Examples::
 
815
 
 
816
      $ bzr diff -r submit:
 
817
    """
 
818
 
 
819
    prefix = 'submit:'
 
820
 
 
821
    def _get_submit_location(self, branch):
 
822
        submit_location = branch.get_submit_branch()
 
823
        location_type = 'submit branch'
 
824
        if submit_location is None:
 
825
            submit_location = branch.get_parent()
 
826
            location_type = 'parent branch'
 
827
        if submit_location is None:
 
828
            raise errors.NoSubmitBranch(branch)
 
829
        trace.note('Using %s %s', location_type, submit_location)
 
830
        return submit_location
 
831
 
 
832
    def _match_on(self, branch, revs):
 
833
        trace.mutter('matching ancestor: on: %s, %s', self.spec, branch)
 
834
        return self._find_revision_info(branch,
 
835
            self._get_submit_location(branch))
 
836
 
 
837
    def _as_revision_id(self, context_branch):
 
838
        return self._find_revision_id(context_branch,
 
839
            self._get_submit_location(context_branch))
 
840
 
 
841
 
 
842
revspec_registry = registry.Registry()
 
843
def _register_revspec(revspec):
 
844
    revspec_registry.register(revspec.prefix, revspec)
 
845
 
 
846
_register_revspec(RevisionSpec_revno)
 
847
_register_revspec(RevisionSpec_revid)
 
848
_register_revspec(RevisionSpec_last)
 
849
_register_revspec(RevisionSpec_before)
 
850
_register_revspec(RevisionSpec_tag)
 
851
_register_revspec(RevisionSpec_date)
 
852
_register_revspec(RevisionSpec_ancestor)
 
853
_register_revspec(RevisionSpec_branch)
 
854
_register_revspec(RevisionSpec_submit)
 
855
 
 
856
SPEC_TYPES = symbol_versioning.deprecated_list(
 
857
    symbol_versioning.deprecated_in((1, 12, 0)), "SPEC_TYPES", [])