15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21
from warnings import warn
22
from cStringIO import StringIO
26
from bzrlib.inventory import InventoryEntry
27
import bzrlib.inventory as inventory
28
21
from bzrlib.trace import mutter, note
29
from bzrlib.osutils import (isdir, quotefn, compact_date, rand_bytes,
30
rename, splitpath, sha_file, appendpath,
32
from bzrlib.errors import (BzrError, InvalidRevisionNumber, InvalidRevisionId,
33
NoSuchRevision, HistoryMissing, NotBranchError,
34
DivergedBranches, LockError, UnlistableStore,
35
UnlistableBranch, NoSuchFile)
36
from bzrlib.textui import show_status
37
from bzrlib.revision import Revision
38
from bzrlib.delta import compare_trees
39
from bzrlib.tree import EmptyTree, RevisionTree
40
from bzrlib.inventory import Inventory
41
from bzrlib.store import copy_all
42
from bzrlib.store.compressed_text import CompressedTextStore
43
from bzrlib.store.text import TextStore
44
from bzrlib.store.weave import WeaveStore
45
from bzrlib.transport import Transport, get_transport
50
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
51
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
22
from bzrlib.osutils import isdir, quotefn, compact_date, rand_bytes, splitpath, \
23
sha_file, appendpath, file_kind
24
from bzrlib.errors import BzrError
26
BZR_BRANCH_FORMAT = "Bazaar-NG branch, format 0.0.4\n"
52
27
## TODO: Maybe include checks for common corruption of newlines, etc?
55
# TODO: Some operations like log might retrieve the same revisions
56
# repeatedly to calculate deltas. We could perhaps have a weakref
57
# cache in memory to make this faster. In general anything can be
58
# cached in memory between lock and unlock operations.
60
def find_branch(*ignored, **ignored_too):
61
# XXX: leave this here for about one release, then remove it
62
raise NotImplementedError('find_branch() is not supported anymore, '
63
'please use one of the new branch constructors')
31
def find_branch(f, **args):
32
if f and (f.startswith('http://') or f.startswith('https://')):
34
return remotebranch.RemoteBranch(f, **args)
36
return Branch(f, **args)
39
def find_cached_branch(f, cache_root, **args):
40
from remotebranch import RemoteBranch
41
br = find_branch(f, **args)
42
def cacheify(br, store_name):
43
from meta_store import CachedStore
44
cache_path = os.path.join(cache_root, store_name)
46
new_store = CachedStore(getattr(br, store_name), cache_path)
47
setattr(br, store_name, new_store)
49
if isinstance(br, RemoteBranch):
50
cacheify(br, 'inventory_store')
51
cacheify(br, 'text_store')
52
cacheify(br, 'revision_store')
64
56
def _relpath(base, path):
65
57
"""Return path relative to base, or raise exception.
78
from errors import NotBranchError
86
79
raise NotBranchError("path %r is not within branch %r" % (rp, base))
88
81
return os.sep.join(s)
91
def find_branch_root(t):
92
"""Find the branch root enclosing the transport's base.
94
t is a Transport object.
96
It is not necessary that the base of t exists.
84
def find_branch_root(f=None):
85
"""Find the branch root enclosing f, or pwd.
87
f may be a filename or a URL.
89
It is not necessary that f exists.
98
91
Basically we keep looking up until we find the control directory or
99
run into the root. If there isn't one, raises NotBranchError.
95
elif hasattr(os.path, 'realpath'):
96
f = os.path.realpath(f)
98
f = os.path.abspath(f)
99
if not os.path.exists(f):
100
raise BzrError('%r does not exist' % f)
103
if t.has(bzrlib.BZRDIR):
105
new_t = t.clone('..')
106
if new_t.base == t.base:
106
if os.path.exists(os.path.join(f, bzrlib.BZRDIR)):
108
head, tail = os.path.split(f)
107
110
# reached the root, whatever that may be
108
raise NotBranchError('%s is not in a branch' % orig_base)
111
raise BzrError('%r is not in a branch' % orig_f)
114
class DivergedBranches(Exception):
115
def __init__(self, branch1, branch2):
116
self.branch1 = branch1
117
self.branch2 = branch2
118
Exception.__init__(self, "These branches have diverged.")
121
class NoSuchRevision(BzrError):
122
def __init__(self, branch, revision):
124
self.revision = revision
125
msg = "Branch %s has no revision %d" % (branch, revision)
126
BzrError.__init__(self, msg)
112
129
######################################################################
116
133
"""Branch holding a history of revisions.
119
Base directory/url of the branch.
123
def __init__(self, *ignored, **ignored_too):
124
raise NotImplementedError('The Branch class is abstract')
127
def open_downlevel(base):
128
"""Open a branch which may be of an old format.
130
Only local branches are supported."""
131
return _Branch(get_transport(base), relax_version_check=True)
135
"""Open an existing branch, rooted at 'base' (url)"""
136
t = get_transport(base)
137
mutter("trying to open %r with transport %r", base, t)
141
def open_containing(url):
142
"""Open an existing branch which contains url.
144
This probes for a branch at url, and searches upwards from there.
146
t = get_transport(url)
147
t = find_branch_root(t)
151
def initialize(base):
152
"""Create a new branch, rooted at 'base' (url)"""
153
t = get_transport(base)
154
return _Branch(t, init=True)
156
def setup_caching(self, cache_root):
157
"""Subclasses that care about caching should override this, and set
158
up cached stores located under cache_root.
160
self.cache_root = cache_root
163
class _Branch(Branch):
164
"""A branch stored in the actual filesystem.
166
Note that it's "local" in the context of the filesystem; it doesn't
167
really matter if it's on an nfs/smb/afs/coda/... share, as long as
168
it's writable, and can be accessed via the normal filesystem API.
136
Base directory of the branch.
171
139
None, or 'r' or 'w'
178
146
Lock object from bzrlib.lock.
180
# We actually expect this class to be somewhat short-lived; part of its
181
# purpose is to try to isolate what bits of the branch logic are tied to
182
# filesystem access, so that in a later step, we can extricate them to
183
# a separarte ("storage") class.
184
149
_lock_mode = None
185
150
_lock_count = None
187
_inventory_weave = None
189
# Map some sort of prefix into a namespace
190
# stuff like "revno:10", "revid:", etc.
191
# This should match a prefix with a function which accepts
192
REVISION_NAMESPACES = {}
194
def push_stores(self, branch_to):
195
"""Copy the content of this branches store to branch_to."""
196
if (self._branch_format != branch_to._branch_format
197
or self._branch_format != 4):
198
from bzrlib.fetch import greedy_fetch
199
mutter("falling back to fetch logic to push between %s(%s) and %s(%s)",
200
self, self._branch_format, branch_to, branch_to._branch_format)
201
greedy_fetch(to_branch=branch_to, from_branch=self,
202
revision=self.last_revision())
205
store_pairs = ((self.text_store, branch_to.text_store),
206
(self.inventory_store, branch_to.inventory_store),
207
(self.revision_store, branch_to.revision_store))
209
for from_store, to_store in store_pairs:
210
copy_all(from_store, to_store)
211
except UnlistableStore:
212
raise UnlistableBranch(from_store)
214
def __init__(self, transport, init=False,
215
relax_version_check=False):
153
def __init__(self, base, init=False, find_root=True):
216
154
"""Create new branch object at a particular location.
218
transport -- A Transport object, defining how to access files.
219
(If a string, transport.transport() will be used to
220
create a Transport object)
156
base -- Base directory for the branch.
222
158
init -- If True, create new control files in a previously
223
159
unversioned directory. If False, the branch must already
226
relax_version_check -- If true, the usual check for the branch
227
version is not applied. This is intended only for
228
upgrade/recovery type use; it's not guaranteed that
229
all operations will work on old format branches.
162
find_root -- If true and init is false, find the root of the
163
existing branch containing base.
231
165
In the test suite, creation of new trees is tested using the
232
166
`ScratchBranch` class.
234
assert isinstance(transport, Transport), \
235
"%r is not a Transport" % transport
236
self._transport = transport
168
from bzrlib.store import ImmutableStore
170
self.base = os.path.realpath(base)
238
171
self._make_control()
239
self._check_format(relax_version_check)
241
def get_store(name, compressed=True):
242
# FIXME: This approach of assuming stores are all entirely compressed
243
# or entirely uncompressed is tidy, but breaks upgrade from
244
# some existing branches where there's a mixture; we probably
245
# still want the option to look for both.
246
relpath = self._rel_controlfilename(name)
248
store = CompressedTextStore(self._transport.clone(relpath))
250
store = TextStore(self._transport.clone(relpath))
251
#if self._transport.should_cache():
252
# cache_path = os.path.join(self.cache_root, name)
253
# os.mkdir(cache_path)
254
# store = bzrlib.store.CachedStore(store, cache_path)
257
relpath = self._rel_controlfilename(name)
258
ws = WeaveStore(self._transport.clone(relpath))
259
if self._transport.should_cache():
260
ws.enable_cache = True
263
if self._branch_format == 4:
264
self.inventory_store = get_store('inventory-store')
265
self.text_store = get_store('text-store')
266
self.revision_store = get_store('revision-store')
267
elif self._branch_format == 5:
268
self.control_weaves = get_weave([])
269
self.weave_store = get_weave('weaves')
270
self.revision_store = get_store('revision-store', compressed=False)
173
self.base = find_branch_root(base)
175
self.base = os.path.realpath(base)
176
if not isdir(self.controlfilename('.')):
177
from errors import NotBranchError
178
raise NotBranchError("not a bzr branch: %s" % quotefn(base),
179
['use "bzr init" to initialize a new working tree',
180
'current bzr can only operate from top-of-tree'])
183
self.text_store = ImmutableStore(self.controlfilename('text-store'))
184
self.revision_store = ImmutableStore(self.controlfilename('revision-store'))
185
self.inventory_store = ImmutableStore(self.controlfilename('inventory-store'))
272
188
def __str__(self):
273
return '%s(%r)' % (self.__class__.__name__, self._transport.base)
189
return '%s(%r)' % (self.__class__.__name__, self.base)
276
192
__repr__ = __str__
279
195
def __del__(self):
280
196
if self._lock_mode or self._lock:
281
# XXX: This should show something every time, and be suitable for
282
# headless operation and embedding
197
from warnings import warn
283
198
warn("branch %r was not explicitly unlocked" % self)
284
199
self._lock.unlock()
286
# TODO: It might be best to do this somewhere else,
287
# but it is nice for a Branch object to automatically
288
# cache it's information.
289
# Alternatively, we could have the Transport objects cache requests
290
# See the earlier discussion about how major objects (like Branch)
291
# should never expect their __del__ function to run.
292
if hasattr(self, 'cache_root') and self.cache_root is not None:
295
shutil.rmtree(self.cache_root)
298
self.cache_root = None
302
return self._transport.base
305
base = property(_get_base)
308
203
def lock_write(self):
309
# TODO: Upgrade locking to support using a Transport,
310
# and potentially a remote locking protocol
311
204
if self._lock_mode:
312
205
if self._lock_mode != 'w':
206
from errors import LockError
313
207
raise LockError("can't upgrade to a write lock from %r" %
315
209
self._lock_count += 1
317
self._lock = self._transport.lock_write(
318
self._rel_controlfilename('branch-lock'))
211
from bzrlib.lock import WriteLock
213
self._lock = WriteLock(self.controlfilename('branch-lock'))
319
214
self._lock_mode = 'w'
320
215
self._lock_count = 1
323
219
def lock_read(self):
324
220
if self._lock_mode:
325
221
assert self._lock_mode in ('r', 'w'), \
326
222
"invalid lock mode %r" % self._lock_mode
327
223
self._lock_count += 1
329
self._lock = self._transport.lock_read(
330
self._rel_controlfilename('branch-lock'))
225
from bzrlib.lock import ReadLock
227
self._lock = ReadLock(self.controlfilename('branch-lock'))
331
228
self._lock_mode = 'r'
332
229
self._lock_count = 1
334
233
def unlock(self):
335
234
if not self._lock_mode:
235
from errors import LockError
336
236
raise LockError('branch %r is not locked' % (self))
338
238
if self._lock_count > 1:
374
273
Controlfiles should almost never be opened in write mode but
375
274
rather should be atomically copied and replaced using atomicfile.
379
relpath = self._rel_controlfilename(file_or_path)
380
#TODO: codecs.open() buffers linewise, so it was overloaded with
381
# a much larger buffer, do we need to do the same for getreader/getwriter?
383
return self._transport.get(relpath)
385
raise BzrError("Branch.controlfile(mode='wb') is not supported, use put_controlfiles")
387
return codecs.getreader('utf-8')(self._transport.get(relpath), errors='replace')
389
raise BzrError("Branch.controlfile(mode='w') is not supported, use put_controlfiles")
277
fn = self.controlfilename(file_or_path)
279
if mode == 'rb' or mode == 'wb':
280
return file(fn, mode)
281
elif mode == 'r' or mode == 'w':
282
# open in binary mode anyhow so there's no newline translation;
283
# codecs uses line buffering by default; don't want that.
285
return codecs.open(fn, mode + 'b', 'utf-8',
391
288
raise BzrError("invalid controlfile mode %r" % mode)
393
def put_controlfile(self, path, f, encode=True):
394
"""Write an entry as a controlfile.
396
:param path: The path to put the file, relative to the .bzr control
398
:param f: A file-like or string object whose contents should be copied.
399
:param encode: If true, encode the contents as utf-8
401
self.put_controlfiles([(path, f)], encode=encode)
403
def put_controlfiles(self, files, encode=True):
404
"""Write several entries as controlfiles.
406
:param files: A list of [(path, file)] pairs, where the path is the directory
407
underneath the bzr control directory
408
:param encode: If true, encode the contents as utf-8
412
for path, f in files:
414
if isinstance(f, basestring):
415
f = f.encode('utf-8', 'replace')
417
f = codecs.getwriter('utf-8')(f, errors='replace')
418
path = self._rel_controlfilename(path)
419
ctrl_files.append((path, f))
420
self._transport.put_multi(ctrl_files)
422
292
def _make_control(self):
423
293
from bzrlib.inventory import Inventory
424
from bzrlib.weavefile import write_weave_v5
425
from bzrlib.weave import Weave
294
from bzrlib.xml import pack_xml
427
# Create an empty inventory
429
# if we want per-tree root ids then this is the place to set
430
# them; they're not needed for now and so ommitted for
432
bzrlib.xml5.serializer_v5.write_inventory(Inventory(), sio)
433
empty_inv = sio.getvalue()
435
bzrlib.weavefile.write_weave_v5(Weave(), sio)
436
empty_weave = sio.getvalue()
438
dirs = [[], 'revision-store', 'weaves']
296
os.mkdir(self.controlfilename([]))
297
self.controlfile('README', 'w').write(
440
298
"This is a Bazaar-NG control directory.\n"
441
"Do not change any files in this directory.\n"),
442
('branch-format', BZR_BRANCH_FORMAT_5),
443
('revision-history', ''),
446
('pending-merges', ''),
447
('inventory', empty_inv),
448
('inventory.weave', empty_weave),
449
('ancestry.weave', empty_weave)
451
cfn = self._rel_controlfilename
452
self._transport.mkdir_multi([cfn(d) for d in dirs])
453
self.put_controlfiles(files)
454
mutter('created control directory in ' + self._transport.base)
456
def _check_format(self, relax_version_check):
299
"Do not change any files in this directory.\n")
300
self.controlfile('branch-format', 'w').write(BZR_BRANCH_FORMAT)
301
for d in ('text-store', 'inventory-store', 'revision-store'):
302
os.mkdir(self.controlfilename(d))
303
for f in ('revision-history', 'merged-patches',
304
'pending-merged-patches', 'branch-name',
307
self.controlfile(f, 'w').write('')
308
mutter('created control directory in ' + self.base)
310
pack_xml(Inventory(), self.controlfile('inventory','w'))
313
def _check_format(self):
457
314
"""Check this branch format is supported.
459
The format level is stored, as an integer, in
460
self._branch_format for code that needs to check it later.
316
The current tool only supports the current unstable format.
462
318
In the future, we might need different in-memory Branch
463
319
classes to support downlevel branches. But not yet.
466
fmt = self.controlfile('branch-format', 'r').read()
468
raise NotBranchError(self.base)
469
mutter("got branch format %r", fmt)
470
if fmt == BZR_BRANCH_FORMAT_5:
471
self._branch_format = 5
472
elif fmt == BZR_BRANCH_FORMAT_4:
473
self._branch_format = 4
475
if (not relax_version_check
476
and self._branch_format != 5):
321
# This ignores newlines so that we can open branches created
322
# on Windows from Linux and so on. I think it might be better
323
# to always make all internal files in unix format.
324
fmt = self.controlfile('branch-format', 'r').read()
325
fmt.replace('\r\n', '')
326
if fmt != BZR_BRANCH_FORMAT:
477
327
raise BzrError('sorry, branch format %r not supported' % fmt,
478
328
['use a different bzr version',
479
'or remove the .bzr directory'
480
' and "bzr init" again'])
482
def get_root_id(self):
483
"""Return the id of this branches root"""
484
inv = self.read_working_inventory()
485
return inv.root.file_id
487
def set_root_id(self, file_id):
488
inv = self.read_working_inventory()
489
orig_root_id = inv.root.file_id
490
del inv._byid[inv.root.file_id]
491
inv.root.file_id = file_id
492
inv._byid[inv.root.file_id] = inv.root
495
if entry.parent_id in (None, orig_root_id):
496
entry.parent_id = inv.root.file_id
497
self._write_inventory(inv)
329
'or remove the .bzr directory and "bzr init" again'])
499
333
def read_working_inventory(self):
500
334
"""Read the working inventory."""
335
from bzrlib.inventory import Inventory
336
from bzrlib.xml import unpack_xml
337
from time import time
503
341
# ElementTree does its own conversion from UTF-8, so open in
505
f = self.controlfile('inventory', 'rb')
506
return bzrlib.xml5.serializer_v5.read_inventory(f)
343
inv = unpack_xml(Inventory,
344
self.controlfile('inventory', 'rb'))
345
mutter("loaded inventory of %d items in %f"
346
% (len(inv), time() - before))
514
355
That is to say, the inventory describing changes underway, that
515
356
will be committed to the next revision.
517
from cStringIO import StringIO
358
from bzrlib.atomicfile import AtomicFile
359
from bzrlib.xml import pack_xml
518
361
self.lock_write()
521
bzrlib.xml5.serializer_v5.write_inventory(inv, sio)
523
# Transport handles atomicity
524
self.put_controlfile('inventory', sio)
363
f = AtomicFile(self.controlfilename('inventory'), 'wb')
528
372
mutter('wrote working inventory')
530
375
inventory = property(read_working_inventory, _write_inventory, None,
531
376
"""Inventory for the working copy.""")
533
def add(self, files, ids=None):
379
def add(self, files, verbose=False, ids=None):
534
380
"""Make files versioned.
536
Note that the command line normally calls smart_add instead,
537
which can automatically recurse.
382
Note that the command line normally calls smart_add instead.
539
384
This puts the files in the Added state, so that they will be
540
385
recorded by the next commit.
696
544
return self.working_tree().unknowns()
699
def append_revision(self, *revision_ids):
700
for revision_id in revision_ids:
701
mutter("add {%s} to revision-history" % revision_id)
704
rev_history = self.revision_history()
705
rev_history.extend(revision_ids)
706
self.put_controlfile('revision-history', '\n'.join(rev_history))
710
def has_revision(self, revision_id):
711
"""True if this branch has a copy of the revision.
713
This does not necessarily imply the revision is merge
714
or on the mainline."""
715
return (revision_id is None
716
or revision_id in self.revision_store)
718
def get_revision_xml_file(self, revision_id):
719
"""Return XML file object for revision object."""
720
if not revision_id or not isinstance(revision_id, basestring):
721
raise InvalidRevisionId(revision_id)
726
return self.revision_store[revision_id]
727
except (IndexError, KeyError):
728
raise bzrlib.errors.NoSuchRevision(self, revision_id)
733
get_revision_xml = get_revision_xml_file
735
def get_revision_xml(self, revision_id):
736
return self.get_revision_xml_file(revision_id).read()
547
def append_revision(self, revision_id):
548
from bzrlib.atomicfile import AtomicFile
550
mutter("add {%s} to revision-history" % revision_id)
551
rev_history = self.revision_history() + [revision_id]
553
f = AtomicFile(self.controlfilename('revision-history'))
555
for rev_id in rev_history:
739
562
def get_revision(self, revision_id):
740
563
"""Return the Revision object for a named revision"""
741
xml_file = self.get_revision_xml_file(revision_id)
564
from bzrlib.revision import Revision
565
from bzrlib.xml import unpack_xml
744
r = bzrlib.xml5.serializer_v5.read_revision(xml_file)
745
except SyntaxError, e:
746
raise bzrlib.errors.BzrError('failed to unpack revision_xml',
569
if not revision_id or not isinstance(revision_id, basestring):
570
raise ValueError('invalid revision-id: %r' % revision_id)
571
r = unpack_xml(Revision, self.revision_store[revision_id])
750
575
assert r.revision_id == revision_id
753
def get_revision_delta(self, revno):
754
"""Return the delta for one revision.
756
The delta is relative to its mainline predecessor, or the
757
empty tree for revision 1.
759
assert isinstance(revno, int)
760
rh = self.revision_history()
761
if not (1 <= revno <= len(rh)):
762
raise InvalidRevisionNumber(revno)
764
# revno is 1-based; list is 0-based
766
new_tree = self.revision_tree(rh[revno-1])
768
old_tree = EmptyTree()
770
old_tree = self.revision_tree(rh[revno-2])
772
return compare_trees(old_tree, new_tree)
774
579
def get_revision_sha1(self, revision_id):
775
580
"""Hash the stored value of a revision, and return it."""
779
584
# the revision, (add signatures/remove signatures) and still
780
585
# have all hash pointers stay consistent.
781
586
# But for now, just hash the contents.
782
return bzrlib.osutils.sha_file(self.get_revision_xml_file(revision_id))
784
def get_ancestry(self, revision_id):
785
"""Return a list of revision-ids integrated by a revision.
787
This currently returns a list, but the ordering is not guaranteed:
790
if revision_id is None:
792
w = self.control_weaves.get_weave('inventory')
793
return [None] + map(w.idx_to_name,
794
w.inclusions([w.lookup(revision_id)]))
796
def get_inventory_weave(self):
797
return self.control_weaves.get_weave('inventory')
799
def get_inventory(self, revision_id):
800
"""Get Inventory object by hash."""
801
xml = self.get_inventory_xml(revision_id)
802
return bzrlib.xml5.serializer_v5.read_inventory_from_string(xml)
804
def get_inventory_xml(self, revision_id):
805
"""Get inventory XML as a file object."""
807
assert isinstance(revision_id, basestring), type(revision_id)
808
iw = self.get_inventory_weave()
809
return iw.get_text(iw.lookup(revision_id))
811
raise bzrlib.errors.HistoryMissing(self, 'inventory', revision_id)
813
def get_inventory_sha1(self, revision_id):
587
return sha_file(self.revision_store[revision_id])
590
def get_inventory(self, inventory_id):
591
"""Get Inventory object by hash.
593
TODO: Perhaps for this and similar methods, take a revision
594
parameter which can be either an integer revno or a
596
from bzrlib.inventory import Inventory
597
from bzrlib.xml import unpack_xml
599
return unpack_xml(Inventory, self.inventory_store[inventory_id])
602
def get_inventory_sha1(self, inventory_id):
814
603
"""Return the sha1 hash of the inventory entry
816
return self.get_revision(revision_id).inventory_sha1
605
return sha_file(self.inventory_store[inventory_id])
818
608
def get_revision_inventory(self, revision_id):
819
609
"""Return inventory of a past revision."""
820
# TODO: Unify this with get_inventory()
821
# bzr 0.0.6 and later imposes the constraint that the inventory_id
610
# bzr 0.0.6 imposes the constraint that the inventory_id
822
611
# must be the same as its revision, so this is trivial.
823
612
if revision_id == None:
824
return Inventory(self.get_root_id())
613
from bzrlib.inventory import Inventory
826
616
return self.get_inventory(revision_id)
828
619
def revision_history(self):
829
"""Return sequence of revision hashes on to this branch."""
620
"""Return sequence of revision hashes on to this branch.
622
>>> ScratchBranch().revision_history()
832
627
return [l.rstrip('\r\n') for l in
941
754
if stop_revision is None:
942
755
stop_revision = other_len
944
assert isinstance(stop_revision, int)
945
if stop_revision > other_len:
946
raise bzrlib.errors.NoSuchRevision(self, stop_revision)
756
elif stop_revision > other_len:
757
raise NoSuchRevision(self, stop_revision)
947
759
return other_history[self_len:stop_revision]
949
762
def update_revisions(self, other, stop_revision=None):
950
"""Pull in new perfect-fit revisions."""
951
from bzrlib.fetch import greedy_fetch
952
from bzrlib.revision import get_intervening_revisions
953
if stop_revision is None:
954
stop_revision = other.last_revision()
955
greedy_fetch(to_branch=self, from_branch=other,
956
revision=stop_revision)
957
pullable_revs = self.missing_revisions(
958
other, other.revision_id_to_revno(stop_revision))
960
greedy_fetch(to_branch=self,
962
revision=pullable_revs[-1])
963
self.append_revision(*pullable_revs)
763
"""Pull in all new revisions from other branch.
765
>>> from bzrlib.commit import commit
766
>>> bzrlib.trace.silent = True
767
>>> br1 = ScratchBranch(files=['foo', 'bar'])
770
>>> commit(br1, "lala!", rev_id="REVISION-ID-1", verbose=False)
771
>>> br2 = ScratchBranch()
772
>>> br2.update_revisions(br1)
776
>>> br2.revision_history()
778
>>> br2.update_revisions(br1)
782
>>> br1.text_store.total_size() == br2.text_store.total_size()
785
from bzrlib.progress import ProgressBar
789
from sets import Set as set
793
pb.update('comparing histories')
794
revision_ids = self.missing_revisions(other, stop_revision)
796
if hasattr(other.revision_store, "prefetch"):
797
other.revision_store.prefetch(revision_ids)
798
if hasattr(other.inventory_store, "prefetch"):
799
inventory_ids = [other.get_revision(r).inventory_id
800
for r in revision_ids]
801
other.inventory_store.prefetch(inventory_ids)
806
for rev_id in revision_ids:
808
pb.update('fetching revision', i, len(revision_ids))
809
rev = other.get_revision(rev_id)
810
revisions.append(rev)
811
inv = other.get_inventory(str(rev.inventory_id))
812
for key, entry in inv.iter_entries():
813
if entry.text_id is None:
815
if entry.text_id not in self.text_store:
816
needed_texts.add(entry.text_id)
820
count = self.text_store.copy_multi(other.text_store, needed_texts)
821
print "Added %d texts." % count
822
inventory_ids = [ f.inventory_id for f in revisions ]
823
count = self.inventory_store.copy_multi(other.inventory_store,
825
print "Added %d inventories." % count
826
revision_ids = [ f.revision_id for f in revisions]
827
count = self.revision_store.copy_multi(other.revision_store,
829
for revision_id in revision_ids:
830
self.append_revision(revision_id)
831
print "Added %d revisions." % count
966
834
def commit(self, *args, **kw):
967
from bzrlib.commit import Commit
968
Commit().commit(self, *args, **kw)
970
def revision_id_to_revno(self, revision_id):
971
"""Given a revision id, return its revno"""
972
if revision_id is None:
974
history = self.revision_history()
976
return history.index(revision_id) + 1
978
raise bzrlib.errors.NoSuchRevision(self, revision_id)
835
from bzrlib.commit import commit
836
commit(self, *args, **kw)
980
def get_rev_id(self, revno, history=None):
981
"""Find the revision id of the specified revno."""
839
def lookup_revision(self, revno):
840
"""Return revision hash for revision number."""
985
history = self.revision_history()
986
elif revno <= 0 or revno > len(history):
987
raise bzrlib.errors.NoSuchRevision(self, revno)
988
return history[revno - 1]
845
# list is 0-based; revisions are 1-based
846
return self.revision_history()[revno-1]
848
raise BzrError("no such revision %s" % revno)
990
851
def revision_tree(self, revision_id):
991
852
"""Return Tree for a revision on this branch.
993
854
`revision_id` may be None for the null revision, in which case
994
855
an `EmptyTree` is returned."""
856
from bzrlib.tree import EmptyTree, RevisionTree
995
857
# TODO: refactor this to use an existing revision object
996
858
# so we don't need to read it in twice.
997
859
if revision_id == None:
998
860
return EmptyTree()
1000
862
inv = self.get_revision_inventory(revision_id)
1001
return RevisionTree(self.weave_store, inv, revision_id)
863
return RevisionTree(self.text_store, inv)
1004
866
def working_tree(self):
1005
867
"""Return a `Tree` for the working copy."""
1006
from bzrlib.workingtree import WorkingTree
1007
# TODO: In the future, WorkingTree should utilize Transport
1008
# RobertCollins 20051003 - I don't think it should - working trees are
1009
# much more complex to keep consistent than our careful .bzr subset.
1010
# instead, we should say that working trees are local only, and optimise
1012
return WorkingTree(self._transport.base, self.read_working_inventory())
868
from workingtree import WorkingTree
869
return WorkingTree(self.base, self.read_working_inventory())
1015
872
def basis_tree(self):
1201
def add_pending_merge(self, *revision_ids):
1202
# TODO: Perhaps should check at this point that the
1203
# history of the revision is actually present?
1060
def add_pending_merge(self, revision_id):
1061
from bzrlib.revision import validate_revision_id
1063
validate_revision_id(revision_id)
1204
1065
p = self.pending_merges()
1206
for rev_id in revision_ids:
1212
self.set_pending_merges(p)
1066
if revision_id in p:
1068
p.append(revision_id)
1069
self.set_pending_merges(p)
1214
1072
def set_pending_merges(self, rev_list):
1217
self.put_controlfile('pending-merges', '\n'.join(rev_list))
1222
def get_parent(self):
1223
"""Return the parent location of the branch.
1225
This is the default location for push/pull/missing. The usual
1226
pattern is that the user can override it by specifying a
1230
_locs = ['parent', 'pull', 'x-pull']
1233
return self.controlfile(l, 'r').read().strip('\n')
1235
if e.errno != errno.ENOENT:
1240
def set_parent(self, url):
1241
# TODO: Maybe delete old location files?
1242
1073
from bzrlib.atomicfile import AtomicFile
1243
1074
self.lock_write()
1245
f = AtomicFile(self.controlfilename('parent'))
1076
f = AtomicFile(self.controlfilename('pending-merges'))
1254
def check_revno(self, revno):
1256
Check whether a revno corresponds to any revision.
1257
Zero (the NULL revision) is considered valid.
1260
self.check_real_revno(revno)
1262
def check_real_revno(self, revno):
1264
Check whether a revno corresponds to a real revision.
1265
Zero (the NULL revision) is considered invalid
1267
if revno < 1 or revno > self.revno():
1268
raise InvalidRevisionNumber(revno)
1274
class ScratchBranch(_Branch):
1088
class ScratchBranch(Branch):
1275
1089
"""Special test class: a branch that cleans up after itself.
1277
1091
>>> b = ScratchBranch()