~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revisionspec.py

  • Committer: John Arbash Meinel
  • Date: 2009-06-19 19:26:32 UTC
  • mto: This revision was merged to the branch mainline in revision 4466.
  • Revision ID: john@arbash-meinel.com-20090619192632-1a4ntoq61fkhlp2x
Make a note of the 'worst case' for heads.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
 
18
 
import datetime
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
 
19
18
import re
 
19
 
 
20
from bzrlib.lazy_import import lazy_import
 
21
lazy_import(globals(), """
20
22
import bisect
21
 
from bzrlib.errors import BzrError, NoSuchRevision, NoCommits
 
23
import datetime
 
24
""")
 
25
 
 
26
from bzrlib import (
 
27
    errors,
 
28
    osutils,
 
29
    registry,
 
30
    revision,
 
31
    symbol_versioning,
 
32
    trace,
 
33
    )
 
34
 
22
35
 
23
36
_marker = []
24
37
 
 
38
 
25
39
class RevisionInfo(object):
26
 
    """The results of applying a revision specification to a branch.
 
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.
27
43
 
28
44
    An instance has two useful attributes: revno, and rev_id.
29
45
 
82
98
        return '<bzrlib.revisionspec.RevisionInfo object %s, %s for %r>' % (
83
99
            self.revno, self.rev_id, self.branch)
84
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
 
85
116
# classes in this list should have a "prefix" attribute, against which
86
117
# string specs are matched
87
 
SPEC_TYPES = []
 
118
_revno_regex = None
 
119
 
88
120
 
89
121
class RevisionSpec(object):
90
 
    """A parsed revision specification.
 
122
    """A parsed revision specification."""
 
123
 
 
124
    help_txt = """A parsed revision specification.
91
125
 
92
126
    A revision specification can be an integer, in which case it is
93
127
    assumed to be a revno (though this will translate negative values
104
138
    """
105
139
 
106
140
    prefix = None
107
 
 
108
 
    def __new__(cls, spec, foo=_marker):
109
 
        """Parse a revision specifier.
 
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.
110
150
        """
 
151
        if not isinstance(spec, (type(None), basestring)):
 
152
            raise TypeError('error')
 
153
 
111
154
        if spec is None:
112
 
            return object.__new__(RevisionSpec, spec)
113
 
 
114
 
        try:
115
 
            spec = int(spec)
116
 
        except ValueError:
117
 
            pass
118
 
 
119
 
        if isinstance(spec, int):
120
 
            return object.__new__(RevisionSpec_int, spec)
121
 
        elif isinstance(spec, basestring):
 
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:
122
163
            for spectype in SPEC_TYPES:
 
164
                trace.mutter('Returning RevisionSpec %s for %s',
 
165
                             spectype.__name__, spec)
123
166
                if spec.startswith(spectype.prefix):
124
 
                    return object.__new__(spectype, spec)
125
 
            else:
126
 
                raise BzrError('No namespace registered for string: %r' %
127
 
                               spec)
128
 
        else:
129
 
            raise TypeError('Unhandled revision type %s' % spec)
130
 
 
131
 
    def __init__(self, spec):
 
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
132
195
        if self.prefix and spec.startswith(self.prefix):
133
196
            spec = spec[len(self.prefix):]
134
197
        self.spec = spec
135
198
 
136
199
    def _match_on(self, branch, revs):
137
 
        return RevisionInfo(branch, 0, None)
 
200
        trace.mutter('Returning RevisionSpec._match_on: None')
 
201
        return RevisionInfo(branch, None, None)
138
202
 
139
203
    def _match_on_and_check(self, branch, revs):
140
204
        info = self._match_on(branch, revs)
141
205
        if info:
142
206
            return info
143
 
        elif info == (0, None):
144
 
            # special case - the empty tree
 
207
        elif info == (None, None):
 
208
            # special case - nothing supplied
145
209
            return info
146
210
        elif self.prefix:
147
 
            raise NoSuchRevision(branch, self.prefix + str(self.spec))
 
211
            raise errors.InvalidRevisionSpec(self.user_spec, branch)
148
212
        else:
149
 
            raise NoSuchRevision(branch, str(self.spec))
 
213
            raise errors.InvalidRevisionSpec(self.spec, branch)
150
214
 
151
215
    def in_history(self, branch):
152
216
        if branch:
153
 
            revs = branch.revision_history()
 
217
            if self.wants_revision_history:
 
218
                revs = branch.revision_history()
 
219
            else:
 
220
                revs = None
154
221
        else:
 
222
            # this should never trigger.
 
223
            # TODO: make it a deprecated code path. RBC 20060928
155
224
            revs = None
156
225
        return self._match_on_and_check(branch, revs)
157
226
 
164
233
    # will do what you expect.
165
234
    in_store = in_history
166
235
    in_branch = in_store
167
 
        
 
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
 
168
272
    def __repr__(self):
