126
125
raise errors.UnstackableRepositoryFormat(self.repository._format,
127
126
self.repository.base)
129
def _extend_partial_history(self, stop_index=None, stop_revision=None):
130
"""Extend the partial history to include a given index
132
If a stop_index is supplied, stop when that index has been reached.
133
If a stop_revision is supplied, stop when that revision is
134
encountered. Otherwise, stop when the beginning of history is
137
:param stop_index: The index which should be present. When it is
138
present, history extension will stop.
139
:param stop_revision: The revision id which should be present. When
140
it is encountered, history extension will stop.
142
if len(self._partial_revision_history_cache) == 0:
143
self._partial_revision_history_cache = [self.last_revision()]
144
repository._iter_for_revno(
145
self.repository, self._partial_revision_history_cache,
146
stop_index=stop_index, stop_revision=stop_revision)
147
if self._partial_revision_history_cache[-1] == _mod_revision.NULL_REVISION:
148
self._partial_revision_history_cache.pop()
150
def _get_check_refs(self):
151
"""Get the references needed for check().
155
revid = self.last_revision()
156
return [('revision-existence', revid), ('lefthand-distance', revid)]
159
129
def open(base, _unsupported=False, possible_transports=None):
160
130
"""Open the branch rooted at base.
457
414
# start_revision_id.
458
415
if self._merge_sorted_revisions_cache is None:
459
416
last_revision = self.last_revision()
460
known_graph = self.repository.get_known_graph_ancestry(
462
self._merge_sorted_revisions_cache = known_graph.merge_sort(
417
graph = self.repository.get_graph()
418
parent_map = dict(((key, value) for key, value in
419
graph.iter_ancestry([last_revision]) if value is not None))
420
revision_graph = repository._strip_NULL_ghosts(parent_map)
421
revs = tsort.merge_sort(revision_graph, last_revision, None,
423
# Drop the sequence # before caching
424
self._merge_sorted_revisions_cache = [r[1:] for r in revs]
464
426
filtered = self._filter_merge_sorted_revisions(
465
427
self._merge_sorted_revisions_cache, start_revision_id,
466
428
stop_revision_id, stop_rule)
467
# Make sure we don't return revisions that are not part of the
468
# start_revision_id ancestry.
469
filtered = self._filter_start_non_ancestors(filtered)
470
429
if direction == 'reverse':
472
431
if direction == 'forward':
479
438
"""Iterate over an inclusive range of sorted revisions."""
480
439
rev_iter = iter(merge_sorted_revisions)
481
440
if start_revision_id is not None:
482
for node in rev_iter:
483
rev_id = node.key[-1]
441
for rev_id, depth, revno, end_of_merge in rev_iter:
484
442
if rev_id != start_revision_id:
487
445
# The decision to include the start or not
488
446
# depends on the stop_rule if a stop is provided
489
# so pop this node back into the iterator
490
rev_iter = chain(iter([node]), rev_iter)
448
iter([(rev_id, depth, revno, end_of_merge)]),
492
451
if stop_revision_id is None:
494
for node in rev_iter:
495
rev_id = node.key[-1]
496
yield (rev_id, node.merge_depth, node.revno,
452
for rev_id, depth, revno, end_of_merge in rev_iter:
453
yield rev_id, depth, revno, end_of_merge
498
454
elif stop_rule == 'exclude':
499
for node in rev_iter:
500
rev_id = node.key[-1]
455
for rev_id, depth, revno, end_of_merge in rev_iter:
501
456
if rev_id == stop_revision_id:
503
yield (rev_id, node.merge_depth, node.revno,
458
yield rev_id, depth, revno, end_of_merge
505
459
elif stop_rule == 'include':
506
for node in rev_iter:
507
rev_id = node.key[-1]
508
yield (rev_id, node.merge_depth, node.revno,
460
for rev_id, depth, revno, end_of_merge in rev_iter:
461
yield rev_id, depth, revno, end_of_merge
510
462
if rev_id == stop_revision_id:
512
elif stop_rule == 'with-merges-without-common-ancestry':
513
# We want to exclude all revisions that are already part of the
514
# stop_revision_id ancestry.
515
graph = self.repository.get_graph()
516
ancestors = graph.find_unique_ancestors(start_revision_id,
518
for node in rev_iter:
519
rev_id = node.key[-1]
520
if rev_id not in ancestors:
522
yield (rev_id, node.merge_depth, node.revno,
524
464
elif stop_rule == 'with-merges':
525
465
stop_rev = self.repository.get_revision(stop_revision_id)
526
466
if stop_rev.parent_ids:
527
467
left_parent = stop_rev.parent_ids[0]
529
469
left_parent = _mod_revision.NULL_REVISION
530
# left_parent is the actual revision we want to stop logging at,
531
# since we want to show the merged revisions after the stop_rev too
532
reached_stop_revision_id = False
533
revision_id_whitelist = []
534
for node in rev_iter:
535
rev_id = node.key[-1]
470
for rev_id, depth, revno, end_of_merge in rev_iter:
536
471
if rev_id == left_parent:
537
# reached the left parent after the stop_revision
539
if (not reached_stop_revision_id or
540
rev_id in revision_id_whitelist):
541
yield (rev_id, node.merge_depth, node.revno,
543
if reached_stop_revision_id or rev_id == stop_revision_id:
544
# only do the merged revs of rev_id from now on
545
rev = self.repository.get_revision(rev_id)
547
reached_stop_revision_id = True
548
revision_id_whitelist.extend(rev.parent_ids)
473
yield rev_id, depth, revno, end_of_merge
550
475
raise ValueError('invalid stop_rule %r' % stop_rule)
552
def _filter_start_non_ancestors(self, rev_iter):
553
# If we started from a dotted revno, we want to consider it as a tip
554
# and don't want to yield revisions that are not part of its
555
# ancestry. Given the order guaranteed by the merge sort, we will see
556
# uninteresting descendants of the first parent of our tip before the
558
first = rev_iter.next()
559
(rev_id, merge_depth, revno, end_of_merge) = first
562
# We start at a mainline revision so by definition, all others
563
# revisions in rev_iter are ancestors
564
for node in rev_iter:
569
pmap = self.repository.get_parent_map([rev_id])
570
parents = pmap.get(rev_id, [])
572
whitelist.update(parents)
574
# If there is no parents, there is nothing of interest left
576
# FIXME: It's hard to test this scenario here as this code is never
577
# called in that case. -- vila 20100322
580
for (rev_id, merge_depth, revno, end_of_merge) in rev_iter:
582
if rev_id in whitelist:
583
pmap = self.repository.get_parent_map([rev_id])
584
parents = pmap.get(rev_id, [])
585
whitelist.remove(rev_id)
586
whitelist.update(parents)
588
# We've reached the mainline, there is nothing left to
592
# A revision that is not part of the ancestry of our
595
yield (rev_id, merge_depth, revno, end_of_merge)
597
477
def leave_lock_in_place(self):
598
478
"""Tell this branch object not to release the physical lock when this
599
479
object is unlocked.
769
636
except (errors.NotStacked, errors.UnstackableBranchFormat,
770
637
errors.UnstackableRepositoryFormat):
640
# XXX: Lock correctness - should unlock our old repo if we were
642
# repositories don't offer an interface to remove fallback
643
# repositories today; take the conceptually simpler option and just
645
self.repository = self.bzrdir.find_repository()
646
self.repository.lock_write()
647
# for every revision reference the branch has, ensure it is pulled
649
source_repository = self._get_fallback_repository(old_url)
650
for revision_id in chain([self.last_revision()],
651
self.tags.get_reverse_tag_dict()):
652
self.repository.fetch(source_repository, revision_id,
774
655
self._activate_fallback_location(url)
775
656
# write this out after the repository is stacked to avoid setting a
776
657
# stacked config that doesn't work.
777
658
self._set_config_location('stacked_on_location', url)
780
"""Change a branch to be unstacked, copying data as needed.
782
Don't call this directly, use set_stacked_on_url(None).
784
pb = ui.ui_factory.nested_progress_bar()
786
pb.update("Unstacking")
787
# The basic approach here is to fetch the tip of the branch,
788
# including all available ghosts, from the existing stacked
789
# repository into a new repository object without the fallbacks.
791
# XXX: See <https://launchpad.net/bugs/397286> - this may not be
792
# correct for CHKMap repostiories
793
old_repository = self.repository
794
if len(old_repository._fallback_repositories) != 1:
795
raise AssertionError("can't cope with fallback repositories "
796
"of %r" % (self.repository,))
797
# unlock it, including unlocking the fallback
798
old_repository.unlock()
799
old_repository.lock_read()
801
# Repositories don't offer an interface to remove fallback
802
# repositories today; take the conceptually simpler option and just
803
# reopen it. We reopen it starting from the URL so that we
804
# get a separate connection for RemoteRepositories and can
805
# stream from one of them to the other. This does mean doing
806
# separate SSH connection setup, but unstacking is not a
807
# common operation so it's tolerable.
808
new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
809
new_repository = new_bzrdir.find_repository()
810
self.repository = new_repository
811
if self.repository._fallback_repositories:
812
raise AssertionError("didn't expect %r to have "
813
"fallback_repositories"
814
% (self.repository,))
815
# this is not paired with an unlock because it's just restoring
816
# the previous state; the lock's released when set_stacked_on_url
818
self.repository.lock_write()
819
# XXX: If you unstack a branch while it has a working tree
820
# with a pending merge, the pending-merged revisions will no
821
# longer be present. You can (probably) revert and remerge.
823
# XXX: This only fetches up to the tip of the repository; it
824
# doesn't bring across any tags. That's fairly consistent
825
# with how branch works, but perhaps not ideal.
826
self.repository.fetch(old_repository,
827
revision_id=self.last_revision(),
830
old_repository.unlock()
834
661
def _set_tags_bytes(self, bytes):
835
662
"""Mirror method for _get_tags_bytes.
1002
831
except ValueError:
1003
832
raise errors.NoSuchRevision(self, revision_id)
1006
834
def get_rev_id(self, revno, history=None):
1007
835
"""Find the revision id of the specified revno."""
1009
837
return _mod_revision.NULL_REVISION
1010
last_revno, last_revid = self.last_revision_info()
1011
if revno == last_revno:
1013
if revno <= 0 or revno > last_revno:
839
history = self.revision_history()
840
if revno <= 0 or revno > len(history):
1014
841
raise errors.NoSuchRevision(self, revno)
1015
distance_from_last = last_revno - revno
1016
if len(self._partial_revision_history_cache) <= distance_from_last:
1017
self._extend_partial_history(distance_from_last)
1018
return self._partial_revision_history_cache[distance_from_last]
842
return history[revno - 1]
1020
844
@needs_write_lock
1021
845
def pull(self, source, overwrite=False, stop_revision=None,
1313
1142
Callers will typically also want to check the repository.
1315
:param refs: Calculated refs for this branch as specified by
1316
branch._get_check_refs()
1317
1144
:return: A BranchCheckResult.
1319
result = BranchCheckResult(self)
1146
ret = BranchCheckResult(self)
1147
mainline_parent_id = None
1320
1148
last_revno, last_revision_id = self.last_revision_info()
1321
actual_revno = refs[('lefthand-distance', last_revision_id)]
1322
if actual_revno != last_revno:
1323
result.errors.append(errors.BzrCheckError(
1324
'revno does not match len(mainline) %s != %s' % (
1325
last_revno, actual_revno)))
1326
# TODO: We should probably also check that self.revision_history
1327
# matches the repository for older branch formats.
1328
# If looking for the code that cross-checks repository parents against
1329
# the iter_reverse_revision_history output, that is now a repository
1149
real_rev_history = []
1151
for revid in self.repository.iter_reverse_revision_history(
1153
real_rev_history.append(revid)
1154
except errors.RevisionNotPresent:
1155
ret.ghosts_in_mainline = True
1157
ret.ghosts_in_mainline = False
1158
real_rev_history.reverse()
1159
if len(real_rev_history) != last_revno:
1160
raise errors.BzrCheckError('revno does not match len(mainline)'
1161
' %s != %s' % (last_revno, len(real_rev_history)))
1162
# TODO: We should probably also check that real_rev_history actually
1163
# matches self.revision_history()
1164
for revision_id in real_rev_history:
1166
revision = self.repository.get_revision(revision_id)
1167
except errors.NoSuchRevision, e:
1168
raise errors.BzrCheckError("mainline revision {%s} not in repository"
1170
# In general the first entry on the revision history has no parents.
1171
# But it's not illegal for it to have parents listed; this can happen
1172
# in imports from Arch when the parents weren't reachable.
1173
if mainline_parent_id is not None:
1174
if mainline_parent_id not in revision.parent_ids:
1175
raise errors.BzrCheckError("previous revision {%s} not listed among "
1177
% (mainline_parent_id, revision_id))
1178
mainline_parent_id = revision_id
1333
1181
def _get_checkout_format(self):
1334
1182
"""Return the most suitable metadir for a checkout of this branch.
1565
1405
"""Return the short format description for this format."""
1566
1406
raise NotImplementedError(self.get_format_description)
1568
def _run_post_branch_init_hooks(self, a_bzrdir, name, branch):
1569
hooks = Branch.hooks['post_branch_init']
1572
params = BranchInitHookParams(self, a_bzrdir, name, branch)
1576
def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
1577
lock_type='metadir', set_format=True):
1408
def _initialize_helper(self, a_bzrdir, utf8_files, lock_type='metadir',
1578
1410
"""Initialize a branch in a bzrdir, with specified files
1580
1412
:param a_bzrdir: The bzrdir to initialize the branch in
1581
1413
:param utf8_files: The files to create as a list of
1582
1414
(filename, content) tuples
1583
:param name: Name of colocated branch to create, if any
1584
1415
:param set_format: If True, set the format with
1585
1416
self.get_format_string. (BzrBranch4 has its format set
1587
1418
:return: a branch in this format
1589
mutter('creating branch %r in %s', self, a_bzrdir.user_url)
1590
branch_transport = a_bzrdir.get_branch_transport(self, name=name)
1420
mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
1421
branch_transport = a_bzrdir.get_branch_transport(self)
1592
1423
'metadir': ('lock', lockdir.LockDir),
1593
1424
'branch4': ('branch-lock', lockable_files.TransportLock),
1776
1597
"multiple hooks installed for transform_fallback_location, "
1777
1598
"all are called with the url returned from the previous hook."
1778
1599
"The order is however undefined.", (1, 9), None))
1779
self.create_hook(HookPoint('automatic_tag_name',
1780
"Called to determine an automatic tag name for a revision."
1781
"automatic_tag_name is called with (branch, revision_id) and "
1782
"should return a tag name or None if no tag name could be "
1783
"determined. The first non-None tag name returned will be used.",
1785
self.create_hook(HookPoint('post_branch_init',
1786
"Called after new branch initialization completes. "
1787
"post_branch_init is called with a "
1788
"bzrlib.branch.BranchInitHookParams. "
1789
"Note that init, branch and checkout (both heavyweight and "
1790
"lightweight) will all trigger this hook.", (2, 2), None))
1791
self.create_hook(HookPoint('post_switch',
1792
"Called after a checkout switches branch. "
1793
"post_switch is called with a "
1794
"bzrlib.branch.SwitchHookParams.", (2, 2), None))
1798
1602
# install the default hooks into the Branch class.
1837
1641
self.old_revno, self.old_revid, self.new_revno, self.new_revid)
1840
class BranchInitHookParams(object):
1841
"""Object holding parameters passed to *_branch_init hooks.
1843
There are 4 fields that hooks may wish to access:
1845
:ivar format: the branch format
1846
:ivar bzrdir: the BzrDir where the branch will be/has been initialized
1847
:ivar name: name of colocated branch, if any (or None)
1848
:ivar branch: the branch created
1850
Note that for lightweight checkouts, the bzrdir and format fields refer to
1851
the checkout, hence they are different from the corresponding fields in
1852
branch, which refer to the original branch.
1855
def __init__(self, format, a_bzrdir, name, branch):
1856
"""Create a group of BranchInitHook parameters.
1858
:param format: the branch format
1859
:param a_bzrdir: the BzrDir where the branch will be/has been
1861
:param name: name of colocated branch, if any (or None)
1862
:param branch: the branch created
1864
Note that for lightweight checkouts, the bzrdir and format fields refer
1865
to the checkout, hence they are different from the corresponding fields
1866
in branch, which refer to the original branch.
1868
self.format = format
1869
self.bzrdir = a_bzrdir
1871
self.branch = branch
1873
def __eq__(self, other):
1874
return self.__dict__ == other.__dict__
1878
return "<%s of %s>" % (self.__class__.__name__, self.branch)
1880
return "<%s of format:%s bzrdir:%s>" % (
1881
self.__class__.__name__, self.branch,
1882
self.format, self.bzrdir)
1885
class SwitchHookParams(object):
1886
"""Object holding parameters passed to *_switch hooks.
1888
There are 4 fields that hooks may wish to access:
1890
:ivar control_dir: BzrDir of the checkout to change
1891
:ivar to_branch: branch that the checkout is to reference
1892
:ivar force: skip the check for local commits in a heavy checkout
1893
:ivar revision_id: revision ID to switch to (or None)
1896
def __init__(self, control_dir, to_branch, force, revision_id):
1897
"""Create a group of SwitchHook parameters.
1899
:param control_dir: BzrDir of the checkout to change
1900
:param to_branch: branch that the checkout is to reference
1901
:param force: skip the check for local commits in a heavy checkout
1902
:param revision_id: revision ID to switch to (or None)
1904
self.control_dir = control_dir
1905
self.to_branch = to_branch
1907
self.revision_id = revision_id
1909
def __eq__(self, other):
1910
return self.__dict__ == other.__dict__
1913
return "<%s for %s to (%s, %s)>" % (self.__class__.__name__,
1914
self.control_dir, self.to_branch,
1918
1644
class BzrBranchFormat4(BranchFormat):
1919
1645
"""Bzr branch format 4.
1973
1698
return self.get_format_string()
1975
def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
1700
def open(self, a_bzrdir, _found=False, ignore_fallbacks=False):
1976
1701
"""See BranchFormat.open()."""
1978
format = BranchFormat.find_format(a_bzrdir, name=name)
1703
format = BranchFormat.find_format(a_bzrdir)
1979
1704
if format.__class__ != self.__class__:
1980
1705
raise AssertionError("wrong format %r found for %r" %
1981
1706
(format, self))
1982
transport = a_bzrdir.get_branch_transport(None, name=name)
1708
transport = a_bzrdir.get_branch_transport(None)
1984
1709
control_files = lockable_files.LockableFiles(transport, 'lock',
1985
1710
lockdir.LockDir)
1986
1711
return self._branch_class()(_format=self,
1987
1712
_control_files=control_files,
1989
1713
a_bzrdir=a_bzrdir,
1990
1714
_repository=a_bzrdir.find_repository(),
1991
1715
ignore_fallbacks=ignore_fallbacks)
1992
1716
except errors.NoSuchFile:
1993
raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
1717
raise errors.NotBranchError(path=transport.base)
1995
1719
def __init__(self):
1996
1720
super(BranchFormatMetadir, self).__init__()
2171
1887
def get_reference(self, a_bzrdir):
2172
1888
"""See BranchFormat.get_reference()."""
2173
1889
transport = a_bzrdir.get_branch_transport(None)
2174
return transport.get_bytes('location')
1890
return transport.get('location').read()
2176
1892
def set_reference(self, a_bzrdir, to_branch):
2177
1893
"""See BranchFormat.set_reference()."""
2178
1894
transport = a_bzrdir.get_branch_transport(None)
2179
1895
location = transport.put_bytes('location', to_branch.base)
2181
def initialize(self, a_bzrdir, name=None, target_branch=None):
1897
def initialize(self, a_bzrdir, target_branch=None):
2182
1898
"""Create a branch of this format in a_bzrdir."""
2183
1899
if target_branch is None:
2184
1900
# this format does not implement branch itself, thus the implicit
2185
1901
# creation contract must see it as uninitializable
2186
1902
raise errors.UninitializableFormat(self)
2187
mutter('creating branch reference in %s', a_bzrdir.user_url)
2188
branch_transport = a_bzrdir.get_branch_transport(self, name=name)
1903
mutter('creating branch reference in %s', a_bzrdir.transport.base)
1904
branch_transport = a_bzrdir.get_branch_transport(self)
2189
1905
branch_transport.put_bytes('location',
2190
target_branch.bzrdir.user_url)
1906
target_branch.bzrdir.root_transport.base)
2191
1907
branch_transport.put_bytes('format', self.get_format_string())
2193
a_bzrdir, name, _found=True,
1909
a_bzrdir, _found=True,
2194
1910
possible_transports=[target_branch.bzrdir.root_transport])
2195
self._run_post_branch_init_hooks(a_bzrdir, name, branch)
2198
1912
def __init__(self):
2199
1913
super(BranchReferenceFormat, self).__init__()
2205
1919
def clone(to_bzrdir, revision_id=None,
2206
1920
repository_policy=None):
2207
1921
"""See Branch.clone()."""
2208
return format.initialize(to_bzrdir, target_branch=a_branch)
1922
return format.initialize(to_bzrdir, a_branch)
2209
1923
# cannot obey revision_id limits when cloning a reference ...
2210
1924
# FIXME RBC 20060210 either nuke revision_id for clone, or
2211
1925
# emit some sort of warning/error to the caller ?!
2214
def open(self, a_bzrdir, name=None, _found=False, location=None,
1928
def open(self, a_bzrdir, _found=False, location=None,
2215
1929
possible_transports=None, ignore_fallbacks=False):
2216
1930
"""Return the branch that the branch reference in a_bzrdir points at.
2218
1932
:param a_bzrdir: A BzrDir that contains a branch.
2219
:param name: Name of colocated branch to open, if any
2220
1933
:param _found: a private parameter, do not use it. It is used to
2221
1934
indicate if format probing has already be done.
2222
1935
:param ignore_fallbacks: when set, no fallback branches will be opened
2269
1981
BranchFormat.register_format(__format6)
2270
1982
BranchFormat.register_format(__format7)
2271
1983
BranchFormat.register_format(__format8)
2272
BranchFormat.set_default_format(__format7)
1984
BranchFormat.set_default_format(__format6)
2273
1985
_legacy_formats = [BzrBranchFormat4(),
2275
1987
network_format_registry.register(
2276
1988
_legacy_formats[0].network_name(), _legacy_formats[0].__class__)
2279
class BranchWriteLockResult(object):
2280
"""The result of write locking a branch.
2282
:ivar branch_token: The token obtained from the underlying branch lock, or
2284
:ivar unlock: A callable which will unlock the lock.
2287
def __init__(self, unlock, branch_token):
2288
self.branch_token = branch_token
2289
self.unlock = unlock
2292
return "BranchWriteLockResult(%s, %s)" % (self.branch_token,
2296
class BzrBranch(Branch, _RelockDebugMixin):
1991
class BzrBranch(Branch):
2297
1992
"""A branch stored in the actual filesystem.
2299
1994
Note that it's "local" in the context of the filesystem; it doesn't
2305
2000
:ivar repository: Repository for this branch.
2306
2001
:ivar base: The url of the base directory for this branch; the one
2307
2002
containing the .bzr directory.
2308
:ivar name: Optional colocated branch name as it exists in the control
2312
2005
def __init__(self, _format=None,
2313
_control_files=None, a_bzrdir=None, name=None,
2314
_repository=None, ignore_fallbacks=False):
2006
_control_files=None, a_bzrdir=None, _repository=None,
2007
ignore_fallbacks=False):
2315
2008
"""Create new branch object at a particular location."""
2316
2009
if a_bzrdir is None:
2317
2010
raise ValueError('a_bzrdir must be supplied')
2319
2012
self.bzrdir = a_bzrdir
2320
2013
self._base = self.bzrdir.transport.clone('..').base
2322
2014
# XXX: We should be able to just do
2323
2015
# self.base = self.bzrdir.root_transport.base
2324
2016
# but this does not quite work yet -- mbp 20080522
2352
2040
return self.control_files.is_locked()
2354
2042
def lock_write(self, token=None):
2355
"""Lock the branch for write operations.
2357
:param token: A token to permit reacquiring a previously held and
2359
:return: A BranchWriteLockResult.
2361
if not self.is_locked():
2362
self._note_lock('w')
2363
2043
# All-in-one needs to always unlock/lock.
2364
2044
repo_control = getattr(self.repository, 'control_files', None)
2365
2045
if self.control_files == repo_control or not self.is_locked():
2366
self.repository._warn_if_deprecated(self)
2367
2046
self.repository.lock_write()
2368
2047
took_lock = True
2370
2049
took_lock = False
2372
return BranchWriteLockResult(self.unlock,
2373
self.control_files.lock_write(token=token))
2051
return self.control_files.lock_write(token=token)
2376
2054
self.repository.unlock()
2379
2057
def lock_read(self):
2380
"""Lock the branch for read operations.
2382
:return: An object with an unlock method which will release the lock
2385
if not self.is_locked():
2386
self._note_lock('r')
2387
2058
# All-in-one needs to always unlock/lock.
2388
2059
repo_control = getattr(self.repository, 'control_files', None)
2389
2060
if self.control_files == repo_control or not self.is_locked():
2390
self.repository._warn_if_deprecated(self)
2391
2061
self.repository.lock_read()
2392
2062
took_lock = True
2394
2064
took_lock = False
2396
2066
self.control_files.lock_read()
2400
2069
self.repository.unlock()
2403
@only_raises(errors.LockNotHeld, errors.LockBroken)
2404
2072
def unlock(self):
2406
2074
self.control_files.unlock()
2774
2444
self._extend_partial_history(stop_index=last_revno-1)
2775
2445
return list(reversed(self._partial_revision_history_cache))
2447
def _extend_partial_history(self, stop_index=None, stop_revision=None):
2448
"""Extend the partial history to include a given index
2450
If a stop_index is supplied, stop when that index has been reached.
2451
If a stop_revision is supplied, stop when that revision is
2452
encountered. Otherwise, stop when the beginning of history is
2455
:param stop_index: The index which should be present. When it is
2456
present, history extension will stop.
2457
:param revision_id: The revision id which should be present. When
2458
it is encountered, history extension will stop.
2460
repo = self.repository
2461
if len(self._partial_revision_history_cache) == 0:
2462
iterator = repo.iter_reverse_revision_history(self.last_revision())
2464
start_revision = self._partial_revision_history_cache[-1]
2465
iterator = repo.iter_reverse_revision_history(start_revision)
2466
#skip the last revision in the list
2467
next_revision = iterator.next()
2468
for revision_id in iterator:
2469
self._partial_revision_history_cache.append(revision_id)
2470
if (stop_index is not None and
2471
len(self._partial_revision_history_cache) > stop_index):
2473
if revision_id == stop_revision:
2777
2476
def _write_revision_history(self, history):
2778
2477
"""Factored out of set_revision_history.