23
23
from bzrlib.osutils import isdir, quotefn, compact_date, rand_bytes, \
25
25
sha_file, appendpath, file_kind
27
from bzrlib.errors import BzrError, InvalidRevisionNumber, InvalidRevisionId, \
28
DivergedBranches, NotBranchError
26
from bzrlib.errors import BzrError, InvalidRevisionNumber, InvalidRevisionId
29
27
from bzrlib.textui import show_status
30
28
from bzrlib.revision import Revision
29
from bzrlib.xml import unpack_xml
31
30
from bzrlib.delta import compare_trees
32
31
from bzrlib.tree import EmptyTree, RevisionTree
38
33
BZR_BRANCH_FORMAT = "Bazaar-NG branch, format 0.0.4\n"
39
34
## TODO: Maybe include checks for common corruption of newlines, etc?
43
38
# repeatedly to calculate deltas. We could perhaps have a weakref
44
39
# cache in memory to make this faster.
46
# TODO: please move the revision-string syntax stuff out of the branch
47
# object; it's clutter
50
42
def find_branch(f, **args):
51
43
if f and (f.startswith('http://') or f.startswith('https://')):
52
from bzrlib.remotebranch import RemoteBranch
53
return RemoteBranch(f, **args)
45
return remotebranch.RemoteBranch(f, **args)
55
47
return Branch(f, **args)
58
50
def find_cached_branch(f, cache_root, **args):
59
from bzrlib.remotebranch import RemoteBranch
51
from remotebranch import RemoteBranch
60
52
br = find_branch(f, **args)
61
53
def cacheify(br, store_name):
62
from bzrlib.meta_store import CachedStore
54
from meta_store import CachedStore
63
55
cache_path = os.path.join(cache_root, store_name)
64
56
os.mkdir(cache_path)
65
57
new_store = CachedStore(getattr(br, store_name), cache_path)
127
119
head, tail = os.path.split(f)
129
121
# reached the root, whatever that may be
130
raise NotBranchError('%s is not in a branch' % orig_f)
122
raise BzrError('%r is not in a branch' % orig_f)
125
class DivergedBranches(Exception):
126
def __init__(self, branch1, branch2):
127
self.branch1 = branch1
128
self.branch2 = branch2
129
Exception.__init__(self, "These branches have diverged.")
132
class NoSuchRevision(BzrError):
133
def __init__(self, branch, revision):
135
self.revision = revision
136
msg = "Branch %s has no revision %d" % (branch, revision)
137
BzrError.__init__(self, msg)
136
140
######################################################################
250
258
self._lock = None
251
259
self._lock_mode = self._lock_count = None
253
262
def abspath(self, name):
254
263
"""Return absolute filename for something in the branch"""
255
264
return os.path.join(self.base, name)
257
267
def relpath(self, path):
258
268
"""Return path relative to this branch of something inside it.
260
270
Raises an error if path is not in this branch."""
261
271
return _relpath(self.base, path)
263
274
def controlfilename(self, file_or_path):
264
275
"""Return location relative to branch."""
265
276
if isinstance(file_or_path, basestring):
354
364
def read_working_inventory(self):
355
365
"""Read the working inventory."""
356
366
from bzrlib.inventory import Inventory
367
from bzrlib.xml import unpack_xml
368
from time import time
359
372
# ElementTree does its own conversion from UTF-8, so open in
361
f = self.controlfile('inventory', 'rb')
362
return bzrlib.xml.serializer_v4.read_inventory(f)
374
inv = unpack_xml(Inventory,
375
self.controlfile('inventory', 'rb'))
376
mutter("loaded inventory of %d items in %f"
377
% (len(inv), time() - before))
390
407
"""Inventory for the working copy.""")
393
def add(self, files, ids=None):
410
def add(self, files, verbose=False, ids=None):
394
411
"""Make files versioned.
396
Note that the command line normally calls smart_add instead,
397
which can automatically recurse.
413
Note that the command line normally calls smart_add instead.
399
415
This puts the files in the Added state, so that they will be
400
416
recorded by the next commit.
570
def get_revision_xml_file(self, revision_id):
571
"""Return XML file object for revision object."""
572
if not revision_id or not isinstance(revision_id, basestring):
573
raise InvalidRevisionId(revision_id)
578
return self.revision_store[revision_id]
579
except (IndexError, KeyError):
580
raise bzrlib.errors.NoSuchRevision(self, revision_id)
586
get_revision_xml = get_revision_xml_file
589
594
def get_revision(self, revision_id):
590
595
"""Return the Revision object for a named revision"""
591
xml_file = self.get_revision_xml_file(revision_id)
594
r = bzrlib.xml.serializer_v4.read_revision(xml_file)
595
except SyntaxError, e:
596
raise bzrlib.errors.BzrError('failed to unpack revision_xml',
598
if not revision_id or not isinstance(revision_id, basestring):
599
raise InvalidRevisionId(revision_id)
600
r = unpack_xml(Revision, self.revision_store[revision_id])
600
604
assert r.revision_id == revision_id
642
646
parameter which can be either an integer revno or a
644
648
from bzrlib.inventory import Inventory
646
f = self.get_inventory_xml_file(inventory_id)
647
return bzrlib.xml.serializer_v4.read_inventory(f)
650
def get_inventory_xml(self, inventory_id):
651
"""Get inventory XML as a file object."""
652
return self.inventory_store[inventory_id]
654
get_inventory_xml_file = get_inventory_xml
649
from bzrlib.xml import unpack_xml
651
return unpack_xml(Inventory, self.inventory_store[inventory_id])
657
654
def get_inventory_sha1(self, inventory_id):
658
655
"""Return the sha1 hash of the inventory entry
660
return sha_file(self.get_inventory_xml(inventory_id))
657
return sha_file(self.inventory_store[inventory_id])
663
660
def get_revision_inventory(self, revision_id):
688
685
def common_ancestor(self, other, self_revno=None, other_revno=None):
690
>>> from bzrlib.commit import commit
691
688
>>> sb = ScratchBranch(files=['foo', 'foo~'])
692
689
>>> sb.common_ancestor(sb) == (None, None)
694
>>> commit(sb, "Committing first revision", verbose=False)
691
>>> commit.commit(sb, "Committing first revision", verbose=False)
695
692
>>> sb.common_ancestor(sb)[0]
697
694
>>> clone = sb.clone()
698
>>> commit(sb, "Committing second revision", verbose=False)
695
>>> commit.commit(sb, "Committing second revision", verbose=False)
699
696
>>> sb.common_ancestor(sb)[0]
701
698
>>> sb.common_ancestor(clone)[0]
703
>>> commit(clone, "Committing divergent second revision",
700
>>> commit.commit(clone, "Committing divergent second revision",
704
701
... verbose=False)
705
702
>>> sb.common_ancestor(clone)[0]
788
785
if stop_revision is None:
789
786
stop_revision = other_len
790
787
elif stop_revision > other_len:
791
raise bzrlib.errors.NoSuchRevision(self, stop_revision)
788
raise NoSuchRevision(self, stop_revision)
793
790
return other_history[self_len:stop_revision]
796
793
def update_revisions(self, other, stop_revision=None):
797
794
"""Pull in all new revisions from other branch.
796
>>> from bzrlib.commit import commit
797
>>> bzrlib.trace.silent = True
798
>>> br1 = ScratchBranch(files=['foo', 'bar'])
801
>>> commit(br1, "lala!", rev_id="REVISION-ID-1", verbose=False)
802
>>> br2 = ScratchBranch()
803
>>> br2.update_revisions(br1)
807
>>> br2.revision_history()
809
>>> br2.update_revisions(br1)
813
>>> br1.text_store.total_size() == br2.text_store.total_size()
799
from bzrlib.fetch import greedy_fetch
800
from bzrlib.revision import get_intervening_revisions
802
pb = bzrlib.ui.ui_factory.progress_bar()
816
from bzrlib.progress import ProgressBar
803
820
pb.update('comparing histories')
804
if stop_revision is None:
805
other_revision = other.last_patch()
807
other_revision = other.lookup_revision(stop_revision)
808
count = greedy_fetch(self, other, other_revision, pb)[0]
810
revision_ids = self.missing_revisions(other, stop_revision)
811
except DivergedBranches, e:
813
revision_ids = get_intervening_revisions(self.last_patch(),
814
other_revision, self)
815
assert self.last_patch() not in revision_ids
816
except bzrlib.errors.NotAncestor:
819
self.append_revision(*revision_ids)
822
def install_revisions(self, other, revision_ids, pb):
821
revision_ids = self.missing_revisions(other, stop_revision)
823
823
if hasattr(other.revision_store, "prefetch"):
824
824
other.revision_store.prefetch(revision_ids)
825
825
if hasattr(other.inventory_store, "prefetch"):
827
for rev_id in revision_ids:
829
revision = other.get_revision(rev_id).inventory_id
830
inventory_ids.append(revision)
831
except bzrlib.errors.NoSuchRevision:
826
inventory_ids = [other.get_revision(r).inventory_id
827
for r in revision_ids]
833
828
other.inventory_store.prefetch(inventory_ids)
836
pb = bzrlib.ui.ui_factory.progress_bar()
839
831
needed_texts = set()
843
for i, rev_id in enumerate(revision_ids):
844
pb.update('fetching revision', i+1, len(revision_ids))
846
rev = other.get_revision(rev_id)
847
except bzrlib.errors.NoSuchRevision:
833
for rev_id in revision_ids:
835
pb.update('fetching revision', i, len(revision_ids))
836
rev = other.get_revision(rev_id)
851
837
revisions.append(rev)
852
838
inv = other.get_inventory(str(rev.inventory_id))
853
839
for key, entry in inv.iter_entries():
861
count, cp_fail = self.text_store.copy_multi(other.text_store,
863
#print "Added %d texts." % count
847
count = self.text_store.copy_multi(other.text_store, needed_texts)
848
print "Added %d texts." % count
864
849
inventory_ids = [ f.inventory_id for f in revisions ]
865
count, cp_fail = self.inventory_store.copy_multi(other.inventory_store,
867
#print "Added %d inventories." % count
850
count = self.inventory_store.copy_multi(other.inventory_store,
852
print "Added %d inventories." % count
868
853
revision_ids = [ f.revision_id for f in revisions]
870
count, cp_fail = self.revision_store.copy_multi(other.revision_store,
873
assert len(cp_fail) == 0
874
return count, failures
854
count = self.revision_store.copy_multi(other.revision_store,
856
for revision_id in revision_ids:
857
self.append_revision(revision_id)
858
print "Added %d revisions." % count
877
861
def commit(self, *args, **kw):
878
862
from bzrlib.commit import commit
879
863
commit(self, *args, **kw)
882
866
def lookup_revision(self, revision):
883
867
"""Return the revision identifier for a given revision information."""
884
revno, info = self._get_revision_info(revision)
868
revno, info = self.get_revision_info(revision)
888
def revision_id_to_revno(self, revision_id):
889
"""Given a revision id, return its revno"""
890
history = self.revision_history()
892
return history.index(revision_id) + 1
894
raise bzrlib.errors.NoSuchRevision(self, revision_id)
897
871
def get_revision_info(self, revision):
898
872
"""Return (revno, revision id) for revision identifier.
902
876
revision can also be a string, in which case it is parsed for something like
903
877
'date:' or 'revid:' etc.
905
revno, rev_id = self._get_revision_info(revision)
907
raise bzrlib.errors.NoSuchRevision(self, revision)
910
def get_rev_id(self, revno, history=None):
911
"""Find the revision id of the specified revno."""
915
history = self.revision_history()
916
elif revno <= 0 or revno > len(history):
917
raise bzrlib.errors.NoSuchRevision(self, revno)
918
return history[revno - 1]
920
def _get_revision_info(self, revision):
921
"""Return (revno, revision id) for revision specifier.
923
revision can be an integer, in which case it is assumed to be revno
924
(though this will translate negative values into positive ones)
925
revision can also be a string, in which case it is parsed for something
926
like 'date:' or 'revid:' etc.
928
A revid is always returned. If it is None, the specifier referred to
929
the null revision. If the revid does not occur in the revision
930
history, revno will be None.
933
879
if revision is None:
940
886
revs = self.revision_history()
941
887
if isinstance(revision, int):
890
# Mabye we should do this first, but we don't need it if revision == 0
943
892
revno = len(revs) + revision + 1
946
rev_id = self.get_rev_id(revno, revs)
947
895
elif isinstance(revision, basestring):
948
896
for prefix, func in Branch.REVISION_NAMESPACES.iteritems():
949
897
if revision.startswith(prefix):
950
result = func(self, revs, revision)
952
revno, rev_id = result
955
rev_id = self.get_rev_id(revno, revs)
898
revno = func(self, revs, revision)
958
raise BzrError('No namespace registered for string: %r' %
961
raise TypeError('Unhandled revision type %s' % revision)
901
raise BzrError('No namespace registered for string: %r' % revision)
965
raise bzrlib.errors.NoSuchRevision(self, revision)
903
if revno is None or revno <= 0 or revno > len(revs):
904
raise BzrError("no such revision %s" % revision)
905
return revno, revs[revno-1]
968
907
def _namespace_revno(self, revs, revision):
969
908
"""Lookup a revision by revision number"""
970
909
assert revision.startswith('revno:')
972
return (int(revision[6:]),)
911
return int(revision[6:])
973
912
except ValueError:
975
914
REVISION_NAMESPACES['revno:'] = _namespace_revno
977
916
def _namespace_revid(self, revs, revision):
978
917
assert revision.startswith('revid:')
979
rev_id = revision[len('revid:'):]
981
return revs.index(rev_id) + 1, rev_id
919
return revs.index(revision[6:]) + 1
982
920
except ValueError:
984
922
REVISION_NAMESPACES['revid:'] = _namespace_revid
986
924
def _namespace_last(self, revs, revision):
1073
1011
# TODO: Handle timezone.
1074
1012
dt = datetime.datetime.fromtimestamp(r.timestamp)
1075
1013
if first >= dt and (last is None or dt >= last):
1078
1016
for i in range(len(revs)):
1079
1017
r = self.get_revision(revs[i])
1080
1018
# TODO: Handle timezone.
1081
1019
dt = datetime.datetime.fromtimestamp(r.timestamp)
1082
1020
if first <= dt and (last is None or dt <= last):
1084
1022
REVISION_NAMESPACES['date:'] = _namespace_date
1087
def _namespace_ancestor(self, revs, revision):
1088
from revision import common_ancestor, MultipleRevisionSources
1089
other_branch = find_branch(_trim_namespace('ancestor', revision))
1090
revision_a = self.last_patch()
1091
revision_b = other_branch.last_patch()
1092
for r, b in ((revision_a, self), (revision_b, other_branch)):
1094
raise bzrlib.errors.NoCommits(b)
1095
revision_source = MultipleRevisionSources(self, other_branch)
1096
result = common_ancestor(revision_a, revision_b, revision_source)
1098
revno = self.revision_id_to_revno(result)
1099
except bzrlib.errors.NoSuchRevision:
1104
REVISION_NAMESPACES['ancestor:'] = _namespace_ancestor
1106
1024
def revision_tree(self, revision_id):
1107
1025
"""Return Tree for a revision on this branch.
1344
def get_parent(self):
1345
"""Return the parent location of the branch.
1347
This is the default location for push/pull/missing. The usual
1348
pattern is that the user can override it by specifying a
1352
_locs = ['parent', 'pull', 'x-pull']
1355
return self.controlfile(l, 'r').read().strip('\n')
1357
if e.errno != errno.ENOENT:
1362
def set_parent(self, url):
1363
# TODO: Maybe delete old location files?
1364
from bzrlib.atomicfile import AtomicFile
1367
f = AtomicFile(self.controlfilename('parent'))
1376
def check_revno(self, revno):
1378
Check whether a revno corresponds to any revision.
1379
Zero (the NULL revision) is considered valid.
1382
self.check_real_revno(revno)
1384
def check_real_revno(self, revno):
1386
Check whether a revno corresponds to a real revision.
1387
Zero (the NULL revision) is considered invalid
1389
if revno < 1 or revno > self.revno():
1390
raise InvalidRevisionNumber(revno)
1395
1259
class ScratchBranch(Branch):
1396
1260
"""Special test class: a branch that cleans up after itself.
1515
1377
"""Return a new tree-root file id."""
1516
1378
return gen_file_id('TREE_ROOT')
1519
def copy_branch(branch_from, to_location, revision=None):
1520
"""Copy branch_from into the existing directory to_location.
1523
If not None, only revisions up to this point will be copied.
1524
The head of the new branch will be that revision.
1527
The name of a local directory that exists but is empty.
1529
from bzrlib.merge import merge
1531
assert isinstance(branch_from, Branch)
1532
assert isinstance(to_location, basestring)
1534
br_to = Branch(to_location, init=True)
1535
br_to.set_root_id(branch_from.get_root_id())
1536
if revision is None:
1537
revno = branch_from.revno()
1539
revno, rev_id = branch_from.get_revision_info(revision)
1540
br_to.update_revisions(branch_from, stop_revision=revno)
1541
merge((to_location, -1), (to_location, 0), this_dir=to_location,
1542
check_clean=False, ignore_zero=True)
1543
br_to.set_parent(branch_from.base)
1546
def _trim_namespace(namespace, spec):
1547
full_namespace = namespace + ':'
1548
assert spec.startswith(full_namespace)
1549
return spec[len(full_namespace):]