~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revisionspec.py

  • Committer: Robert Collins
  • Date: 2006-07-20 13:00:31 UTC
  • mto: (1852.9.1 Tree.compare().)
  • mto: This revision was merged to the branch mainline in revision 1890.
  • Revision ID: robertc@robertcollins.net-20060720130031-d26103a427ea10f3
StartĀ treeĀ implementationĀ tests.

Show diffs side-by-side

added added

removed removed

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