~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revisionspec.py

  • Committer: Aaron Bentley
  • Date: 2005-09-12 16:33:28 UTC
  • mto: (1185.1.16)
  • mto: This revision was merged to the branch mainline in revision 1390.
  • Revision ID: abentley@panoramicfeedback.com-20050912163328-8cd13823e289d44f
Narrowed exception handling for remote file retrieval

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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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", [])