169
273
        # this is mostly for helping with testing
170
 
        return '<%s %s%s>' % (self.__class__.__name__,
171
 
                              self.prefix or '',
172
 
                              self.spec)
173
 
    
 
274
        return '<%s %s>' % (self.__class__.__name__,
 
275
                              self.user_spec)
 
276
 
174
277
    def needs_branch(self):
175
278
        """Whether this revision spec needs a branch.
176
279
 
178
281
        """
179
282
        return True
180
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
 
181
292
# private API
182
293
 
183
 
class RevisionSpec_int(RevisionSpec):
184
 
    """Spec is a number.  Special case."""
185
 
    def __init__(self, spec):
186
 
        self.spec = int(spec)
187
 
 
188
 
    def _match_on(self, branch, revs):
189
 
        if self.spec < 0:
190
 
            revno = len(revs) + self.spec + 1
191
 
        else:
192
 
            revno = self.spec
193
 
        rev_id = branch.get_rev_id(revno, revs)
194
 
        return RevisionInfo(branch, revno, rev_id)
195
 
 
196
 
 
197
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
    """
198
315
    prefix = 'revno:'
 
316
    wants_revision_history = False
199
317
 
200
318
    def _match_on(self, branch, revs):
201
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):
202
394
        if self.spec.find(':') == -1:
203
 
            try:
204
 
                return RevisionInfo(branch, int(self.spec))
205
 
            except ValueError:
206
 
                return RevisionInfo(branch, None)
 
395
            return None
207
396
        else:
208
 
            from branch import Branch
209
 
            revname = self.spec[self.spec.find(':')+1:]
210
 
            other_branch = Branch.open_containing(revname)[0]
211
 
            try:
212
 
                revno = int(self.spec[:self.spec.find(':')])
213
 
            except ValueError:
214
 
                return RevisionInfo(other_branch, None)
215
 
            revid = other_branch.get_rev_id(revno)
216
 
            return RevisionInfo(other_branch, revno)
217
 
        
218
 
    def needs_branch(self):
219
 
        return self.spec.find(':') == -1
220
 
 
221
 
SPEC_TYPES.append(RevisionSpec_revno)
 
397
            return self.spec[self.spec.find(':')+1:]
 
398
 
 
399
# Old compatibility
 
400
RevisionSpec_int = RevisionSpec_revno
 
401
 
222
402
 
223
403
 
224
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
 
225
417
    prefix = 'revid:'
226
418
 
227
419
    def _match_on(self, branch, revs):
228
 
        try:
229
 
            return RevisionInfo(branch, revs.index(self.spec) + 1, self.spec)
230
 
        except ValueError:
231
 
            return RevisionInfo(branch, None, self.spec)
232
 
 
233
 
SPEC_TYPES.append(RevisionSpec_revid)
 
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
 
234
429
 
235
430
 
236
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
    """
237
443
 
238
444
    prefix = 'last:'
239
445
 
240
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
 
241
458
        try:
242
459
            offset = int(self.spec)
243
 
        except ValueError:
244
 
            return RevisionInfo(branch, None)
245
 
        else:
246
 
            if offset <= 0:
247
 
                raise BzrError('You must supply a positive value for --revision last:XXX')
248
 
            return RevisionInfo(branch, len(revs) - offset + 1)
249
 
 
250
 
SPEC_TYPES.append(RevisionSpec_last)
 
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
 
251
480
 
252
481
 
253
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
    """
254
503
 
255
504
    prefix = 'before:'
256
 
    
 
505
 
257
506
    def _match_on(self, branch, revs):
258
 
        r = RevisionSpec(self.spec)._match_on(branch, revs)
259
 
        if (r.revno is None) or (r.revno == 0):
260
 
            return r
261
 
        return RevisionInfo(branch, r.revno - 1)
262
 
 
263
 
SPEC_TYPES.append(RevisionSpec_before)
 
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
 
264
554
 
265
555
 
266
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
 
267
564
    prefix = 'tag:'
268
565
 
269
566
    def _match_on(self, branch, revs):
270
 
        raise BzrError('tag: namespace registered, but not implemented.')
271
 
 
272
 
SPEC_TYPES.append(RevisionSpec_tag)
273
 
 
274
 
 
275
 
class RevisionSpec_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
 
276
582
    def __init__(self, revs, branch):
277
583
        self.revs = revs
278
584
        self.branch = branch
 
585
 
279
586
    def __getitem__(self, index):
 
587
        """Get the date of the index'd item"""
280
588
        r = self.branch.repository.get_revision(self.revs[index])
281
589
        # TODO: Handle timezone.
282
590
        return datetime.datetime.fromtimestamp(r.timestamp)
 
591
 
283
592
    def __len__(self):
284
593
        return len(self.revs)
285
594
 
286
595
 
