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
21
28
from bzrlib.trace import mutter, note
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"
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"
27
52
## TODO: Maybe include checks for common corruption of newlines, etc?
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')
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')
56
64
def _relpath(base, path):
57
65
"""Return path relative to base, or raise exception.
78
from errors import NotBranchError
79
86
raise NotBranchError("path %r is not within branch %r" % (rp, base))
81
88
return os.sep.join(s)
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.
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.
91
98
Basically we keep looking up until we find the control directory or
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)
99
run into the root. If there isn't one, raises NotBranchError.
106
if os.path.exists(os.path.join(f, bzrlib.BZRDIR)):
108
head, tail = os.path.split(f)
103
if t.has(bzrlib.BZRDIR):
105
new_t = t.clone('..')
106
if new_t.base == t.base:
110
107
# reached the root, whatever that may be
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)
108
raise NotBranchError('%s is not in a branch' % orig_base)
129
112
######################################################################
133
116
"""Branch holding a history of revisions.
136
Base directory of the branch.
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.
139
171
None, or 'r' or 'w'
146
178
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.
149
184
_lock_mode = None
150
185
_lock_count = None
187
_inventory_weave = None
153
def __init__(self, base, init=False, find_root=True):
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):
154
216
"""Create new branch object at a particular location.
156
base -- Base directory for the branch.
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)
158
222
init -- If True, create new control files in a previously
159
223
unversioned directory. If False, the branch must already
162
find_root -- If true and init is false, find the root of the
163
existing branch containing base.
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.
165
231
In the test suite, creation of new trees is tested using the
166
232
`ScratchBranch` class.
168
from bzrlib.store import ImmutableStore
234
assert isinstance(transport, Transport), \
235
"%r is not a Transport" % transport
236
self._transport = transport
170
self.base = os.path.realpath(base)
171
238
self._make_control()
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'))
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)
188
272
def __str__(self):
189
return '%s(%r)' % (self.__class__.__name__, self.base)
273
return '%s(%r)' % (self.__class__.__name__, self._transport.base)
192
276
__repr__ = __str__
195
279
def __del__(self):
196
280
if self._lock_mode or self._lock:
197
from warnings import warn
281
# XXX: This should show something every time, and be suitable for
282
# headless operation and embedding
198
283
warn("branch %r was not explicitly unlocked" % self)
199
284
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)
203
308
def lock_write(self):
309
# TODO: Upgrade locking to support using a Transport,
310
# and potentially a remote locking protocol
204
311
if self._lock_mode:
205
312
if self._lock_mode != 'w':
206
from errors import LockError
207
313
raise LockError("can't upgrade to a write lock from %r" %
209
315
self._lock_count += 1
211
from bzrlib.lock import WriteLock
213
self._lock = WriteLock(self.controlfilename('branch-lock'))
317
self._lock = self._transport.lock_write(
318
self._rel_controlfilename('branch-lock'))
214
319
self._lock_mode = 'w'
215
320
self._lock_count = 1
219
323
def lock_read(self):
220
324
if self._lock_mode:
221
325
assert self._lock_mode in ('r', 'w'), \
222
326
"invalid lock mode %r" % self._lock_mode
223
327
self._lock_count += 1
225
from bzrlib.lock import ReadLock
227
self._lock = ReadLock(self.controlfilename('branch-lock'))
329
self._lock = self._transport.lock_read(
330
self._rel_controlfilename('branch-lock'))
228
331
self._lock_mode = 'r'
229
332
self._lock_count = 1
233
334
def unlock(self):
234
335
if not self._lock_mode:
235
from errors import LockError
236
336
raise LockError('branch %r is not locked' % (self))
238
338
if self._lock_count > 1:
273
374
Controlfiles should almost never be opened in write mode but
274
375
rather should be atomically copied and replaced using atomicfile.
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',
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")
288
391
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)
292
422
def _make_control(self):
293
423
from bzrlib.inventory import Inventory
294
from bzrlib.xml import pack_xml
424
from bzrlib.weavefile import write_weave_v5
425
from bzrlib.weave import Weave
296
os.mkdir(self.controlfilename([]))
297
self.controlfile('README', 'w').write(
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']
298
440
"This is a Bazaar-NG control directory.\n"
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):
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):
314
457
"""Check this branch format is supported.
316
The current tool only supports the current unstable format.
459
The format level is stored, as an integer, in
460
self._branch_format for code that needs to check it later.
318
462
In the future, we might need different in-memory Branch
319
463
classes to support downlevel branches. But not yet.
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:
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):
327
477
raise BzrError('sorry, branch format %r not supported' % fmt,
328
478
['use a different bzr version',
329
'or remove the .bzr directory and "bzr init" again'])
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)
333
499
def read_working_inventory(self):
334
500
"""Read the working inventory."""
335
from bzrlib.inventory import Inventory
336
from bzrlib.xml import unpack_xml
337
from time import time
341
503
# ElementTree does its own conversion from UTF-8, so open in
343
inv = unpack_xml(Inventory,
344
self.controlfile('inventory', 'rb'))
345
mutter("loaded inventory of %d items in %f"
346
% (len(inv), time() - before))
505
f = self.controlfile('inventory', 'rb')
506
return bzrlib.xml5.serializer_v5.read_inventory(f)
355
514
That is to say, the inventory describing changes underway, that
356
515
will be committed to the next revision.
358
from bzrlib.atomicfile import AtomicFile
359
from bzrlib.xml import pack_xml
517
from cStringIO import StringIO
361
518
self.lock_write()
363
f = AtomicFile(self.controlfilename('inventory'), 'wb')
521
bzrlib.xml5.serializer_v5.write_inventory(inv, sio)
523
# Transport handles atomicity
524
self.put_controlfile('inventory', sio)
372
528
mutter('wrote working inventory')
375
530
inventory = property(read_working_inventory, _write_inventory, None,
376
531
"""Inventory for the working copy.""")
379
def add(self, files, verbose=False, ids=None):
533
def add(self, files, ids=None):
380
534
"""Make files versioned.
382
Note that the command line normally calls smart_add instead.
536
Note that the command line normally calls smart_add instead,
537
which can automatically recurse.
384
539
This puts the files in the Added state, so that they will be
385
540
recorded by the next commit.
544
696
return self.working_tree().unknowns()
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:
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()
562
739
def get_revision(self, revision_id):
563
740
"""Return the Revision object for a named revision"""
564
from bzrlib.revision import Revision
565
from bzrlib.xml import unpack_xml
741
xml_file = self.get_revision_xml_file(revision_id)
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])
744
r = bzrlib.xml5.serializer_v5.read_revision(xml_file)
745
except SyntaxError, e:
746
raise bzrlib.errors.BzrError('failed to unpack revision_xml',
575
750
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)
579
774
def get_revision_sha1(self, revision_id):
580
775
"""Hash the stored value of a revision, and return it."""
584
779
# the revision, (add signatures/remove signatures) and still
585
780
# have all hash pointers stay consistent.
586
781
# But for now, just hash the contents.
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):
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):
603
814
"""Return the sha1 hash of the inventory entry
605
return sha_file(self.inventory_store[inventory_id])
816
return self.get_revision(revision_id).inventory_sha1
608
818
def get_revision_inventory(self, revision_id):
609
819
"""Return inventory of a past revision."""
610
# bzr 0.0.6 imposes the constraint that the inventory_id
820
# TODO: Unify this with get_inventory()
821
# bzr 0.0.6 and later imposes the constraint that the inventory_id
611
822
# must be the same as its revision, so this is trivial.
612
823
if revision_id == None:
613
from bzrlib.inventory import Inventory
824
return Inventory(self.get_root_id())
616
826
return self.get_inventory(revision_id)
619
828
def revision_history(self):
620
"""Return sequence of revision hashes on to this branch.
622
>>> ScratchBranch().revision_history()
829
"""Return sequence of revision hashes on to this branch."""
627
832
return [l.rstrip('\r\n') for l in
754
941
if stop_revision is None:
755
942
stop_revision = other_len
756
elif stop_revision > other_len:
757
raise NoSuchRevision(self, stop_revision)
944
assert isinstance(stop_revision, int)
945
if stop_revision > other_len:
946
raise bzrlib.errors.NoSuchRevision(self, stop_revision)
759
947
return other_history[self_len:stop_revision]
762
949
def update_revisions(self, other, stop_revision=None):
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
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)
834
966
def commit(self, *args, **kw):
835
from bzrlib.commit import commit
836
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)
839
def lookup_revision(self, revno):
840
"""Return revision hash for revision number."""
980
def get_rev_id(self, revno, history=None):
981
"""Find the revision id of the specified revno."""
845
# list is 0-based; revisions are 1-based
846
return self.revision_history()[revno-1]
848
raise BzrError("no such revision %s" % revno)
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]
851
990
def revision_tree(self, revision_id):
852
991
"""Return Tree for a revision on this branch.
854
993
`revision_id` may be None for the null revision, in which case
855
994
an `EmptyTree` is returned."""
856
from bzrlib.tree import EmptyTree, RevisionTree
857
995
# TODO: refactor this to use an existing revision object
858
996
# so we don't need to read it in twice.
859
997
if revision_id == None:
860
998
return EmptyTree()
862
1000
inv = self.get_revision_inventory(revision_id)
863
return RevisionTree(self.text_store, inv)
1001
return RevisionTree(self.weave_store, inv, revision_id)
866
1004
def working_tree(self):
867
1005
"""Return a `Tree` for the working copy."""
868
from workingtree import WorkingTree
869
return WorkingTree(self.base, self.read_working_inventory())
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())
872
1015
def basis_tree(self):
1060
def add_pending_merge(self, revision_id):
1061
from bzrlib.revision import validate_revision_id
1063
validate_revision_id(revision_id)
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?
1065
1204
p = self.pending_merges()
1066
if revision_id in p:
1068
p.append(revision_id)
1069
self.set_pending_merges(p)
1206
for rev_id in revision_ids:
1212
self.set_pending_merges(p)
1072
1214
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?
1073
1242
from bzrlib.atomicfile import AtomicFile
1074
1243
self.lock_write()
1076
f = AtomicFile(self.controlfilename('pending-merges'))
1245
f = AtomicFile(self.controlfilename('parent'))
1088
class ScratchBranch(Branch):
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):
1089
1275
"""Special test class: a branch that cleans up after itself.
1091
1277
>>> b = ScratchBranch()