14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
from bzrlib.lazy_import import lazy_import
17
from __future__ import absolute_import
20
from .lazy_import import lazy_import
21
21
lazy_import(globals(), """
26
branch as _mod_branch,
32
from breezy.i18n import gettext
39
43
class RevisionInfo(object):
40
44
"""The results of applying a revision specification to a branch."""
53
57
or treat the result as a tuple.
56
def __init__(self, branch, revno, rev_id=_marker):
60
def __init__(self, branch, revno=None, rev_id=None):
57
61
self.branch = branch
62
self._has_revno = (revno is not None)
65
if self.rev_id is None and self._revno is not None:
60
66
# allow caller to be lazy
61
if self.revno is None:
64
self.rev_id = branch.get_rev_id(self.revno)
68
def __nonzero__(self):
69
# first the easy ones...
67
self.rev_id = branch.get_rev_id(self._revno)
71
if not self._has_revno and self.rev_id is not None:
73
self._revno = self.branch.revision_id_to_revno(self.rev_id)
74
except errors.NoSuchRevision:
76
self._has_revno = True
70
80
if self.rev_id is None:
72
if self.revno is not None:
74
82
# TODO: otherwise, it should depend on how I was built -
75
83
# if it's in_history(branch), then check revision_history(),
76
84
# if it's in_store(branch), do the check below
77
85
return self.branch.repository.has_revision(self.rev_id)
87
__nonzero__ = __bool__
90
100
def __eq__(self, other):
91
101
if type(other) not in (tuple, list, type(self)):
93
if type(other) is type(self) and self.branch is not other.branch:
103
if isinstance(other, type(self)) and self.branch is not other.branch:
95
105
return tuple(self) == tuple(other)
97
107
def __repr__(self):
98
return '<bzrlib.revisionspec.RevisionInfo object %s, %s for %r>' % (
108
return '<breezy.revisionspec.RevisionInfo object %s, %s for %r>' % (
99
109
self.revno, self.rev_id, self.branch)
102
def from_revision_id(branch, revision_id, revs):
112
def from_revision_id(branch, revision_id, revs=symbol_versioning.DEPRECATED_PARAMETER):
103
113
"""Construct a RevisionInfo given just the id.
105
115
Use this if you don't know or care what the revno is.
107
if revision_id == revision.NULL_REVISION:
108
return RevisionInfo(branch, 0, revision_id)
110
revno = revs.index(revision_id) + 1
113
return RevisionInfo(branch, revno, revision_id)
117
if symbol_versioning.deprecated_passed(revs):
118
symbol_versioning.warn(
119
'RevisionInfo.from_revision_id(revs) was deprecated in 2.5.',
122
return RevisionInfo(branch, revno=None, rev_id=revision_id)
119
125
class RevisionSpec(object):
166
173
spectype.__name__, spec)
167
174
return spectype(spec, _internal=True)
169
for spectype in SPEC_TYPES:
170
if spec.startswith(spectype.prefix):
171
trace.mutter('Returning RevisionSpec %s for %s',
172
spectype.__name__, spec)
173
return spectype(spec, _internal=True)
174
176
# Otherwise treat it as a DWIM, build the RevisionSpec object and
175
177
# wait for _match_on to be called.
176
178
return RevisionSpec_dwim(spec, _internal=True)
312
327
"""Run the lookup and see what we can get."""
314
329
# First, see if it's a revno
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:
330
if self._revno_regex.match(self.spec) is not None:
320
332
return self._try_spectype(RevisionSpec_revno, branch)
321
333
except RevisionSpec_revno.dwim_catchable_exceptions:
324
336
# Next see what has been registered
337
for objgetter in self._possible_revspecs:
338
rs_class = objgetter.get_obj()
340
return self._try_spectype(rs_class, branch)
341
except rs_class.dwim_catchable_exceptions:
344
# Try the old (deprecated) dwim list:
325
345
for rs_class in dwim_revspecs:
327
347
return self._try_spectype(rs_class, branch)
333
353
# really relevant.
334
354
raise errors.InvalidRevisionSpec(self.spec, branch)
357
def append_possible_revspec(cls, revspec):
358
"""Append a possible DWIM revspec.
360
:param revspec: Revision spec to try.
362
cls._possible_revspecs.append(registry._ObjectGetter(revspec))
365
def append_possible_lazy_revspec(cls, module_name, member_name):
366
"""Append a possible lazily loaded DWIM revspec.
368
:param module_name: Name of the module with the revspec
369
:param member_name: Name of the revspec within the module
371
cls._possible_revspecs.append(
372
registry._LazyObjectGetter(module_name, member_name))
337
375
class RevisionSpec_revno(RevisionSpec):
338
376
"""Selects a revision using a number."""
356
394
your history is very long.
358
396
prefix = 'revno:'
359
wants_revision_history = False
361
398
def _match_on(self, branch, revs):
362
399
"""Lookup a revision by revision number"""
363
branch, revno, revision_id = self._lookup(branch, revs)
400
branch, revno, revision_id = self._lookup(branch)
364
401
return RevisionInfo(branch, revno, revision_id)
366
def _lookup(self, branch, revs_or_none):
403
def _lookup(self, branch):
367
404
loc = self.spec.find(':')
369
406
revno_spec = self.spec
421
454
revno = last_revno + revno + 1
423
revision_id = branch.get_rev_id(revno, revs_or_none)
456
revision_id = branch.get_rev_id(revno)
424
457
except errors.NoSuchRevision:
425
458
raise errors.InvalidRevisionSpec(self.user_spec, branch)
426
459
return branch, revno, revision_id
428
461
def _as_revision_id(self, context_branch):
429
462
# We would have the revno here, but we don't really care
430
branch, revno, revision_id = self._lookup(context_branch, None)
463
branch, revno, revision_id = self._lookup(context_branch)
431
464
return revision_id
433
466
def needs_branch(self):
443
476
RevisionSpec_int = RevisionSpec_revno
447
class RevisionSpec_revid(RevisionSpec):
479
class RevisionIDSpec(RevisionSpec):
481
def _match_on(self, branch, revs):
482
revision_id = self.as_revision_id(branch)
483
return RevisionInfo.from_revision_id(branch, revision_id)
486
class RevisionSpec_revid(RevisionIDSpec):
448
487
"""Selects a revision using the revision id."""
450
489
help_txt = """Selects a revision using the revision id.
460
499
prefix = 'revid:'
462
def _match_on(self, branch, revs):
501
def _as_revision_id(self, context_branch):
463
502
# self.spec comes straight from parsing the command line arguments,
464
503
# so we expect it to be a Unicode string. Switch it to the internal
465
504
# representation.
466
revision_id = osutils.safe_revision_id(self.spec, warn=False)
467
return RevisionInfo.from_revision_id(branch, revision_id, revs)
469
def _as_revision_id(self, context_branch):
470
505
return osutils.safe_revision_id(self.spec, warn=False)
489
524
def _match_on(self, branch, revs):
490
revno, revision_id = self._revno_and_revision_id(branch, revs)
525
revno, revision_id = self._revno_and_revision_id(branch)
491
526
return RevisionInfo(branch, revno, revision_id)
493
def _revno_and_revision_id(self, context_branch, revs_or_none):
528
def _revno_and_revision_id(self, context_branch):
494
529
last_revno, last_revision_id = context_branch.last_revision_info()
496
531
if self.spec == '':
573
604
return RevisionInfo(branch, revno, revision_id)
575
606
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)
607
base_revision_id = RevisionSpec.from_string(self.spec)._as_revision_id(context_branch)
578
608
if base_revision_id == revision.NULL_REVISION:
579
609
raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
580
610
'cannot go before the null: revision')
621
650
class _RevListToTimestamps(object):
622
651
"""This takes a list of revisions, and allows you to bisect by date"""
624
__slots__ = ['revs', 'branch']
653
__slots__ = ['branch']
626
def __init__(self, revs, branch):
655
def __init__(self, branch):
628
656
self.branch = branch
630
658
def __getitem__(self, index):
631
659
"""Get the date of the index'd item"""
632
r = self.branch.repository.get_revision(self.revs[index])
660
r = self.branch.repository.get_revision(self.branch.get_rev_id(index))
633
661
# TODO: Handle timezone.
634
662
return datetime.datetime.fromtimestamp(r.timestamp)
636
664
def __len__(self):
637
return len(self.revs)
665
return self.branch.revno()
640
668
class RevisionSpec_date(RevisionSpec):
714
742
hour=hour, minute=minute, second=second)
715
743
branch.lock_read()
717
rev = bisect.bisect(_RevListToTimestamps(revs, branch), dt)
745
rev = bisect.bisect(_RevListToTimestamps(branch), dt, 1)
748
if rev == branch.revno():
721
749
raise errors.InvalidRevisionSpec(self.user_spec, branch)
723
return RevisionInfo(branch, rev + 1)
750
return RevisionInfo(branch, rev)
757
784
def _find_revision_info(branch, other_location):
758
785
revision_id = RevisionSpec_ancestor._find_revision_id(branch,
761
revno = branch.revision_id_to_revno(revision_id)
762
except errors.NoSuchRevision:
764
return RevisionInfo(branch, revno, revision_id)
787
return RevisionInfo(branch, None, revision_id)
767
790
def _find_revision_id(branch, other_location):
768
from bzrlib.branch import Branch
791
from .branch import Branch
770
793
branch.lock_read()
808
831
dwim_catchable_exceptions = (errors.NotBranchError,)
810
833
def _match_on(self, branch, revs):
811
from bzrlib.branch import Branch
834
from .branch import Branch
812
835
other_branch = Branch.open(self.spec)
813
836
revision_b = other_branch.last_revision()
814
837
if revision_b in (None, revision.NULL_REVISION):
815
838
raise errors.NoCommits(other_branch)
816
# pull in the remote revisions so we can diff
817
branch.fetch(other_branch, revision_b)
819
revno = branch.revision_id_to_revno(revision_b)
820
except errors.NoSuchRevision:
822
return RevisionInfo(branch, revno, revision_b)
840
branch = other_branch
843
# pull in the remote revisions so we can diff
844
branch.fetch(other_branch, revision_b)
845
except errors.ReadOnlyError:
846
branch = other_branch
847
return RevisionInfo(branch, None, revision_b)
824
849
def _as_revision_id(self, context_branch):
825
from bzrlib.branch import Branch
850
from .branch import Branch
826
851
other_branch = Branch.open(self.spec)
827
852
last_revision = other_branch.last_revision()
828
853
last_revision = revision.ensure_null(last_revision)
884
916
self._get_submit_location(context_branch))
919
class RevisionSpec_annotate(RevisionIDSpec):
923
help_txt = """Select the revision that last modified the specified line.
925
Select the revision that last modified the specified line. Line is
926
specified as path:number. Path is a relative path to the file. Numbers
927
start at 1, and are relative to the current version, not the last-
928
committed version of the file.
931
def _raise_invalid(self, numstring, context_branch):
932
raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
933
'No such line: %s' % numstring)
935
def _as_revision_id(self, context_branch):
936
path, numstring = self.spec.rsplit(':', 1)
938
index = int(numstring) - 1
940
self._raise_invalid(numstring, context_branch)
941
tree, file_path = workingtree.WorkingTree.open_containing(path)
944
file_id = tree.path2id(file_path)
946
raise errors.InvalidRevisionSpec(self.user_spec,
947
context_branch, "File '%s' is not versioned." %
949
revision_ids = [r for (r, l) in tree.annotate_iter(file_id)]
953
revision_id = revision_ids[index]
955
self._raise_invalid(numstring, context_branch)
956
if revision_id == revision.CURRENT_REVISION:
957
raise errors.InvalidRevisionSpec(self.user_spec, context_branch,
958
'Line %s has not been committed.' % numstring)
962
class RevisionSpec_mainline(RevisionIDSpec):
964
help_txt = """Select mainline revision that merged the specified revision.
966
Select the revision that merged the specified revision into mainline.
971
def _as_revision_id(self, context_branch):
972
revspec = RevisionSpec.from_string(self.spec)
973
if revspec.get_branch() is None:
974
spec_branch = context_branch
976
spec_branch = _mod_branch.Branch.open(revspec.get_branch())
977
revision_id = revspec.as_revision_id(spec_branch)
978
graph = context_branch.repository.get_graph()
979
result = graph.find_lefthand_merger(revision_id,
980
context_branch.last_revision())
982
raise errors.InvalidRevisionSpec(self.user_spec, context_branch)
887
986
# The order in which we want to DWIM a revision spec without any prefix.
888
987
# revno is always tried first and isn't listed here, this is used by
889
988
# RevisionSpec_dwim._match_on
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
989
dwim_revspecs = symbol_versioning.deprecated_list(
990
symbol_versioning.deprecated_in((2, 4, 0)), "dwim_revspecs", [])
992
RevisionSpec_dwim.append_possible_revspec(RevisionSpec_tag)
993
RevisionSpec_dwim.append_possible_revspec(RevisionSpec_revid)
994
RevisionSpec_dwim.append_possible_revspec(RevisionSpec_date)
995
RevisionSpec_dwim.append_possible_revspec(RevisionSpec_branch)
898
997
revspec_registry = registry.Registry()
899
998
def _register_revspec(revspec):
908
1007
_register_revspec(RevisionSpec_ancestor)
909
1008
_register_revspec(RevisionSpec_branch)
910
1009
_register_revspec(RevisionSpec_submit)
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", [])
1010
_register_revspec(RevisionSpec_annotate)
1011
_register_revspec(RevisionSpec_mainline)