15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
from copy import deepcopy
18
19
from cStringIO import StringIO
20
from bzrlib.lazy_import import lazy_import
21
lazy_import(globals(), """
20
from unittest import TestSuite
22
21
from warnings import warn
28
config as _mod_config,
33
revision as _mod_revision,
40
from bzrlib.config import BranchConfig, TreeConfig
41
from bzrlib.lockable_files import LockableFiles, TransportLock
42
from bzrlib.tag import (
24
from bzrlib import bzrdir, errors, lockdir, osutils, revision, \
28
from bzrlib.config import TreeConfig
48
29
from bzrlib.decorators import needs_read_lock, needs_write_lock
49
from bzrlib.errors import (BzrError, BzrCheckError, DivergedBranches,
50
HistoryMissing, InvalidRevisionId,
51
InvalidRevisionNumber, LockError, NoSuchFile,
30
import bzrlib.errors as errors
31
from bzrlib.errors import (BzrError, BzrCheckError, DivergedBranches,
32
HistoryMissing, InvalidRevisionId,
33
InvalidRevisionNumber, LockError, NoSuchFile,
52
34
NoSuchRevision, NoWorkingTree, NotVersionedError,
53
NotBranchError, UninitializableFormat,
54
UnlistableStore, UnlistableBranch,
35
NotBranchError, UninitializableFormat,
36
UnlistableStore, UnlistableBranch,
56
from bzrlib.hooks import Hooks
38
from bzrlib.lockable_files import LockableFiles, TransportLock
57
39
from bzrlib.symbol_versioning import (deprecated_function,
59
41
DEPRECATED_PARAMETER,
61
zero_eight, zero_nine, zero_sixteen,
43
zero_eight, zero_nine,
63
45
from bzrlib.trace import mutter, note
66
48
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
67
49
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
68
BZR_BRANCH_FORMAT_6 = "Bazaar Branch Format 6 (bzr 0.15)\n"
50
BZR_BRANCH_FORMAT_6 = "Bazaar-NG branch, format 6\n"
71
53
# TODO: Maybe include checks for common corruption of newlines, etc?
164
131
return bzrdir.BzrDir.create_standalone_workingtree(base).branch
166
@deprecated_function(zero_eight)
167
133
def setup_caching(self, cache_root):
168
134
"""Subclasses that care about caching should override this, and set
169
135
up cached stores located under cache_root.
171
NOTE: This is unused.
137
# seems to be unused, 2006-01-13 mbp
138
warn('%s is deprecated' % self.setup_caching)
139
self.cache_root = cache_root
175
141
def get_config(self):
176
return BranchConfig(self)
142
return bzrlib.config.BranchConfig(self)
178
144
def _get_nick(self):
179
145
return self.get_config().get_nickname()
181
147
def _set_nick(self, nick):
182
self.get_config().set_user_option('nickname', nick, warn_masked=True)
148
self.get_config().set_user_option('nickname', nick)
184
150
nick = property(_get_nick, _set_nick)
186
152
def is_locked(self):
187
raise NotImplementedError(self.is_locked)
153
raise NotImplementedError('is_locked is abstract')
189
155
def lock_write(self):
190
raise NotImplementedError(self.lock_write)
156
raise NotImplementedError('lock_write is abstract')
192
158
def lock_read(self):
193
raise NotImplementedError(self.lock_read)
159
raise NotImplementedError('lock_read is abstract')
195
161
def unlock(self):
196
raise NotImplementedError(self.unlock)
162
raise NotImplementedError('unlock is abstract')
198
164
def peek_lock_mode(self):
199
165
"""Return lock mode for the Branch: 'r', 'w' or None"""
200
166
raise NotImplementedError(self.peek_lock_mode)
202
168
def get_physical_lock_status(self):
203
raise NotImplementedError(self.get_physical_lock_status)
206
def get_revision_id_to_revno_map(self):
207
"""Return the revision_id => dotted revno map.
209
This will be regenerated on demand, but will be cached.
211
:return: A dictionary mapping revision_id => dotted revno.
212
This dictionary should not be modified by the caller.
214
if self._revision_id_to_revno_cache is not None:
215
mapping = self._revision_id_to_revno_cache
217
mapping = self._gen_revno_map()
218
self._cache_revision_id_to_revno(mapping)
219
# TODO: jam 20070417 Since this is being cached, should we be returning
221
# I would rather not, and instead just declare that users should not
222
# modify the return value.
225
def _gen_revno_map(self):
226
"""Create a new mapping from revision ids to dotted revnos.
228
Dotted revnos are generated based on the current tip in the revision
230
This is the worker function for get_revision_id_to_revno_map, which
231
just caches the return value.
233
:return: A dictionary mapping revision_id => dotted revno.
235
last_revision = self.last_revision()
236
revision_graph = self.repository.get_revision_graph(last_revision)
237
merge_sorted_revisions = tsort.merge_sort(
242
revision_id_to_revno = dict((rev_id, revno)
243
for seq_num, rev_id, depth, revno, end_of_merge
244
in merge_sorted_revisions)
245
return revision_id_to_revno
247
def leave_lock_in_place(self):
248
"""Tell this branch object not to release the physical lock when this
251
If lock_write doesn't return a token, then this method is not supported.
253
self.control_files.leave_in_place()
255
def dont_leave_lock_in_place(self):
256
"""Tell this branch object to release the physical lock when this
257
object is unlocked, even if it didn't originally acquire it.
259
If lock_write doesn't return a token, then this method is not supported.
261
self.control_files.dont_leave_in_place()
169
raise NotImplementedError('get_physical_lock_status is abstract')
263
171
def abspath(self, name):
264
172
"""Return absolute filename for something in the branch
363
269
raise InvalidRevisionNumber(revno)
364
270
return self.repository.get_revision_delta(rh[revno-1])
366
@deprecated_method(zero_sixteen)
367
272
def get_root_id(self):
368
"""Return the id of this branches root
370
Deprecated: branches don't have root ids-- trees do.
371
Use basis_tree().get_root_id() instead.
373
raise NotImplementedError(self.get_root_id)
273
"""Return the id of this branches root"""
274
raise NotImplementedError('get_root_id is abstract')
375
276
def print_file(self, file, revision_id):
376
277
"""Print `file` to stdout."""
377
raise NotImplementedError(self.print_file)
278
raise NotImplementedError('print_file is abstract')
379
280
def append_revision(self, *revision_ids):
380
raise NotImplementedError(self.append_revision)
281
raise NotImplementedError('append_revision is abstract')
382
283
def set_revision_history(self, rev_history):
383
raise NotImplementedError(self.set_revision_history)
385
def _cache_revision_history(self, rev_history):
386
"""Set the cached revision history to rev_history.
388
The revision_history method will use this cache to avoid regenerating
389
the revision history.
391
This API is semi-public; it only for use by subclasses, all other code
392
should consider it to be private.
394
self._revision_history_cache = rev_history
396
def _cache_revision_id_to_revno(self, revision_id_to_revno):
397
"""Set the cached revision_id => revno map to revision_id_to_revno.
399
This API is semi-public; it only for use by subclasses, all other code
400
should consider it to be private.
402
self._revision_id_to_revno_cache = revision_id_to_revno
404
def _clear_cached_state(self):
405
"""Clear any cached data on this branch, e.g. cached revision history.
407
This means the next call to revision_history will need to call
408
_gen_revision_history.
410
This API is semi-public; it only for use by subclasses, all other code
411
should consider it to be private.
413
self._revision_history_cache = None
414
self._revision_id_to_revno_cache = None
416
def _gen_revision_history(self):
417
"""Return sequence of revision hashes on to this branch.
419
Unlike revision_history, this method always regenerates or rereads the
420
revision history, i.e. it does not cache the result, so repeated calls
423
Concrete subclasses should override this instead of revision_history so
424
that subclasses do not need to deal with caching logic.
426
This API is semi-public; it only for use by subclasses, all other code
427
should consider it to be private.
429
raise NotImplementedError(self._gen_revision_history)
284
raise NotImplementedError('set_revision_history is abstract')
432
286
def revision_history(self):
433
"""Return sequence of revision hashes on to this branch.
435
This method will cache the revision history for as long as it is safe to
438
if self._revision_history_cache is not None:
439
history = self._revision_history_cache
441
history = self._gen_revision_history()
442
self._cache_revision_history(history)
287
"""Return sequence of revision hashes on to this branch."""
288
raise NotImplementedError('revision_history is abstract')
446
291
"""Return current revision number for this branch.
767
571
mainline_parent_id = revision_id
768
572
return BranchCheckResult(self)
770
def _get_checkout_format(self):
771
"""Return the most suitable metadir for a checkout of this branch.
772
Weaves are used if this branch's repository uses weaves.
774
if isinstance(self.bzrdir, bzrdir.BzrDirPreSplitOut):
775
from bzrlib.repofmt import weaverepo
776
format = bzrdir.BzrDirMetaFormat1()
777
format.repository_format = weaverepo.RepositoryFormat7()
779
format = self.repository.bzrdir.checkout_metadir()
780
format.set_branch_format(self._format)
783
def create_checkout(self, to_location, revision_id=None,
785
"""Create a checkout of a branch.
787
:param to_location: The url to produce the checkout at
788
:param revision_id: The revision to check out
789
:param lightweight: If True, produce a lightweight checkout, otherwise,
790
produce a bound branch (heavyweight checkout)
791
:return: The tree of the created checkout
793
t = transport.get_transport(to_location)
796
format = self._get_checkout_format()
797
checkout = format.initialize_on_transport(t)
798
BranchReferenceFormat().initialize(checkout, self)
800
format = self._get_checkout_format()
801
checkout_branch = bzrdir.BzrDir.create_branch_convenience(
802
to_location, force_new_tree=False, format=format)
803
checkout = checkout_branch.bzrdir
804
checkout_branch.bind(self)
805
# pull up to the specified revision_id to set the initial
806
# branch tip correctly, and seed it with history.
807
checkout_branch.pull(self, stop_revision=revision_id)
808
tree = checkout.create_workingtree(revision_id)
809
basis_tree = tree.basis_tree()
810
basis_tree.lock_read()
812
for path, file_id in basis_tree.iter_references():
813
reference_parent = self.reference_parent(file_id, path)
814
reference_parent.create_checkout(tree.abspath(path),
815
basis_tree.get_reference_revision(file_id, path),
821
def reference_parent(self, file_id, path):
822
"""Return the parent branch for a tree-reference file_id
823
:param file_id: The file_id of the tree reference
824
:param path: The path of the file_id in the tree
825
:return: A branch associated with the file_id
827
# FIXME should provide multiple branches, based on config
828
return Branch.open(self.bzrdir.root_transport.clone(path).base)
830
def supports_tags(self):
831
return self._format.supports_tags()
834
575
class BranchFormat(object):
835
576
"""An encapsulation of the initialization and open routines for a format.
967
658
def __str__(self):
968
659
return self.get_format_string().rstrip()
970
def supports_tags(self):
971
"""True if this format supports tags stored in the branch"""
972
return False # by default
974
# XXX: Probably doesn't really belong here -- mbp 20070212
975
def _initialize_control_files(self, a_bzrdir, utf8_files, lock_filename,
977
branch_transport = a_bzrdir.get_branch_transport(self)
978
control_files = lockable_files.LockableFiles(branch_transport,
979
lock_filename, lock_class)
980
control_files.create_lock()
981
control_files.lock_write()
983
for filename, content in utf8_files:
984
control_files.put_utf8(filename, content)
986
control_files.unlock()
989
class BranchHooks(Hooks):
990
"""A dictionary mapping hook name to a list of callables for branch hooks.
992
e.g. ['set_rh'] Is the list of items to be called when the
993
set_revision_history function is invoked.
997
"""Create the default hooks.
999
These are all empty initially, because by default nothing should get
1002
Hooks.__init__(self)
1003
# Introduced in 0.15:
1004
# invoked whenever the revision history has been set
1005
# with set_revision_history. The api signature is
1006
# (branch, revision_history), and the branch will
1009
# invoked after a push operation completes.
1010
# the api signature is
1012
# containing the members
1013
# (source, local, master, old_revno, old_revid, new_revno, new_revid)
1014
# where local is the local target branch or None, master is the target
1015
# master branch, and the rest should be self explanatory. The source
1016
# is read locked and the target branches write locked. Source will
1017
# be the local low-latency branch.
1018
self['post_push'] = []
1019
# invoked after a pull operation completes.
1020
# the api signature is
1022
# containing the members
1023
# (source, local, master, old_revno, old_revid, new_revno, new_revid)
1024
# where local is the local branch or None, master is the target
1025
# master branch, and the rest should be self explanatory. The source
1026
# is read locked and the target branches write locked. The local
1027
# branch is the low-latency branch.
1028
self['post_pull'] = []
1029
# invoked after a commit operation completes.
1030
# the api signature is
1031
# (local, master, old_revno, old_revid, new_revno, new_revid)
1032
# old_revid is NULL_REVISION for the first commit to a branch.
1033
self['post_commit'] = []
1034
# invoked after a uncommit operation completes.
1035
# the api signature is
1036
# (local, master, old_revno, old_revid, new_revno, new_revid) where
1037
# local is the local branch or None, master is the target branch,
1038
# and an empty branch recieves new_revno of 0, new_revid of None.
1039
self['post_uncommit'] = []
1042
# install the default hooks into the Branch class.
1043
Branch.hooks = BranchHooks()
1046
662
class BzrBranchFormat4(BranchFormat):
1047
663
"""Bzr branch format 4.
1280
861
it's writable, and can be accessed via the normal filesystem API.
1283
def __init__(self, _format=None,
864
def __init__(self, transport=DEPRECATED_PARAMETER, init=DEPRECATED_PARAMETER,
865
relax_version_check=DEPRECATED_PARAMETER, _format=None,
1284
866
_control_files=None, a_bzrdir=None, _repository=None):
1285
"""Create new branch object at a particular location."""
1286
Branch.__init__(self)
867
"""Create new branch object at a particular location.
869
transport -- A Transport object, defining how to access files.
871
init -- If True, create new control files in a previously
872
unversioned directory. If False, the branch must already
875
relax_version_check -- If true, the usual check for the branch
876
version is not applied. This is intended only for
877
upgrade/recovery type use; it's not guaranteed that
878
all operations will work on old format branches.
1287
880
if a_bzrdir is None:
1288
raise ValueError('a_bzrdir must be supplied')
881
self.bzrdir = bzrdir.BzrDir.open(transport.base)
1290
883
self.bzrdir = a_bzrdir
1291
# self._transport used to point to the directory containing the
1292
# control directory, but was not used - now it's just the transport
1293
# for the branch control files. mbp 20070212
1294
self._base = self.bzrdir.transport.clone('..').base
884
self._transport = self.bzrdir.transport.clone('..')
885
self._base = self._transport.base
1295
886
self._format = _format
1296
887
if _control_files is None:
1297
888
raise ValueError('BzrBranch _control_files is None')
1298
889
self.control_files = _control_files
1299
self._transport = _control_files._transport
890
if deprecated_passed(init):
891
warn("BzrBranch.__init__(..., init=XXX): The init parameter is "
892
"deprecated as of bzr 0.8. Please use Branch.create().",
896
# this is slower than before deprecation, oh well never mind.
898
self._initialize(transport.base)
899
self._check_format(_format)
900
if deprecated_passed(relax_version_check):
901
warn("BzrBranch.__init__(..., relax_version_check=XXX_: The "
902
"relax_version_check parameter is deprecated as of bzr 0.8. "
903
"Please use BzrDir.open_downlevel, or a BzrBranchFormat's "
907
if (not relax_version_check
908
and not self._format.is_supported()):
909
raise errors.UnsupportedFormatError(format=fmt)
910
if deprecated_passed(transport):
911
warn("BzrBranch.__init__(transport=XXX...): The transport "
912
"parameter is deprecated as of bzr 0.8. "
913
"Please use Branch.open, or bzrdir.open_branch().",
1300
916
self.repository = _repository
1302
918
def __str__(self):
1369
1026
@needs_write_lock
1370
1027
def append_revision(self, *revision_ids):
1371
1028
"""See Branch.append_revision."""
1372
revision_ids = [osutils.safe_revision_id(r) for r in revision_ids]
1373
1029
for revision_id in revision_ids:
1374
_mod_revision.check_not_reserved_id(revision_id)
1375
1030
mutter("add {%s} to revision-history" % revision_id)
1376
1031
rev_history = self.revision_history()
1377
1032
rev_history.extend(revision_ids)
1378
1033
self.set_revision_history(rev_history)
1380
def _write_revision_history(self, history):
1381
"""Factored out of set_revision_history.
1383
This performs the actual writing to disk.
1384
It is intended to be called by BzrBranch5.set_revision_history."""
1385
self.control_files.put_bytes(
1386
'revision-history', '\n'.join(history))
1388
1035
@needs_write_lock
1389
1036
def set_revision_history(self, rev_history):
1390
1037
"""See Branch.set_revision_history."""
1391
rev_history = [osutils.safe_revision_id(r) for r in rev_history]
1392
self._clear_cached_state()
1393
self._write_revision_history(rev_history)
1394
self._cache_revision_history(rev_history)
1395
for hook in Branch.hooks['set_rh']:
1396
hook(self, rev_history)
1038
self.control_files.put_utf8(
1039
'revision-history', '\n'.join(rev_history))
1040
transaction = self.get_transaction()
1041
history = transaction.map.find_revision_history()
1042
if history is not None:
1043
# update the revision history in the identity map.
1044
history[:] = list(rev_history)
1045
# this call is disabled because revision_history is
1046
# not really an object yet, and the transaction is for objects.
1047
# transaction.register_dirty(history)
1049
transaction.map.add_revision_history(rev_history)
1050
# this call is disabled because revision_history is
1051
# not really an object yet, and the transaction is for objects.
1052
# transaction.register_clean(history)
1055
def revision_history(self):
1056
"""See Branch.revision_history."""
1057
transaction = self.get_transaction()
1058
history = transaction.map.find_revision_history()
1059
if history is not None:
1060
mutter("cache hit for revision-history in %s", self)
1061
return list(history)
1062
history = [l.rstrip('\r\n') for l in
1063
self.control_files.get_utf8('revision-history').readlines()]
1064
transaction.map.add_revision_history(history)
1065
# this call is disabled because revision_history is
1066
# not really an object yet, and the transaction is for objects.
1067
# transaction.register_clean(history, precious=True)
1068
return list(history)
1398
1070
@needs_write_lock
1399
def set_last_revision_info(self, revno, revision_id):
1400
revision_id = osutils.safe_revision_id(revision_id)
1401
history = self._lefthand_history(revision_id)
1402
assert len(history) == revno, '%d != %d' % (len(history), revno)
1403
self.set_revision_history(history)
1405
def _gen_revision_history(self):
1406
history = self.control_files.get('revision-history').read().split('\n')
1407
if history[-1:] == ['']:
1408
# There shouldn't be a trailing newline, but just in case.
1412
def _lefthand_history(self, revision_id, last_rev=None,
1071
def generate_revision_history(self, revision_id, last_rev=None,
1073
"""Create a new revision history that will finish with revision_id.
1075
:param revision_id: the new tip to use.
1076
:param last_rev: The previous last_revision. If not None, then this
1077
must be a ancestory of revision_id, or DivergedBranches is raised.
1078
:param other_branch: The other branch that DivergedBranches should
1079
raise with respect to.
1414
1081
# stop_revision must be a descendant of last_revision
1415
1082
stop_graph = self.repository.get_revision_graph(revision_id)
1416
if (last_rev is not None and last_rev != _mod_revision.NULL_REVISION
1417
and last_rev not in stop_graph):
1083
if last_rev is not None and last_rev not in stop_graph:
1418
1084
# our previous tip is not merged into stop_revision
1419
1085
raise errors.DivergedBranches(self, other_branch)
1420
1086
# make a new revision history from the graph
1421
1087
current_rev_id = revision_id
1422
1088
new_history = []
1423
while current_rev_id not in (None, _mod_revision.NULL_REVISION):
1089
while current_rev_id not in (None, revision.NULL_REVISION):
1424
1090
new_history.append(current_rev_id)
1425
1091
current_rev_id_parents = stop_graph[current_rev_id]
1488
1136
return self.bzrdir.open_workingtree()
1490
1138
@needs_write_lock
1491
def pull(self, source, overwrite=False, stop_revision=None,
1492
_hook_master=None, run_hooks=True):
1495
:param _hook_master: Private parameter - set the branch to
1496
be supplied as the master to push hooks.
1497
:param run_hooks: Private parameter - if false, this branch
1498
is being called because it's the master of the primary branch,
1499
so it should not run its hooks.
1501
result = PullResult()
1502
result.source_branch = source
1503
result.target_branch = self
1139
def pull(self, source, overwrite=False, stop_revision=None):
1140
"""See Branch.pull."""
1504
1141
source.lock_read()
1506
result.old_revno, result.old_revid = self.last_revision_info()
1143
old_count = len(self.revision_history())
1508
self.update_revisions(source, stop_revision)
1145
self.update_revisions(source,stop_revision)
1509
1146
except DivergedBranches:
1510
1147
if not overwrite:
1513
if stop_revision is None:
1514
stop_revision = source.last_revision()
1515
self.generate_revision_history(stop_revision)
1516
result.tag_conflicts = source.tags.merge_to(self.tags)
1517
result.new_revno, result.new_revid = self.last_revision_info()
1519
result.master_branch = _hook_master
1520
result.local_branch = self
1522
result.master_branch = self
1523
result.local_branch = None
1525
for hook in Branch.hooks['post_pull']:
1150
self.set_revision_history(source.revision_history())
1151
new_count = len(self.revision_history())
1152
return new_count - old_count
1528
1154
source.unlock()
1531
def _get_parent_location(self):
1156
def get_parent(self):
1157
"""See Branch.get_parent."""
1532
1159
_locs = ['parent', 'pull', 'x-pull']
1160
assert self.base[-1] == '/'
1533
1161
for l in _locs:
1535
return self.control_files.get(l).read().strip('\n')
1163
parent = self.control_files.get(l).read().strip('\n')
1536
1164
except NoSuchFile:
1541
def push(self, target, overwrite=False, stop_revision=None,
1542
_override_hook_source_branch=None):
1545
This is the basic concrete implementation of push()
1547
:param _override_hook_source_branch: If specified, run
1548
the hooks passing this Branch as the source, rather than self.
1549
This is for use of RemoteBranch, where push is delegated to the
1550
underlying vfs-based Branch.
1552
# TODO: Public option to disable running hooks - should be trivial but
1556
result = self._push_with_bound_branches(target, overwrite,
1558
_override_hook_source_branch=_override_hook_source_branch)
1563
def _push_with_bound_branches(self, target, overwrite,
1565
_override_hook_source_branch=None):
1566
"""Push from self into target, and into target's master if any.
1568
This is on the base BzrBranch class even though it doesn't support
1569
bound branches because the *target* might be bound.
1572
if _override_hook_source_branch:
1573
result.source_branch = _override_hook_source_branch
1574
for hook in Branch.hooks['post_push']:
1577
bound_location = target.get_bound_location()
1578
if bound_location and target.base != bound_location:
1579
# there is a master branch.
1581
# XXX: Why the second check? Is it even supported for a branch to
1582
# be bound to itself? -- mbp 20070507
1583
master_branch = target.get_master_branch()
1584
master_branch.lock_write()
1586
# push into the master from this branch.
1587
self._basic_push(master_branch, overwrite, stop_revision)
1588
# and push into the target branch from this. Note that we push from
1589
# this branch again, because its considered the highest bandwidth
1591
result = self._basic_push(target, overwrite, stop_revision)
1592
result.master_branch = master_branch
1593
result.local_branch = target
1597
master_branch.unlock()
1600
result = self._basic_push(target, overwrite, stop_revision)
1601
# TODO: Why set master_branch and local_branch if there's no
1602
# binding? Maybe cleaner to just leave them unset? -- mbp
1604
result.master_branch = target
1605
result.local_branch = None
1609
def _basic_push(self, target, overwrite, stop_revision):
1610
"""Basic implementation of push without bound branches or hooks.
1612
Must be called with self read locked and target write locked.
1614
result = PushResult()
1615
result.source_branch = self
1616
result.target_branch = target
1617
result.old_revno, result.old_revid = target.last_revision_info()
1619
target.update_revisions(self, stop_revision)
1620
except DivergedBranches:
1624
target.set_revision_history(self.revision_history())
1625
result.tag_conflicts = self.tags.merge_to(target.tags)
1626
result.new_revno, result.new_revid = target.last_revision_info()
1629
def get_parent(self):
1630
"""See Branch.get_parent."""
1632
assert self.base[-1] == '/'
1633
parent = self._get_parent_location()
1636
# This is an old-format absolute path to a local branch
1637
# turn it into a url
1638
if parent.startswith('/'):
1639
parent = urlutils.local_path_to_url(parent.decode('utf8'))
1166
# This is an old-format absolute path to a local branch
1167
# turn it into a url
1168
if parent.startswith('/'):
1169
parent = urlutils.local_path_to_url(parent.decode('utf8'))
1641
1170
return urlutils.join(self.base[:-1], parent)
1642
except errors.InvalidURLJoin, e:
1643
raise errors.InaccessibleParent(parent, self.base)
1173
def get_push_location(self):
1174
"""See Branch.get_push_location."""
1175
push_loc = self.get_config().get_user_option('push_location')
1645
1178
def set_push_location(self, location):
1646
1179
"""See Branch.set_push_location."""
1647
self.get_config().set_user_option(
1648
'push_location', location,
1649
store=_mod_config.STORE_LOCATION_NORECURSE)
1180
self.get_config().set_user_option('push_location', location,
1651
1183
@needs_write_lock
1652
1184
def set_parent(self, url):
1823
1339
if master is not None:
1824
1340
old_tip = self.last_revision()
1825
1341
self.pull(master, overwrite=True)
1826
if old_tip in self.repository.get_ancestry(
1827
_mod_revision.ensure_null(self.last_revision()),
1342
if old_tip in self.repository.get_ancestry(self.last_revision()):
1834
class BzrBranchExperimental(BzrBranch5):
1835
"""Bzr experimental branch format
1838
- a revision-history file.
1840
- a lock dir guarding the branch itself
1841
- all of this stored in a branch/ subdirectory
1842
- works with shared repositories.
1843
- a tag dictionary in the branch
1845
This format is new in bzr 0.15, but shouldn't be used for real data,
1848
This class acts as it's own BranchFormat.
1851
_matchingbzrdir = bzrdir.BzrDirMetaFormat1()
1854
def get_format_string(cls):
1855
"""See BranchFormat.get_format_string()."""
1856
return "Bazaar-NG branch format experimental\n"
1859
def get_format_description(cls):
1860
"""See BranchFormat.get_format_description()."""
1861
return "Experimental branch format"
1864
def get_reference(cls, a_bzrdir):
1865
"""Get the target reference of the branch in a_bzrdir.
1867
format probing must have been completed before calling
1868
this method - it is assumed that the format of the branch
1869
in a_bzrdir is correct.
1871
:param a_bzrdir: The bzrdir to get the branch data from.
1872
:return: None if the branch is not a reference branch.
1877
def _initialize_control_files(cls, a_bzrdir, utf8_files, lock_filename,
1879
branch_transport = a_bzrdir.get_branch_transport(cls)
1880
control_files = lockable_files.LockableFiles(branch_transport,
1881
lock_filename, lock_class)
1882
control_files.create_lock()
1883
control_files.lock_write()
1885
for filename, content in utf8_files:
1886
control_files.put_utf8(filename, content)
1888
control_files.unlock()
1891
def initialize(cls, a_bzrdir):
1892
"""Create a branch of this format in a_bzrdir."""
1893
utf8_files = [('format', cls.get_format_string()),
1894
('revision-history', ''),
1895
('branch-name', ''),
1898
cls._initialize_control_files(a_bzrdir, utf8_files,
1899
'lock', lockdir.LockDir)
1900
return cls.open(a_bzrdir, _found=True)
1903
def open(cls, a_bzrdir, _found=False):
1904
"""Return the branch object for a_bzrdir
1906
_found is a private parameter, do not use it. It is used to indicate
1907
if format probing has already be done.
1910
format = BranchFormat.find_format(a_bzrdir)
1911
assert format.__class__ == cls
1912
transport = a_bzrdir.get_branch_transport(None)
1913
control_files = lockable_files.LockableFiles(transport, 'lock',
1915
return cls(_format=cls,
1916
_control_files=control_files,
1918
_repository=a_bzrdir.find_repository())
1921
def is_supported(cls):
1924
def _make_tags(self):
1925
return BasicTags(self)
1928
def supports_tags(cls):
1932
BranchFormat.register_format(BzrBranchExperimental)
1935
class BzrBranch6(BzrBranch5):
1938
def last_revision_info(self):
1939
revision_string = self.control_files.get('last-revision').read()
1940
revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
1941
revision_id = cache_utf8.get_cached_utf8(revision_id)
1943
return revno, revision_id
1945
def last_revision(self):
1946
"""Return last revision id, or None"""
1947
revision_id = self.last_revision_info()[1]
1948
if revision_id == _mod_revision.NULL_REVISION:
1952
def _write_last_revision_info(self, revno, revision_id):
1953
"""Simply write out the revision id, with no checks.
1955
Use set_last_revision_info to perform this safely.
1957
Does not update the revision_history cache.
1958
Intended to be called by set_last_revision_info and
1959
_write_revision_history.
1961
if revision_id is None:
1962
revision_id = 'null:'
1963
out_string = '%d %s\n' % (revno, revision_id)
1964
self.control_files.put_bytes('last-revision', out_string)
1967
def set_last_revision_info(self, revno, revision_id):
1968
revision_id = osutils.safe_revision_id(revision_id)
1969
if self._get_append_revisions_only():
1970
self._check_history_violation(revision_id)
1971
self._write_last_revision_info(revno, revision_id)
1972
self._clear_cached_state()
1974
def _check_history_violation(self, revision_id):
1975
last_revision = _mod_revision.ensure_null(self.last_revision())
1976
if _mod_revision.is_null(last_revision):
1978
if last_revision not in self._lefthand_history(revision_id):
1979
raise errors.AppendRevisionsOnlyViolation(self.base)
1981
def _gen_revision_history(self):
1982
"""Generate the revision history from last revision
1984
history = list(self.repository.iter_reverse_revision_history(
1985
self.last_revision()))
1989
def _write_revision_history(self, history):
1990
"""Factored out of set_revision_history.
1992
This performs the actual writing to disk, with format-specific checks.
1993
It is intended to be called by BzrBranch5.set_revision_history.
1995
if len(history) == 0:
1996
last_revision = 'null:'
1998
if history != self._lefthand_history(history[-1]):
1999
raise errors.NotLefthandHistory(history)
2000
last_revision = history[-1]
2001
if self._get_append_revisions_only():
2002
self._check_history_violation(last_revision)
2003
self._write_last_revision_info(len(history), last_revision)
2006
def append_revision(self, *revision_ids):
2007
revision_ids = [osutils.safe_revision_id(r) for r in revision_ids]
2008
if len(revision_ids) == 0:
2010
prev_revno, prev_revision = self.last_revision_info()
2011
for revision in self.repository.get_revisions(revision_ids):
2012
if prev_revision == _mod_revision.NULL_REVISION:
2013
if revision.parent_ids != []:
2014
raise errors.NotLeftParentDescendant(self, prev_revision,
2015
revision.revision_id)
2017
if revision.parent_ids[0] != prev_revision:
2018
raise errors.NotLeftParentDescendant(self, prev_revision,
2019
revision.revision_id)
2020
prev_revision = revision.revision_id
2021
self.set_last_revision_info(prev_revno + len(revision_ids),
2025
def _set_parent_location(self, url):
2026
"""Set the parent branch"""
2027
self._set_config_location('parent_location', url, make_relative=True)
2030
def _get_parent_location(self):
2031
"""Set the parent branch"""
2032
return self._get_config_location('parent_location')
2034
def set_push_location(self, location):
2035
"""See Branch.set_push_location."""
2036
self._set_config_location('push_location', location)
2038
def set_bound_location(self, location):
2039
"""See Branch.set_push_location."""
2041
config = self.get_config()
2042
if location is None:
2043
if config.get_user_option('bound') != 'True':
2046
config.set_user_option('bound', 'False', warn_masked=True)
2049
self._set_config_location('bound_location', location,
2051
config.set_user_option('bound', 'True', warn_masked=True)
2054
def _get_bound_location(self, bound):
2055
"""Return the bound location in the config file.
2057
Return None if the bound parameter does not match"""
2058
config = self.get_config()
2059
config_bound = (config.get_user_option('bound') == 'True')
2060
if config_bound != bound:
2062
return self._get_config_location('bound_location', config=config)
2064
def get_bound_location(self):
2065
"""See Branch.set_push_location."""
2066
return self._get_bound_location(True)
2068
def get_old_bound_location(self):
2069
"""See Branch.get_old_bound_location"""
2070
return self._get_bound_location(False)
2072
def set_append_revisions_only(self, enabled):
2077
self.get_config().set_user_option('append_revisions_only', value,
2080
def _get_append_revisions_only(self):
2081
value = self.get_config().get_user_option('append_revisions_only')
2082
return value == 'True'
2084
def _synchronize_history(self, destination, revision_id):
2085
"""Synchronize last revision and revision history between branches.
2087
This version is most efficient when the destination is also a
2088
BzrBranch6, but works for BzrBranch5, as long as the destination's
2089
repository contains all the lefthand ancestors of the intended
2090
last_revision. If not, set_last_revision_info will fail.
2092
:param destination: The branch to copy the history into
2093
:param revision_id: The revision-id to truncate history at. May
2094
be None to copy complete history.
2096
if revision_id is None:
2097
revno, revision_id = self.last_revision_info()
2099
# To figure out the revno for a random revision, we need to build
2100
# the revision history, and count its length.
2101
# We don't care about the order, just how long it is.
2102
# Alternatively, we could start at the current location, and count
2103
# backwards. But there is no guarantee that we will find it since
2104
# it may be a merged revision.
2105
revno = len(list(self.repository.iter_reverse_revision_history(
2107
destination.set_last_revision_info(revno, revision_id)
2109
def _make_tags(self):
2110
return BasicTags(self)
2113
######################################################################
2114
# results of operations
2117
class _Result(object):
2119
def _show_tag_conficts(self, to_file):
2120
if not getattr(self, 'tag_conflicts', None):
2122
to_file.write('Conflicting tags:\n')
2123
for name, value1, value2 in self.tag_conflicts:
2124
to_file.write(' %s\n' % (name, ))
2127
class PullResult(_Result):
2128
"""Result of a Branch.pull operation.
2130
:ivar old_revno: Revision number before pull.
2131
:ivar new_revno: Revision number after pull.
2132
:ivar old_revid: Tip revision id before pull.
2133
:ivar new_revid: Tip revision id after pull.
2134
:ivar source_branch: Source (local) branch object.
2135
:ivar master_branch: Master branch of the target, or None.
2136
:ivar target_branch: Target/destination branch object.
2140
# DEPRECATED: pull used to return the change in revno
2141
return self.new_revno - self.old_revno
2143
def report(self, to_file):
2144
if self.old_revid == self.new_revid:
2145
to_file.write('No revisions to pull.\n')
2147
to_file.write('Now on revision %d.\n' % self.new_revno)
2148
self._show_tag_conficts(to_file)
2151
class PushResult(_Result):
2152
"""Result of a Branch.push operation.
2154
:ivar old_revno: Revision number before push.
2155
:ivar new_revno: Revision number after push.
2156
:ivar old_revid: Tip revision id before push.
2157
:ivar new_revid: Tip revision id after push.
2158
:ivar source_branch: Source branch object.
2159
:ivar master_branch: Master branch of the target, or None.
2160
:ivar target_branch: Target/destination branch object.
2164
# DEPRECATED: push used to return the change in revno
2165
return self.new_revno - self.old_revno
2167
def report(self, to_file):
2168
"""Write a human-readable description of the result."""
2169
if self.old_revid == self.new_revid:
2170
to_file.write('No new revisions to push.\n')
2172
to_file.write('Pushed up to revision %d.\n' % self.new_revno)
2173
self._show_tag_conficts(to_file)
1348
class BranchTestProviderAdapter(object):
1349
"""A tool to generate a suite testing multiple branch formats at once.
1351
This is done by copying the test once for each transport and injecting
1352
the transport_server, transport_readonly_server, and branch_format
1353
classes into each copy. Each copy is also given a new id() to make it
1357
def __init__(self, transport_server, transport_readonly_server, formats):
1358
self._transport_server = transport_server
1359
self._transport_readonly_server = transport_readonly_server
1360
self._formats = formats
1362
def adapt(self, test):
1363
result = TestSuite()
1364
for branch_format, bzrdir_format in self._formats:
1365
new_test = deepcopy(test)
1366
new_test.transport_server = self._transport_server
1367
new_test.transport_readonly_server = self._transport_readonly_server
1368
new_test.bzrdir_format = bzrdir_format
1369
new_test.branch_format = branch_format
1370
def make_new_test_id():
1371
new_id = "%s(%s)" % (new_test.id(), branch_format.__class__.__name__)
1372
return lambda: new_id
1373
new_test.id = make_new_test_id()
1374
result.addTest(new_test)
2176
1378
class BranchCheckResult(object):
added
removed
bzrlib/branch.py