287
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
    """
288
616
    prefix = 'date:'
289
617
    _date_re = re.compile(
290
618
            r'(?P<date>(?P<year>\d\d\d\d)-(?P<month>\d\d)-(?P<day>\d\d))?'
293
621
        )
294
622
 
295
623
    def _match_on(self, branch, revs):
296
 
        """
297
 
        Spec for date revisions:
 
624
        """Spec for date revisions:
298
625
          date:value
299
626
          value can be 'yesterday', 'today', 'tomorrow' or a YYYY-MM-DD string.
300
627
          matches the first entry after a given date (either at midnight or
301
628
          at a specified time).
302
 
 
303
 
          So the proper way of saying 'give me all entries for today' is:
304
 
              -r date:yesterday..date:today
305
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
306
633
        today = datetime.datetime.fromordinal(datetime.date.today().toordinal())
307
634
        if self.spec.lower() == 'yesterday':
308
635
            dt = today - datetime.timedelta(days=1)
313
640
        else:
314
641
            m = self._date_re.match(self.spec)
315
642
            if not m or (not m.group('date') and not m.group('time')):
316
 
                raise BzrError('Invalid revision date %r' % self.spec)
317
 
 
318
 
            if m.group('date'):
319
 
                year, month, day = int(m.group('year')), int(m.group('month')), int(m.group('day'))
320
 
            else:
321
 
                year, month, day = today.year, today.month, today.day
322
 
            if m.group('time'):
323
 
                hour = int(m.group('hour'))
324
 
                minute = int(m.group('minute'))
325
 
                if m.group('second'):
326
 
                    second = int(m.group('second'))
327
 
                else:
328
 
                    second = 0
329
 
            else:
330
 
                hour, minute, second = 0,0,0
 
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')
331
668
 
332
669
            dt = datetime.datetime(year=year, month=month, day=day,
333
670
                    hour=hour, minute=minute, second=second)
334
671
        branch.lock_read()
335
672
        try:
336
 
            rev = bisect.bisect(RevisionSpec_revs(revs, branch), dt)
 
673
            rev = bisect.bisect(_RevListToTimestamps(revs, branch), dt)
337
674
        finally:
338
675
            branch.unlock()
339
676
        if rev == len(revs):
340
 
            return RevisionInfo(branch, None)
 
677
            raise errors.InvalidRevisionSpec(self.user_spec, branch)
341
678
        else:
342
679
            return RevisionInfo(branch, rev + 1)
343
680
 
344
 
SPEC_TYPES.append(RevisionSpec_date)
345
681
 
346
682
 
347
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
    """
348
703
    prefix = 'ancestor:'
349
704
 
350
705
    def _match_on(self, branch, revs):
351
 
        from branch import Branch
352
 
        from revision import common_ancestor, MultipleRevisionSources
353
 
        other_branch = Branch.open_containing(self.spec)[0]
354
 
        revision_a = branch.last_revision()
355
 
        revision_b = other_branch.last_revision()
356
 
        for r, b in ((revision_a, branch), (revision_b, other_branch)):
357
 
            if r is None:
358
 
                raise NoCommits(b)
359
 
        revision_source = MultipleRevisionSources(branch.repository,
360
 
                                                  other_branch.repository)
361
 
        rev_id = common_ancestor(revision_a, revision_b, revision_source)
 
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)
362
716
        try:
363
 
            revno = branch.revision_id_to_revno(rev_id)
364
 
        except NoSuchRevision:
 
717
            revno = branch.revision_id_to_revno(revision_id)
 
718
        except errors.NoSuchRevision:
365
719
            revno = None
366
 
        return RevisionInfo(branch, revno, rev_id)
367
 
        
368
 
SPEC_TYPES.append(RevisionSpec_ancestor)
 
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
 
369
751
 
370
752
class RevisionSpec_branch(RevisionSpec):
371
 
    """A branch: revision specifier.
372
 
 
373
 
    This takes the path to a branch and returns its tip revision id.
 
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
374
762
    """
375
763
    prefix = 'branch:'
376
764
 
377
765
    def _match_on(self, branch, revs):
378
 
        from branch import Branch
379
 
        other_branch = Branch.open_containing(self.spec)[0]
 
766
        from bzrlib.branch import Branch
 
767
        other_branch = Branch.open(self.spec)
380
768
        revision_b = other_branch.last_revision()
381
 
        if revision_b is None:
382
 
            raise NoCommits(other_branch)
 
769
        if revision_b in (None, revision.NULL_REVISION):
 
770
            raise errors.NoCommits(other_branch)
383
771
        # pull in the remote revisions so we can diff
384
772
        branch.fetch(other_branch, revision_b)
385
773
        try:
386
774
            revno = branch.revision_id_to_revno(revision_b)
387
 
        except NoSuchRevision:
 
775
        except errors.NoSuchRevision:
388
776
            revno = None
389
777
        return RevisionInfo(branch, revno, revision_b)
390
 
        
391
 
SPEC_TYPES.append(RevisionSpec_branch)
 
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", [])