/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

Return mapping in revision_id_bzr_to_foreign() as required by the interface.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007, 2008 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
 
18
 
import sys
19
 
 
20
 
from bzrlib.lazy_import import lazy_import
21
 
lazy_import(globals(), """
22
 
from itertools import chain
23
 
from bzrlib import (
24
 
        bzrdir,
25
 
        cache_utf8,
26
 
        config as _mod_config,
27
 
        debug,
28
 
        errors,
29
 
        lockdir,
30
 
        lockable_files,
31
 
        repository,
32
 
        revision as _mod_revision,
33
 
        transport,
34
 
        tsort,
35
 
        ui,
36
 
        urlutils,
37
 
        )
38
 
from bzrlib.config import BranchConfig
39
 
from bzrlib.repofmt.pack_repo import RepositoryFormatKnitPack5RichRoot
40
 
from bzrlib.tag import (
41
 
    BasicTags,
42
 
    DisabledTags,
43
 
    )
44
 
""")
45
 
 
46
 
from bzrlib.decorators import needs_read_lock, needs_write_lock
47
 
from bzrlib.hooks import Hooks
48
 
from bzrlib.inter import InterObject
49
 
from bzrlib import registry
50
 
from bzrlib.symbol_versioning import (
51
 
    deprecated_in,
52
 
    deprecated_method,
53
 
    )
54
 
from bzrlib.trace import mutter, mutter_callsite, note, is_quiet
55
 
 
56
 
 
57
 
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
58
 
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
59
 
BZR_BRANCH_FORMAT_6 = "Bazaar Branch Format 6 (bzr 0.15)\n"
60
 
 
61
 
 
62
 
# TODO: Maybe include checks for common corruption of newlines, etc?
63
 
 
64
 
# TODO: Some operations like log might retrieve the same revisions
65
 
# repeatedly to calculate deltas.  We could perhaps have a weakref
66
 
# cache in memory to make this faster.  In general anything can be
67
 
# cached in memory between lock and unlock operations. .. nb thats
68
 
# what the transaction identity map provides
69
 
 
70
 
 
71
 
######################################################################
72
 
# branch objects
73
 
 
74
 
class Branch(object):
75
 
    """Branch holding a history of revisions.
76
 
 
77
 
    base
78
 
        Base directory/url of the branch.
79
 
 
80
 
    hooks: An instance of BranchHooks.
81
 
    """
82
 
    # this is really an instance variable - FIXME move it there
83
 
    # - RBC 20060112
84
 
    base = None
85
 
 
86
 
    def __init__(self, *ignored, **ignored_too):
87
 
        self.tags = self._format.make_tags(self)
88
 
        self._revision_history_cache = None
89
 
        self._revision_id_to_revno_cache = None
90
 
        self._partial_revision_id_to_revno_cache = {}
91
 
        self._last_revision_info_cache = None
92
 
        self._merge_sorted_revisions_cache = None
93
 
        self._open_hook()
94
 
        hooks = Branch.hooks['open']
95
 
        for hook in hooks:
96
 
            hook(self)
97
 
 
98
 
    def _open_hook(self):
99
 
        """Called by init to allow simpler extension of the base class."""
100
 
 
101
 
    def break_lock(self):
102
 
        """Break a lock if one is present from another instance.
103
 
 
104
 
        Uses the ui factory to ask for confirmation if the lock may be from
105
 
        an active process.
106
 
 
107
 
        This will probe the repository for its lock as well.
108
 
        """
109
 
        self.control_files.break_lock()
110
 
        self.repository.break_lock()
111
 
        master = self.get_master_branch()
112
 
        if master is not None:
113
 
            master.break_lock()
114
 
 
115
 
    @staticmethod
116
 
    def open(base, _unsupported=False, possible_transports=None):
117
 
        """Open the branch rooted at base.
118
 
 
119
 
        For instance, if the branch is at URL/.bzr/branch,
120
 
        Branch.open(URL) -> a Branch instance.
121
 
        """
122
 
        control = bzrdir.BzrDir.open(base, _unsupported,
123
 
                                     possible_transports=possible_transports)
124
 
        return control.open_branch(_unsupported)
125
 
 
126
 
    @staticmethod
127
 
    def open_from_transport(transport, _unsupported=False):
128
 
        """Open the branch rooted at transport"""
129
 
        control = bzrdir.BzrDir.open_from_transport(transport, _unsupported)
130
 
        return control.open_branch(_unsupported)
131
 
 
132
 
    @staticmethod
133
 
    def open_containing(url, possible_transports=None):
134
 
        """Open an existing branch which contains url.
135
 
 
136
 
        This probes for a branch at url, and searches upwards from there.
137
 
 
138
 
        Basically we keep looking up until we find the control directory or
139
 
        run into the root.  If there isn't one, raises NotBranchError.
140
 
        If there is one and it is either an unrecognised format or an unsupported
141
 
        format, UnknownFormatError or UnsupportedFormatError are raised.
142
 
        If there is one, it is returned, along with the unused portion of url.
143
 
        """
144
 
        control, relpath = bzrdir.BzrDir.open_containing(url,
145
 
                                                         possible_transports)
146
 
        return control.open_branch(), relpath
147
 
 
148
 
    def _push_should_merge_tags(self):
149
 
        """Should _basic_push merge this branch's tags into the target?
150
 
 
151
 
        The default implementation returns False if this branch has no tags,
152
 
        and True the rest of the time.  Subclasses may override this.
153
 
        """
154
 
        return self.supports_tags() and self.tags.get_tag_dict()
155
 
 
156
 
    def get_config(self):
157
 
        return BranchConfig(self)
158
 
 
159
 
    def _get_tags_bytes(self):
160
 
        """Get the bytes of a serialised tags dict.
161
 
 
162
 
        Note that not all branches support tags, nor do all use the same tags
163
 
        logic: this method is specific to BasicTags. Other tag implementations
164
 
        may use the same method name and behave differently, safely, because
165
 
        of the double-dispatch via
166
 
        format.make_tags->tags_instance->get_tags_dict.
167
 
 
168
 
        :return: The bytes of the tags file.
169
 
        :seealso: Branch._set_tags_bytes.
170
 
        """
171
 
        return self._transport.get_bytes('tags')
172
 
 
173
 
    def _get_nick(self, local=False, possible_transports=None):
174
 
        config = self.get_config()
175
 
        # explicit overrides master, but don't look for master if local is True
176
 
        if not local and not config.has_explicit_nickname():
177
 
            try:
178
 
                master = self.get_master_branch(possible_transports)
179
 
                if master is not None:
180
 
                    # return the master branch value
181
 
                    return master.nick
182
 
            except errors.BzrError, e:
183
 
                # Silently fall back to local implicit nick if the master is
184
 
                # unavailable
185
 
                mutter("Could not connect to bound branch, "
186
 
                    "falling back to local nick.\n " + str(e))
187
 
        return config.get_nickname()
188
 
 
189
 
    def _set_nick(self, nick):
190
 
        self.get_config().set_user_option('nickname', nick, warn_masked=True)
191
 
 
192
 
    nick = property(_get_nick, _set_nick)
193
 
 
194
 
    def is_locked(self):
195
 
        raise NotImplementedError(self.is_locked)
196
 
 
197
 
    def _lefthand_history(self, revision_id, last_rev=None,
198
 
                          other_branch=None):
199
 
        if 'evil' in debug.debug_flags:
200
 
            mutter_callsite(4, "_lefthand_history scales with history.")
201
 
        # stop_revision must be a descendant of last_revision
202
 
        graph = self.repository.get_graph()
203
 
        if last_rev is not None:
204
 
            if not graph.is_ancestor(last_rev, revision_id):
205
 
                # our previous tip is not merged into stop_revision
206
 
                raise errors.DivergedBranches(self, other_branch)
207
 
        # make a new revision history from the graph
208
 
        parents_map = graph.get_parent_map([revision_id])
209
 
        if revision_id not in parents_map:
210
 
            raise errors.NoSuchRevision(self, revision_id)
211
 
        current_rev_id = revision_id
212
 
        new_history = []
213
 
        check_not_reserved_id = _mod_revision.check_not_reserved_id
214
 
        # Do not include ghosts or graph origin in revision_history
215
 
        while (current_rev_id in parents_map and
216
 
               len(parents_map[current_rev_id]) > 0):
217
 
            check_not_reserved_id(current_rev_id)
218
 
            new_history.append(current_rev_id)
219
 
            current_rev_id = parents_map[current_rev_id][0]
220
 
            parents_map = graph.get_parent_map([current_rev_id])
221
 
        new_history.reverse()
222
 
        return new_history
223
 
 
224
 
    def lock_write(self):
225
 
        raise NotImplementedError(self.lock_write)
226
 
 
227
 
    def lock_read(self):
228
 
        raise NotImplementedError(self.lock_read)
229
 
 
230
 
    def unlock(self):
231
 
        raise NotImplementedError(self.unlock)
232
 
 
233
 
    def peek_lock_mode(self):
234
 
        """Return lock mode for the Branch: 'r', 'w' or None"""
235
 
        raise NotImplementedError(self.peek_lock_mode)
236
 
 
237
 
    def get_physical_lock_status(self):
238
 
        raise NotImplementedError(self.get_physical_lock_status)
239
 
 
240
 
    @needs_read_lock
241
 
    def dotted_revno_to_revision_id(self, revno, _cache_reverse=False):
242
 
        """Return the revision_id for a dotted revno.
243
 
 
244
 
        :param revno: a tuple like (1,) or (1,1,2)
245
 
        :param _cache_reverse: a private parameter enabling storage
246
 
           of the reverse mapping in a top level cache. (This should
247
 
           only be done in selective circumstances as we want to
248
 
           avoid having the mapping cached multiple times.)
249
 
        :return: the revision_id
250
 
        :raises errors.NoSuchRevision: if the revno doesn't exist
251
 
        """
252
 
        rev_id = self._do_dotted_revno_to_revision_id(revno)
253
 
        if _cache_reverse:
254
 
            self._partial_revision_id_to_revno_cache[rev_id] = revno
255
 
        return rev_id
256
 
 
257
 
    def _do_dotted_revno_to_revision_id(self, revno):
258
 
        """Worker function for dotted_revno_to_revision_id.
259
 
 
260
 
        Subclasses should override this if they wish to
261
 
        provide a more efficient implementation.
262
 
        """
263
 
        if len(revno) == 1:
264
 
            return self.get_rev_id(revno[0])
265
 
        revision_id_to_revno = self.get_revision_id_to_revno_map()
266
 
        revision_ids = [revision_id for revision_id, this_revno
267
 
                        in revision_id_to_revno.iteritems()
268
 
                        if revno == this_revno]
269
 
        if len(revision_ids) == 1:
270
 
            return revision_ids[0]
271
 
        else:
272
 
            revno_str = '.'.join(map(str, revno))
273
 
            raise errors.NoSuchRevision(self, revno_str)
274
 
 
275
 
    @needs_read_lock
276
 
    def revision_id_to_dotted_revno(self, revision_id):
277
 
        """Given a revision id, return its dotted revno.
278
 
 
279
 
        :return: a tuple like (1,) or (400,1,3).
280
 
        """
281
 
        return self._do_revision_id_to_dotted_revno(revision_id)
282
 
 
283
 
    def _do_revision_id_to_dotted_revno(self, revision_id):
284
 
        """Worker function for revision_id_to_revno."""
285
 
        # Try the caches if they are loaded
286
 
        result = self._partial_revision_id_to_revno_cache.get(revision_id)
287
 
        if result is not None:
288
 
            return result
289
 
        if self._revision_id_to_revno_cache:
290
 
            result = self._revision_id_to_revno_cache.get(revision_id)
291
 
            if result is None:
292
 
                raise errors.NoSuchRevision(self, revision_id)
293
 
        # Try the mainline as it's optimised
294
 
        try:
295
 
            revno = self.revision_id_to_revno(revision_id)
296
 
            return (revno,)
297
 
        except errors.NoSuchRevision:
298
 
            # We need to load and use the full revno map after all
299
 
            result = self.get_revision_id_to_revno_map().get(revision_id)
300
 
            if result is None:
301
 
                raise errors.NoSuchRevision(self, revision_id)
302
 
        return result
303
 
 
304
 
    @needs_read_lock
305
 
    def get_revision_id_to_revno_map(self):
306
 
        """Return the revision_id => dotted revno map.
307
 
 
308
 
        This will be regenerated on demand, but will be cached.
309
 
 
310
 
        :return: A dictionary mapping revision_id => dotted revno.
311
 
            This dictionary should not be modified by the caller.
312
 
        """
313
 
        if self._revision_id_to_revno_cache is not None:
314
 
            mapping = self._revision_id_to_revno_cache
315
 
        else:
316
 
            mapping = self._gen_revno_map()
317
 
            self._cache_revision_id_to_revno(mapping)
318
 
        # TODO: jam 20070417 Since this is being cached, should we be returning
319
 
        #       a copy?
320
 
        # I would rather not, and instead just declare that users should not
321
 
        # modify the return value.
322
 
        return mapping
323
 
 
324
 
    def _gen_revno_map(self):
325
 
        """Create a new mapping from revision ids to dotted revnos.
326
 
 
327
 
        Dotted revnos are generated based on the current tip in the revision
328
 
        history.
329
 
        This is the worker function for get_revision_id_to_revno_map, which
330
 
        just caches the return value.
331
 
 
332
 
        :return: A dictionary mapping revision_id => dotted revno.
333
 
        """
334
 
        revision_id_to_revno = dict((rev_id, revno)
335
 
            for rev_id, depth, revno, end_of_merge
336
 
             in self.iter_merge_sorted_revisions())
337
 
        return revision_id_to_revno
338
 
 
339
 
    @needs_read_lock
340
 
    def iter_merge_sorted_revisions(self, start_revision_id=None,
341
 
            stop_revision_id=None, stop_rule='exclude', direction='reverse'):
342
 
        """Walk the revisions for a branch in merge sorted order.
343
 
 
344
 
        Merge sorted order is the output from a merge-aware,
345
 
        topological sort, i.e. all parents come before their
346
 
        children going forward; the opposite for reverse.
347
 
 
348
 
        :param start_revision_id: the revision_id to begin walking from.
349
 
            If None, the branch tip is used.
350
 
        :param stop_revision_id: the revision_id to terminate the walk
351
 
            after. If None, the rest of history is included.
352
 
        :param stop_rule: if stop_revision_id is not None, the precise rule
353
 
            to use for termination:
354
 
            * 'exclude' - leave the stop revision out of the result (default)
355
 
            * 'include' - the stop revision is the last item in the result
356
 
            * 'with-merges' - include the stop revision and all of its
357
 
              merged revisions in the result
358
 
        :param direction: either 'reverse' or 'forward':
359
 
            * reverse means return the start_revision_id first, i.e.
360
 
              start at the most recent revision and go backwards in history
361
 
            * forward returns tuples in the opposite order to reverse.
362
 
              Note in particular that forward does *not* do any intelligent
363
 
              ordering w.r.t. depth as some clients of this API may like.
364
 
              (If required, that ought to be done at higher layers.)
365
 
 
366
 
        :return: an iterator over (revision_id, depth, revno, end_of_merge)
367
 
            tuples where:
368
 
 
369
 
            * revision_id: the unique id of the revision
370
 
            * depth: How many levels of merging deep this node has been
371
 
              found.
372
 
            * revno_sequence: This field provides a sequence of
373
 
              revision numbers for all revisions. The format is:
374
 
              (REVNO, BRANCHNUM, BRANCHREVNO). BRANCHNUM is the number of the
375
 
              branch that the revno is on. From left to right the REVNO numbers
376
 
              are the sequence numbers within that branch of the revision.
377
 
            * end_of_merge: When True the next node (earlier in history) is
378
 
              part of a different merge.
379
 
        """
380
 
        # Note: depth and revno values are in the context of the branch so
381
 
        # we need the full graph to get stable numbers, regardless of the
382
 
        # start_revision_id.
383
 
        if self._merge_sorted_revisions_cache is None:
384
 
            last_revision = self.last_revision()
385
 
            graph = self.repository.get_graph()
386
 
            parent_map = dict(((key, value) for key, value in
387
 
                     graph.iter_ancestry([last_revision]) if value is not None))
388
 
            revision_graph = repository._strip_NULL_ghosts(parent_map)
389
 
            revs = tsort.merge_sort(revision_graph, last_revision, None,
390
 
                generate_revno=True)
391
 
            # Drop the sequence # before caching
392
 
            self._merge_sorted_revisions_cache = [r[1:] for r in revs]
393
 
 
394
 
        filtered = self._filter_merge_sorted_revisions(
395
 
            self._merge_sorted_revisions_cache, start_revision_id,
396
 
            stop_revision_id, stop_rule)
397
 
        if direction == 'reverse':
398
 
            return filtered
399
 
        if direction == 'forward':
400
 
            return reversed(list(filtered))
401
 
        else:
402
 
            raise ValueError('invalid direction %r' % direction)
403
 
 
404
 
    def _filter_merge_sorted_revisions(self, merge_sorted_revisions,
405
 
        start_revision_id, stop_revision_id, stop_rule):
406
 
        """Iterate over an inclusive range of sorted revisions."""
407
 
        rev_iter = iter(merge_sorted_revisions)
408
 
        if start_revision_id is not None:
409
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
410
 
                if rev_id != start_revision_id:
411
 
                    continue
412
 
                else:
413
 
                    # The decision to include the start or not
414
 
                    # depends on the stop_rule if a stop is provided
415
 
                    rev_iter = chain(
416
 
                        iter([(rev_id, depth, revno, end_of_merge)]),
417
 
                        rev_iter)
418
 
                    break
419
 
        if stop_revision_id is None:
420
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
421
 
                yield rev_id, depth, revno, end_of_merge
422
 
        elif stop_rule == 'exclude':
423
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
424
 
                if rev_id == stop_revision_id:
425
 
                    return
426
 
                yield rev_id, depth, revno, end_of_merge
427
 
        elif stop_rule == 'include':
428
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
429
 
                yield rev_id, depth, revno, end_of_merge
430
 
                if rev_id == stop_revision_id:
431
 
                    return
432
 
        elif stop_rule == 'with-merges':
433
 
            stop_rev = self.repository.get_revision(stop_revision_id)
434
 
            if stop_rev.parent_ids:
435
 
                left_parent = stop_rev.parent_ids[0]
436
 
            else:
437
 
                left_parent = _mod_revision.NULL_REVISION
438
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
439
 
                if rev_id == left_parent:
440
 
                    return
441
 
                yield rev_id, depth, revno, end_of_merge
442
 
        else:
443
 
            raise ValueError('invalid stop_rule %r' % stop_rule)
444
 
 
445
 
    def leave_lock_in_place(self):
446
 
        """Tell this branch object not to release the physical lock when this
447
 
        object is unlocked.
448
 
 
449
 
        If lock_write doesn't return a token, then this method is not supported.
450
 
        """
451
 
        self.control_files.leave_in_place()
452
 
 
453
 
    def dont_leave_lock_in_place(self):
454
 
        """Tell this branch object to release the physical lock when this
455
 
        object is unlocked, even if it didn't originally acquire it.
456
 
 
457
 
        If lock_write doesn't return a token, then this method is not supported.
458
 
        """
459
 
        self.control_files.dont_leave_in_place()
460
 
 
461
 
    def bind(self, other):
462
 
        """Bind the local branch the other branch.
463
 
 
464
 
        :param other: The branch to bind to
465
 
        :type other: Branch
466
 
        """
467
 
        raise errors.UpgradeRequired(self.base)
468
 
 
469
 
    @needs_write_lock
470
 
    def fetch(self, from_branch, last_revision=None, pb=None):
471
 
        """Copy revisions from from_branch into this branch.
472
 
 
473
 
        :param from_branch: Where to copy from.
474
 
        :param last_revision: What revision to stop at (None for at the end
475
 
                              of the branch.
476
 
        :param pb: An optional progress bar to use.
477
 
        :return: None
478
 
        """
479
 
        if self.base == from_branch.base:
480
 
            return (0, [])
481
 
        if pb is None:
482
 
            nested_pb = ui.ui_factory.nested_progress_bar()
483
 
            pb = nested_pb
484
 
        else:
485
 
            nested_pb = None
486
 
 
487
 
        from_branch.lock_read()
488
 
        try:
489
 
            if last_revision is None:
490
 
                pb.update('get source history')
491
 
                last_revision = from_branch.last_revision()
492
 
                last_revision = _mod_revision.ensure_null(last_revision)
493
 
            return self.repository.fetch(from_branch.repository,
494
 
                                         revision_id=last_revision,
495
 
                                         pb=nested_pb)
496
 
        finally:
497
 
            if nested_pb is not None:
498
 
                nested_pb.finished()
499
 
            from_branch.unlock()
500
 
 
501
 
    def get_bound_location(self):
502
 
        """Return the URL of the branch we are bound to.
503
 
 
504
 
        Older format branches cannot bind, please be sure to use a metadir
505
 
        branch.
506
 
        """
507
 
        return None
508
 
 
509
 
    def get_old_bound_location(self):
510
 
        """Return the URL of the branch we used to be bound to
511
 
        """
512
 
        raise errors.UpgradeRequired(self.base)
513
 
 
514
 
    def get_commit_builder(self, parents, config=None, timestamp=None,
515
 
                           timezone=None, committer=None, revprops=None,
516
 
                           revision_id=None):
517
 
        """Obtain a CommitBuilder for this branch.
518
 
 
519
 
        :param parents: Revision ids of the parents of the new revision.
520
 
        :param config: Optional configuration to use.
521
 
        :param timestamp: Optional timestamp recorded for commit.
522
 
        :param timezone: Optional timezone for timestamp.
523
 
        :param committer: Optional committer to set for commit.
524
 
        :param revprops: Optional dictionary of revision properties.
525
 
        :param revision_id: Optional revision id.
526
 
        """
527
 
 
528
 
        if config is None:
529
 
            config = self.get_config()
530
 
 
531
 
        return self.repository.get_commit_builder(self, parents, config,
532
 
            timestamp, timezone, committer, revprops, revision_id)
533
 
 
534
 
    def get_master_branch(self, possible_transports=None):
535
 
        """Return the branch we are bound to.
536
 
 
537
 
        :return: Either a Branch, or None
538
 
        """
539
 
        return None
540
 
 
541
 
    def get_revision_delta(self, revno):
542
 
        """Return the delta for one revision.
543
 
 
544
 
        The delta is relative to its mainline predecessor, or the
545
 
        empty tree for revision 1.
546
 
        """
547
 
        rh = self.revision_history()
548
 
        if not (1 <= revno <= len(rh)):
549
 
            raise errors.InvalidRevisionNumber(revno)
550
 
        return self.repository.get_revision_delta(rh[revno-1])
551
 
 
552
 
    def get_stacked_on_url(self):
553
 
        """Get the URL this branch is stacked against.
554
 
 
555
 
        :raises NotStacked: If the branch is not stacked.
556
 
        :raises UnstackableBranchFormat: If the branch does not support
557
 
            stacking.
558
 
        """
559
 
        raise NotImplementedError(self.get_stacked_on_url)
560
 
 
561
 
    def print_file(self, file, revision_id):
562
 
        """Print `file` to stdout."""
563
 
        raise NotImplementedError(self.print_file)
564
 
 
565
 
    def set_revision_history(self, rev_history):
566
 
        raise NotImplementedError(self.set_revision_history)
567
 
 
568
 
    def set_stacked_on_url(self, url):
569
 
        """Set the URL this branch is stacked against.
570
 
 
571
 
        :raises UnstackableBranchFormat: If the branch does not support
572
 
            stacking.
573
 
        :raises UnstackableRepositoryFormat: If the repository does not support
574
 
            stacking.
575
 
        """
576
 
        raise NotImplementedError(self.set_stacked_on_url)
577
 
 
578
 
    def _set_tags_bytes(self, bytes):
579
 
        """Mirror method for _get_tags_bytes.
580
 
 
581
 
        :seealso: Branch._get_tags_bytes.
582
 
        """
583
 
        return _run_with_write_locked_target(self, self._transport.put_bytes,
584
 
            'tags', bytes)
585
 
 
586
 
    def _cache_revision_history(self, rev_history):
587
 
        """Set the cached revision history to rev_history.
588
 
 
589
 
        The revision_history method will use this cache to avoid regenerating
590
 
        the revision history.
591
 
 
592
 
        This API is semi-public; it only for use by subclasses, all other code
593
 
        should consider it to be private.
594
 
        """
595
 
        self._revision_history_cache = rev_history
596
 
 
597
 
    def _cache_revision_id_to_revno(self, revision_id_to_revno):
598
 
        """Set the cached revision_id => revno map to revision_id_to_revno.
599
 
 
600
 
        This API is semi-public; it only for use by subclasses, all other code
601
 
        should consider it to be private.
602
 
        """
603
 
        self._revision_id_to_revno_cache = revision_id_to_revno
604
 
 
605
 
    def _clear_cached_state(self):
606
 
        """Clear any cached data on this branch, e.g. cached revision history.
607
 
 
608
 
        This means the next call to revision_history will need to call
609
 
        _gen_revision_history.
610
 
 
611
 
        This API is semi-public; it only for use by subclasses, all other code
612
 
        should consider it to be private.
613
 
        """
614
 
        self._revision_history_cache = None
615
 
        self._revision_id_to_revno_cache = None
616
 
        self._last_revision_info_cache = None
617
 
        self._merge_sorted_revisions_cache = None
618
 
 
619
 
    def _gen_revision_history(self):
620
 
        """Return sequence of revision hashes on to this branch.
621
 
 
622
 
        Unlike revision_history, this method always regenerates or rereads the
623
 
        revision history, i.e. it does not cache the result, so repeated calls
624
 
        may be expensive.
625
 
 
626
 
        Concrete subclasses should override this instead of revision_history so
627
 
        that subclasses do not need to deal with caching logic.
628
 
 
629
 
        This API is semi-public; it only for use by subclasses, all other code
630
 
        should consider it to be private.
631
 
        """
632
 
        raise NotImplementedError(self._gen_revision_history)
633
 
 
634
 
    @needs_read_lock
635
 
    def revision_history(self):
636
 
        """Return sequence of revision ids on this branch.
637
 
 
638
 
        This method will cache the revision history for as long as it is safe to
639
 
        do so.
640
 
        """
641
 
        if 'evil' in debug.debug_flags:
642
 
            mutter_callsite(3, "revision_history scales with history.")
643
 
        if self._revision_history_cache is not None:
644
 
            history = self._revision_history_cache
645
 
        else:
646
 
            history = self._gen_revision_history()
647
 
            self._cache_revision_history(history)
648
 
        return list(history)
649
 
 
650
 
    def revno(self):
651
 
        """Return current revision number for this branch.
652
 
 
653
 
        That is equivalent to the number of revisions committed to
654
 
        this branch.
655
 
        """
656
 
        return self.last_revision_info()[0]
657
 
 
658
 
    def unbind(self):
659
 
        """Older format branches cannot bind or unbind."""
660
 
        raise errors.UpgradeRequired(self.base)
661
 
 
662
 
    def set_append_revisions_only(self, enabled):
663
 
        """Older format branches are never restricted to append-only"""
664
 
        raise errors.UpgradeRequired(self.base)
665
 
 
666
 
    def last_revision(self):
667
 
        """Return last revision id, or NULL_REVISION."""
668
 
        return self.last_revision_info()[1]
669
 
 
670
 
    @needs_read_lock
671
 
    def last_revision_info(self):
672
 
        """Return information about the last revision.
673
 
 
674
 
        :return: A tuple (revno, revision_id).
675
 
        """
676
 
        if self._last_revision_info_cache is None:
677
 
            self._last_revision_info_cache = self._last_revision_info()
678
 
        return self._last_revision_info_cache
679
 
 
680
 
    def _last_revision_info(self):
681
 
        rh = self.revision_history()
682
 
        revno = len(rh)
683
 
        if revno:
684
 
            return (revno, rh[-1])
685
 
        else:
686
 
            return (0, _mod_revision.NULL_REVISION)
687
 
 
688
 
    @deprecated_method(deprecated_in((1, 6, 0)))
689
 
    def missing_revisions(self, other, stop_revision=None):
690
 
        """Return a list of new revisions that would perfectly fit.
691
 
 
692
 
        If self and other have not diverged, return a list of the revisions
693
 
        present in other, but missing from self.
694
 
        """
695
 
        self_history = self.revision_history()
696
 
        self_len = len(self_history)
697
 
        other_history = other.revision_history()
698
 
        other_len = len(other_history)
699
 
        common_index = min(self_len, other_len) -1
700
 
        if common_index >= 0 and \
701
 
            self_history[common_index] != other_history[common_index]:
702
 
            raise errors.DivergedBranches(self, other)
703
 
 
704
 
        if stop_revision is None:
705
 
            stop_revision = other_len
706
 
        else:
707
 
            if stop_revision > other_len:
708
 
                raise errors.NoSuchRevision(self, stop_revision)
709
 
        return other_history[self_len:stop_revision]
710
 
 
711
 
    @needs_write_lock
712
 
    def update_revisions(self, other, stop_revision=None, overwrite=False,
713
 
                         graph=None):
714
 
        """Pull in new perfect-fit revisions.
715
 
 
716
 
        :param other: Another Branch to pull from
717
 
        :param stop_revision: Updated until the given revision
718
 
        :param overwrite: Always set the branch pointer, rather than checking
719
 
            to see if it is a proper descendant.
720
 
        :param graph: A Graph object that can be used to query history
721
 
            information. This can be None.
722
 
        :return: None
723
 
        """
724
 
        return InterBranch.get(other, self).update_revisions(stop_revision,
725
 
            overwrite, graph)
726
 
 
727
 
    def import_last_revision_info(self, source_repo, revno, revid):
728
 
        """Set the last revision info, importing from another repo if necessary.
729
 
 
730
 
        This is used by the bound branch code to upload a revision to
731
 
        the master branch first before updating the tip of the local branch.
732
 
 
733
 
        :param source_repo: Source repository to optionally fetch from
734
 
        :param revno: Revision number of the new tip
735
 
        :param revid: Revision id of the new tip
736
 
        """
737
 
        if not self.repository.has_same_location(source_repo):
738
 
            self.repository.fetch(source_repo, revision_id=revid)
739
 
        self.set_last_revision_info(revno, revid)
740
 
 
741
 
    def revision_id_to_revno(self, revision_id):
742
 
        """Given a revision id, return its revno"""
743
 
        if _mod_revision.is_null(revision_id):
744
 
            return 0
745
 
        history = self.revision_history()
746
 
        try:
747
 
            return history.index(revision_id) + 1
748
 
        except ValueError:
749
 
            raise errors.NoSuchRevision(self, revision_id)
750
 
 
751
 
    def get_rev_id(self, revno, history=None):
752
 
        """Find the revision id of the specified revno."""
753
 
        if revno == 0:
754
 
            return _mod_revision.NULL_REVISION
755
 
        if history is None:
756
 
            history = self.revision_history()
757
 
        if revno <= 0 or revno > len(history):
758
 
            raise errors.NoSuchRevision(self, revno)
759
 
        return history[revno - 1]
760
 
 
761
 
    def pull(self, source, overwrite=False, stop_revision=None,
762
 
             possible_transports=None, _override_hook_target=None):
763
 
        """Mirror source into this branch.
764
 
 
765
 
        This branch is considered to be 'local', having low latency.
766
 
 
767
 
        :returns: PullResult instance
768
 
        """
769
 
        raise NotImplementedError(self.pull)
770
 
 
771
 
    def push(self, target, overwrite=False, stop_revision=None):
772
 
        """Mirror this branch into target.
773
 
 
774
 
        This branch is considered to be 'local', having low latency.
775
 
        """
776
 
        raise NotImplementedError(self.push)
777
 
 
778
 
    def basis_tree(self):
779
 
        """Return `Tree` object for last revision."""
780
 
        return self.repository.revision_tree(self.last_revision())
781
 
 
782
 
    def get_parent(self):
783
 
        """Return the parent location of the branch.
784
 
 
785
 
        This is the default location for pull/missing.  The usual
786
 
        pattern is that the user can override it by specifying a
787
 
        location.
788
 
        """
789
 
        parent = self._get_parent_location()
790
 
        if parent is None:
791
 
            return parent
792
 
        # This is an old-format absolute path to a local branch
793
 
        # turn it into a url
794
 
        if parent.startswith('/'):
795
 
            parent = urlutils.local_path_to_url(parent.decode('utf8'))
796
 
        try:
797
 
            return urlutils.join(self.base[:-1], parent)
798
 
        except errors.InvalidURLJoin, e:
799
 
            raise errors.InaccessibleParent(parent, self.base)
800
 
 
801
 
    def _get_parent_location(self):
802
 
        raise NotImplementedError(self._get_parent_location)
803
 
 
804
 
    def _set_config_location(self, name, url, config=None,
805
 
                             make_relative=False):
806
 
        if config is None:
807
 
            config = self.get_config()
808
 
        if url is None:
809
 
            url = ''
810
 
        elif make_relative:
811
 
            url = urlutils.relative_url(self.base, url)
812
 
        config.set_user_option(name, url, warn_masked=True)
813
 
 
814
 
    def _get_config_location(self, name, config=None):
815
 
        if config is None:
816
 
            config = self.get_config()
817
 
        location = config.get_user_option(name)
818
 
        if location == '':
819
 
            location = None
820
 
        return location
821
 
 
822
 
    def get_submit_branch(self):
823
 
        """Return the submit location of the branch.
824
 
 
825
 
        This is the default location for bundle.  The usual
826
 
        pattern is that the user can override it by specifying a
827
 
        location.
828
 
        """
829
 
        return self.get_config().get_user_option('submit_branch')
830
 
 
831
 
    def set_submit_branch(self, location):
832
 
        """Return the submit location of the branch.
833
 
 
834
 
        This is the default location for bundle.  The usual
835
 
        pattern is that the user can override it by specifying a
836
 
        location.
837
 
        """
838
 
        self.get_config().set_user_option('submit_branch', location,
839
 
            warn_masked=True)
840
 
 
841
 
    def get_public_branch(self):
842
 
        """Return the public location of the branch.
843
 
 
844
 
        This is is used by merge directives.
845
 
        """
846
 
        return self._get_config_location('public_branch')
847
 
 
848
 
    def set_public_branch(self, location):
849
 
        """Return the submit location of the branch.
850
 
 
851
 
        This is the default location for bundle.  The usual
852
 
        pattern is that the user can override it by specifying a
853
 
        location.
854
 
        """
855
 
        self._set_config_location('public_branch', location)
856
 
 
857
 
    def get_push_location(self):
858
 
        """Return the None or the location to push this branch to."""
859
 
        push_loc = self.get_config().get_user_option('push_location')
860
 
        return push_loc
861
 
 
862
 
    def set_push_location(self, location):
863
 
        """Set a new push location for this branch."""
864
 
        raise NotImplementedError(self.set_push_location)
865
 
 
866
 
    def _run_post_change_branch_tip_hooks(self, old_revno, old_revid):
867
 
        """Run the post_change_branch_tip hooks."""
868
 
        hooks = Branch.hooks['post_change_branch_tip']
869
 
        if not hooks:
870
 
            return
871
 
        new_revno, new_revid = self.last_revision_info()
872
 
        params = ChangeBranchTipParams(
873
 
            self, old_revno, new_revno, old_revid, new_revid)
874
 
        for hook in hooks:
875
 
            hook(params)
876
 
 
877
 
    def _run_pre_change_branch_tip_hooks(self, new_revno, new_revid):
878
 
        """Run the pre_change_branch_tip hooks."""
879
 
        hooks = Branch.hooks['pre_change_branch_tip']
880
 
        if not hooks:
881
 
            return
882
 
        old_revno, old_revid = self.last_revision_info()
883
 
        params = ChangeBranchTipParams(
884
 
            self, old_revno, new_revno, old_revid, new_revid)
885
 
        for hook in hooks:
886
 
            try:
887
 
                hook(params)
888
 
            except errors.TipChangeRejected:
889
 
                raise
890
 
            except Exception:
891
 
                exc_info = sys.exc_info()
892
 
                hook_name = Branch.hooks.get_hook_name(hook)
893
 
                raise errors.HookFailed(
894
 
                    'pre_change_branch_tip', hook_name, exc_info)
895
 
 
896
 
    def set_parent(self, url):
897
 
        raise NotImplementedError(self.set_parent)
898
 
 
899
 
    @needs_write_lock
900
 
    def update(self):
901
 
        """Synchronise this branch with the master branch if any.
902
 
 
903
 
        :return: None or the last_revision pivoted out during the update.
904
 
        """
905
 
        return None
906
 
 
907
 
    def check_revno(self, revno):
908
 
        """\
909
 
        Check whether a revno corresponds to any revision.
910
 
        Zero (the NULL revision) is considered valid.
911
 
        """
912
 
        if revno != 0:
913
 
            self.check_real_revno(revno)
914
 
 
915
 
    def check_real_revno(self, revno):
916
 
        """\
917
 
        Check whether a revno corresponds to a real revision.
918
 
        Zero (the NULL revision) is considered invalid
919
 
        """
920
 
        if revno < 1 or revno > self.revno():
921
 
            raise errors.InvalidRevisionNumber(revno)
922
 
 
923
 
    @needs_read_lock
924
 
    def clone(self, to_bzrdir, revision_id=None, repository_policy=None):
925
 
        """Clone this branch into to_bzrdir preserving all semantic values.
926
 
 
927
 
        Most API users will want 'create_clone_on_transport', which creates a
928
 
        new bzrdir and branch on the fly.
929
 
 
930
 
        revision_id: if not None, the revision history in the new branch will
931
 
                     be truncated to end with revision_id.
932
 
        """
933
 
        result = to_bzrdir.create_branch()
934
 
        if repository_policy is not None:
935
 
            repository_policy.configure_branch(result)
936
 
        self.copy_content_into(result, revision_id=revision_id)
937
 
        return  result
938
 
 
939
 
    @needs_read_lock
940
 
    def sprout(self, to_bzrdir, revision_id=None, repository_policy=None):
941
 
        """Create a new line of development from the branch, into to_bzrdir.
942
 
 
943
 
        to_bzrdir controls the branch format.
944
 
 
945
 
        revision_id: if not None, the revision history in the new branch will
946
 
                     be truncated to end with revision_id.
947
 
        """
948
 
        result = to_bzrdir.create_branch()
949
 
        if repository_policy is not None:
950
 
            repository_policy.configure_branch(result)
951
 
        self.copy_content_into(result, revision_id=revision_id)
952
 
        result.set_parent(self.bzrdir.root_transport.base)
953
 
        return result
954
 
 
955
 
    def _synchronize_history(self, destination, revision_id):
956
 
        """Synchronize last revision and revision history between branches.
957
 
 
958
 
        This version is most efficient when the destination is also a
959
 
        BzrBranch6, but works for BzrBranch5, as long as the destination's
960
 
        repository contains all the lefthand ancestors of the intended
961
 
        last_revision.  If not, set_last_revision_info will fail.
962
 
 
963
 
        :param destination: The branch to copy the history into
964
 
        :param revision_id: The revision-id to truncate history at.  May
965
 
          be None to copy complete history.
966
 
        """
967
 
        source_revno, source_revision_id = self.last_revision_info()
968
 
        if revision_id is None:
969
 
            revno, revision_id = source_revno, source_revision_id
970
 
        elif source_revision_id == revision_id:
971
 
            # we know the revno without needing to walk all of history
972
 
            revno = source_revno
973
 
        else:
974
 
            # To figure out the revno for a random revision, we need to build
975
 
            # the revision history, and count its length.
976
 
            # We don't care about the order, just how long it is.
977
 
            # Alternatively, we could start at the current location, and count
978
 
            # backwards. But there is no guarantee that we will find it since
979
 
            # it may be a merged revision.
980
 
            revno = len(list(self.repository.iter_reverse_revision_history(
981
 
                                                                revision_id)))
982
 
        destination.set_last_revision_info(revno, revision_id)
983
 
 
984
 
    @needs_read_lock
985
 
    def copy_content_into(self, destination, revision_id=None):
986
 
        """Copy the content of self into destination.
987
 
 
988
 
        revision_id: if not None, the revision history in the new branch will
989
 
                     be truncated to end with revision_id.
990
 
        """
991
 
        self._synchronize_history(destination, revision_id)
992
 
        try:
993
 
            parent = self.get_parent()
994
 
        except errors.InaccessibleParent, e:
995
 
            mutter('parent was not accessible to copy: %s', e)
996
 
        else:
997
 
            if parent:
998
 
                destination.set_parent(parent)
999
 
        if self._push_should_merge_tags():
1000
 
            self.tags.merge_to(destination.tags)
1001
 
 
1002
 
    @needs_read_lock
1003
 
    def check(self):
1004
 
        """Check consistency of the branch.
1005
 
 
1006
 
        In particular this checks that revisions given in the revision-history
1007
 
        do actually match up in the revision graph, and that they're all
1008
 
        present in the repository.
1009
 
 
1010
 
        Callers will typically also want to check the repository.
1011
 
 
1012
 
        :return: A BranchCheckResult.
1013
 
        """
1014
 
        mainline_parent_id = None
1015
 
        last_revno, last_revision_id = self.last_revision_info()
1016
 
        real_rev_history = list(self.repository.iter_reverse_revision_history(
1017
 
                                last_revision_id))
1018
 
        real_rev_history.reverse()
1019
 
        if len(real_rev_history) != last_revno:
1020
 
            raise errors.BzrCheckError('revno does not match len(mainline)'
1021
 
                ' %s != %s' % (last_revno, len(real_rev_history)))
1022
 
        # TODO: We should probably also check that real_rev_history actually
1023
 
        #       matches self.revision_history()
1024
 
        for revision_id in real_rev_history:
1025
 
            try:
1026
 
                revision = self.repository.get_revision(revision_id)
1027
 
            except errors.NoSuchRevision, e:
1028
 
                raise errors.BzrCheckError("mainline revision {%s} not in repository"
1029
 
                            % revision_id)
1030
 
            # In general the first entry on the revision history has no parents.
1031
 
            # But it's not illegal for it to have parents listed; this can happen
1032
 
            # in imports from Arch when the parents weren't reachable.
1033
 
            if mainline_parent_id is not None:
1034
 
                if mainline_parent_id not in revision.parent_ids:
1035
 
                    raise errors.BzrCheckError("previous revision {%s} not listed among "
1036
 
                                        "parents of {%s}"
1037
 
                                        % (mainline_parent_id, revision_id))
1038
 
            mainline_parent_id = revision_id
1039
 
        return BranchCheckResult(self)
1040
 
 
1041
 
    def _get_checkout_format(self):
1042
 
        """Return the most suitable metadir for a checkout of this branch.
1043
 
        Weaves are used if this branch's repository uses weaves.
1044
 
        """
1045
 
        if isinstance(self.bzrdir, bzrdir.BzrDirPreSplitOut):
1046
 
            from bzrlib.repofmt import weaverepo
1047
 
            format = bzrdir.BzrDirMetaFormat1()
1048
 
            format.repository_format = weaverepo.RepositoryFormat7()
1049
 
        else:
1050
 
            format = self.repository.bzrdir.checkout_metadir()
1051
 
            format.set_branch_format(self._format)
1052
 
        return format
1053
 
 
1054
 
    def create_clone_on_transport(self, to_transport, revision_id=None,
1055
 
        stacked_on=None):
1056
 
        """Create a clone of this branch and its bzrdir.
1057
 
 
1058
 
        :param to_transport: The transport to clone onto.
1059
 
        :param revision_id: The revision id to use as tip in the new branch.
1060
 
            If None the tip is obtained from this branch.
1061
 
        :param stacked_on: An optional URL to stack the clone on.
1062
 
        """
1063
 
        # XXX: Fix the bzrdir API to allow getting the branch back from the
1064
 
        # clone call. Or something. 20090224 RBC/spiv.
1065
 
        dir_to = self.bzrdir.clone_on_transport(to_transport,
1066
 
            revision_id=revision_id, stacked_on=stacked_on)
1067
 
        return dir_to.open_branch()
1068
 
 
1069
 
    def create_checkout(self, to_location, revision_id=None,
1070
 
                        lightweight=False, accelerator_tree=None,
1071
 
                        hardlink=False):
1072
 
        """Create a checkout of a branch.
1073
 
 
1074
 
        :param to_location: The url to produce the checkout at
1075
 
        :param revision_id: The revision to check out
1076
 
        :param lightweight: If True, produce a lightweight checkout, otherwise,
1077
 
        produce a bound branch (heavyweight checkout)
1078
 
        :param accelerator_tree: A tree which can be used for retrieving file
1079
 
            contents more quickly than the revision tree, i.e. a workingtree.
1080
 
            The revision tree will be used for cases where accelerator_tree's
1081
 
            content is different.
1082
 
        :param hardlink: If true, hard-link files from accelerator_tree,
1083
 
            where possible.
1084
 
        :return: The tree of the created checkout
1085
 
        """
1086
 
        t = transport.get_transport(to_location)
1087
 
        t.ensure_base()
1088
 
        if lightweight:
1089
 
            format = self._get_checkout_format()
1090
 
            checkout = format.initialize_on_transport(t)
1091
 
            from_branch = BranchReferenceFormat().initialize(checkout, self)
1092
 
        else:
1093
 
            format = self._get_checkout_format()
1094
 
            checkout_branch = bzrdir.BzrDir.create_branch_convenience(
1095
 
                to_location, force_new_tree=False, format=format)
1096
 
            checkout = checkout_branch.bzrdir
1097
 
            checkout_branch.bind(self)
1098
 
            # pull up to the specified revision_id to set the initial
1099
 
            # branch tip correctly, and seed it with history.
1100
 
            checkout_branch.pull(self, stop_revision=revision_id)
1101
 
            from_branch=None
1102
 
        tree = checkout.create_workingtree(revision_id,
1103
 
                                           from_branch=from_branch,
1104
 
                                           accelerator_tree=accelerator_tree,
1105
 
                                           hardlink=hardlink)
1106
 
        basis_tree = tree.basis_tree()
1107
 
        basis_tree.lock_read()
1108
 
        try:
1109
 
            for path, file_id in basis_tree.iter_references():
1110
 
                reference_parent = self.reference_parent(file_id, path)
1111
 
                reference_parent.create_checkout(tree.abspath(path),
1112
 
                    basis_tree.get_reference_revision(file_id, path),
1113
 
                    lightweight)
1114
 
        finally:
1115
 
            basis_tree.unlock()
1116
 
        return tree
1117
 
 
1118
 
    @needs_write_lock
1119
 
    def reconcile(self, thorough=True):
1120
 
        """Make sure the data stored in this branch is consistent."""
1121
 
        from bzrlib.reconcile import BranchReconciler
1122
 
        reconciler = BranchReconciler(self, thorough=thorough)
1123
 
        reconciler.reconcile()
1124
 
        return reconciler
1125
 
 
1126
 
    def reference_parent(self, file_id, path):
1127
 
        """Return the parent branch for a tree-reference file_id
1128
 
        :param file_id: The file_id of the tree reference
1129
 
        :param path: The path of the file_id in the tree
1130
 
        :return: A branch associated with the file_id
1131
 
        """
1132
 
        # FIXME should provide multiple branches, based on config
1133
 
        return Branch.open(self.bzrdir.root_transport.clone(path).base)
1134
 
 
1135
 
    def supports_tags(self):
1136
 
        return self._format.supports_tags()
1137
 
 
1138
 
    def _check_if_descendant_or_diverged(self, revision_a, revision_b, graph,
1139
 
                                         other_branch):
1140
 
        """Ensure that revision_b is a descendant of revision_a.
1141
 
 
1142
 
        This is a helper function for update_revisions.
1143
 
 
1144
 
        :raises: DivergedBranches if revision_b has diverged from revision_a.
1145
 
        :returns: True if revision_b is a descendant of revision_a.
1146
 
        """
1147
 
        relation = self._revision_relations(revision_a, revision_b, graph)
1148
 
        if relation == 'b_descends_from_a':
1149
 
            return True
1150
 
        elif relation == 'diverged':
1151
 
            raise errors.DivergedBranches(self, other_branch)
1152
 
        elif relation == 'a_descends_from_b':
1153
 
            return False
1154
 
        else:
1155
 
            raise AssertionError("invalid relation: %r" % (relation,))
1156
 
 
1157
 
    def _revision_relations(self, revision_a, revision_b, graph):
1158
 
        """Determine the relationship between two revisions.
1159
 
 
1160
 
        :returns: One of: 'a_descends_from_b', 'b_descends_from_a', 'diverged'
1161
 
        """
1162
 
        heads = graph.heads([revision_a, revision_b])
1163
 
        if heads == set([revision_b]):
1164
 
            return 'b_descends_from_a'
1165
 
        elif heads == set([revision_a, revision_b]):
1166
 
            # These branches have diverged
1167
 
            return 'diverged'
1168
 
        elif heads == set([revision_a]):
1169
 
            return 'a_descends_from_b'
1170
 
        else:
1171
 
            raise AssertionError("invalid heads: %r" % (heads,))
1172
 
 
1173
 
 
1174
 
class BranchFormat(object):
1175
 
    """An encapsulation of the initialization and open routines for a format.
1176
 
 
1177
 
    Formats provide three things:
1178
 
     * An initialization routine,
1179
 
     * a format string,
1180
 
     * an open routine.
1181
 
 
1182
 
    Formats are placed in an dict by their format string for reference
1183
 
    during branch opening. Its not required that these be instances, they
1184
 
    can be classes themselves with class methods - it simply depends on
1185
 
    whether state is needed for a given format or not.
1186
 
 
1187
 
    Once a format is deprecated, just deprecate the initialize and open
1188
 
    methods on the format class. Do not deprecate the object, as the
1189
 
    object will be created every time regardless.
1190
 
    """
1191
 
 
1192
 
    _default_format = None
1193
 
    """The default format used for new branches."""
1194
 
 
1195
 
    _formats = {}
1196
 
    """The known formats."""
1197
 
 
1198
 
    def __eq__(self, other):
1199
 
        return self.__class__ is other.__class__
1200
 
 
1201
 
    def __ne__(self, other):
1202
 
        return not (self == other)
1203
 
 
1204
 
    @classmethod
1205
 
    def find_format(klass, a_bzrdir):
1206
 
        """Return the format for the branch object in a_bzrdir."""
1207
 
        try:
1208
 
            transport = a_bzrdir.get_branch_transport(None)
1209
 
            format_string = transport.get("format").read()
1210
 
            return klass._formats[format_string]
1211
 
        except errors.NoSuchFile:
1212
 
            raise errors.NotBranchError(path=transport.base)
1213
 
        except KeyError:
1214
 
            raise errors.UnknownFormatError(format=format_string, kind='branch')
1215
 
 
1216
 
    @classmethod
1217
 
    def get_default_format(klass):
1218
 
        """Return the current default format."""
1219
 
        return klass._default_format
1220
 
 
1221
 
    def get_reference(self, a_bzrdir):
1222
 
        """Get the target reference of the branch in a_bzrdir.
1223
 
 
1224
 
        format probing must have been completed before calling
1225
 
        this method - it is assumed that the format of the branch
1226
 
        in a_bzrdir is correct.
1227
 
 
1228
 
        :param a_bzrdir: The bzrdir to get the branch data from.
1229
 
        :return: None if the branch is not a reference branch.
1230
 
        """
1231
 
        return None
1232
 
 
1233
 
    @classmethod
1234
 
    def set_reference(self, a_bzrdir, to_branch):
1235
 
        """Set the target reference of the branch in a_bzrdir.
1236
 
 
1237
 
        format probing must have been completed before calling
1238
 
        this method - it is assumed that the format of the branch
1239
 
        in a_bzrdir is correct.
1240
 
 
1241
 
        :param a_bzrdir: The bzrdir to set the branch reference for.
1242
 
        :param to_branch: branch that the checkout is to reference
1243
 
        """
1244
 
        raise NotImplementedError(self.set_reference)
1245
 
 
1246
 
    def get_format_string(self):
1247
 
        """Return the ASCII format string that identifies this format."""
1248
 
        raise NotImplementedError(self.get_format_string)
1249
 
 
1250
 
    def get_format_description(self):
1251
 
        """Return the short format description for this format."""
1252
 
        raise NotImplementedError(self.get_format_description)
1253
 
 
1254
 
    def _initialize_helper(self, a_bzrdir, utf8_files, lock_type='metadir',
1255
 
                           set_format=True):
1256
 
        """Initialize a branch in a bzrdir, with specified files
1257
 
 
1258
 
        :param a_bzrdir: The bzrdir to initialize the branch in
1259
 
        :param utf8_files: The files to create as a list of
1260
 
            (filename, content) tuples
1261
 
        :param set_format: If True, set the format with
1262
 
            self.get_format_string.  (BzrBranch4 has its format set
1263
 
            elsewhere)
1264
 
        :return: a branch in this format
1265
 
        """
1266
 
        mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
1267
 
        branch_transport = a_bzrdir.get_branch_transport(self)
1268
 
        lock_map = {
1269
 
            'metadir': ('lock', lockdir.LockDir),
1270
 
            'branch4': ('branch-lock', lockable_files.TransportLock),
1271
 
        }
1272
 
        lock_name, lock_class = lock_map[lock_type]
1273
 
        control_files = lockable_files.LockableFiles(branch_transport,
1274
 
            lock_name, lock_class)
1275
 
        control_files.create_lock()
1276
 
        control_files.lock_write()
1277
 
        if set_format:
1278
 
            utf8_files += [('format', self.get_format_string())]
1279
 
        try:
1280
 
            for (filename, content) in utf8_files:
1281
 
                branch_transport.put_bytes(
1282
 
                    filename, content,
1283
 
                    mode=a_bzrdir._get_file_mode())
1284
 
        finally:
1285
 
            control_files.unlock()
1286
 
        return self.open(a_bzrdir, _found=True)
1287
 
 
1288
 
    def initialize(self, a_bzrdir):
1289
 
        """Create a branch of this format in a_bzrdir."""
1290
 
        raise NotImplementedError(self.initialize)
1291
 
 
1292
 
    def is_supported(self):
1293
 
        """Is this format supported?
1294
 
 
1295
 
        Supported formats can be initialized and opened.
1296
 
        Unsupported formats may not support initialization or committing or
1297
 
        some other features depending on the reason for not being supported.
1298
 
        """
1299
 
        return True
1300
 
 
1301
 
    def make_tags(self, branch):
1302
 
        """Create a tags object for branch.
1303
 
 
1304
 
        This method is on BranchFormat, because BranchFormats are reflected
1305
 
        over the wire via network_name(), whereas full Branch instances require
1306
 
        multiple VFS method calls to operate at all.
1307
 
 
1308
 
        The default implementation returns a disabled-tags instance.
1309
 
 
1310
 
        Note that it is normal for branch to be a RemoteBranch when using tags
1311
 
        on a RemoteBranch.
1312
 
        """
1313
 
        return DisabledTags(branch)
1314
 
 
1315
 
    def network_name(self):
1316
 
        """A simple byte string uniquely identifying this format for RPC calls.
1317
 
 
1318
 
        MetaDir branch formats use their disk format string to identify the
1319
 
        repository over the wire. All in one formats such as bzr < 0.8, and
1320
 
        foreign formats like svn/git and hg should use some marker which is
1321
 
        unique and immutable.
1322
 
        """
1323
 
        raise NotImplementedError(self.network_name)
1324
 
 
1325
 
    def open(self, a_bzrdir, _found=False):
1326
 
        """Return the branch object for a_bzrdir
1327
 
 
1328
 
        _found is a private parameter, do not use it. It is used to indicate
1329
 
               if format probing has already be done.
1330
 
        """
1331
 
        raise NotImplementedError(self.open)
1332
 
 
1333
 
    @classmethod
1334
 
    def register_format(klass, format):
1335
 
        """Register a metadir format."""
1336
 
        klass._formats[format.get_format_string()] = format
1337
 
        # Metadir formats have a network name of their format string, and get
1338
 
        # registered as class factories.
1339
 
        network_format_registry.register(format.get_format_string(), format.__class__)
1340
 
 
1341
 
    @classmethod
1342
 
    def set_default_format(klass, format):
1343
 
        klass._default_format = format
1344
 
 
1345
 
    def supports_stacking(self):
1346
 
        """True if this format records a stacked-on branch."""
1347
 
        return False
1348
 
 
1349
 
    @classmethod
1350
 
    def unregister_format(klass, format):
1351
 
        del klass._formats[format.get_format_string()]
1352
 
 
1353
 
    def __str__(self):
1354
 
        return self.get_format_description().rstrip()
1355
 
 
1356
 
    def supports_tags(self):
1357
 
        """True if this format supports tags stored in the branch"""
1358
 
        return False  # by default
1359
 
 
1360
 
 
1361
 
class BranchHooks(Hooks):
1362
 
    """A dictionary mapping hook name to a list of callables for branch hooks.
1363
 
 
1364
 
    e.g. ['set_rh'] Is the list of items to be called when the
1365
 
    set_revision_history function is invoked.
1366
 
    """
1367
 
 
1368
 
    def __init__(self):
1369
 
        """Create the default hooks.
1370
 
 
1371
 
        These are all empty initially, because by default nothing should get
1372
 
        notified.
1373
 
        """
1374
 
        Hooks.__init__(self)
1375
 
        # Introduced in 0.15:
1376
 
        # invoked whenever the revision history has been set
1377
 
        # with set_revision_history. The api signature is
1378
 
        # (branch, revision_history), and the branch will
1379
 
        # be write-locked.
1380
 
        self['set_rh'] = []
1381
 
        # Invoked after a branch is opened. The api signature is (branch).
1382
 
        self['open'] = []
1383
 
        # invoked after a push operation completes.
1384
 
        # the api signature is
1385
 
        # (push_result)
1386
 
        # containing the members
1387
 
        # (source, local, master, old_revno, old_revid, new_revno, new_revid)
1388
 
        # where local is the local target branch or None, master is the target
1389
 
        # master branch, and the rest should be self explanatory. The source
1390
 
        # is read locked and the target branches write locked. Source will
1391
 
        # be the local low-latency branch.
1392
 
        self['post_push'] = []
1393
 
        # invoked after a pull operation completes.
1394
 
        # the api signature is
1395
 
        # (pull_result)
1396
 
        # containing the members
1397
 
        # (source, local, master, old_revno, old_revid, new_revno, new_revid)
1398
 
        # where local is the local branch or None, master is the target
1399
 
        # master branch, and the rest should be self explanatory. The source
1400
 
        # is read locked and the target branches write locked. The local
1401
 
        # branch is the low-latency branch.
1402
 
        self['post_pull'] = []
1403
 
        # invoked before a commit operation takes place.
1404
 
        # the api signature is
1405
 
        # (local, master, old_revno, old_revid, future_revno, future_revid,
1406
 
        #  tree_delta, future_tree).
1407
 
        # old_revid is NULL_REVISION for the first commit to a branch
1408
 
        # tree_delta is a TreeDelta object describing changes from the basis
1409
 
        # revision, hooks MUST NOT modify this delta
1410
 
        # future_tree is an in-memory tree obtained from
1411
 
        # CommitBuilder.revision_tree() and hooks MUST NOT modify this tree
1412
 
        self['pre_commit'] = []
1413
 
        # invoked after a commit operation completes.
1414
 
        # the api signature is
1415
 
        # (local, master, old_revno, old_revid, new_revno, new_revid)
1416
 
        # old_revid is NULL_REVISION for the first commit to a branch.
1417
 
        self['post_commit'] = []
1418
 
        # invoked after a uncommit operation completes.
1419
 
        # the api signature is
1420
 
        # (local, master, old_revno, old_revid, new_revno, new_revid) where
1421
 
        # local is the local branch or None, master is the target branch,
1422
 
        # and an empty branch recieves new_revno of 0, new_revid of None.
1423
 
        self['post_uncommit'] = []
1424
 
        # Introduced in 1.6
1425
 
        # Invoked before the tip of a branch changes.
1426
 
        # the api signature is
1427
 
        # (params) where params is a ChangeBranchTipParams with the members
1428
 
        # (branch, old_revno, new_revno, old_revid, new_revid)
1429
 
        self['pre_change_branch_tip'] = []
1430
 
        # Introduced in 1.4
1431
 
        # Invoked after the tip of a branch changes.
1432
 
        # the api signature is
1433
 
        # (params) where params is a ChangeBranchTipParams with the members
1434
 
        # (branch, old_revno, new_revno, old_revid, new_revid)
1435
 
        self['post_change_branch_tip'] = []
1436
 
        # Introduced in 1.9
1437
 
        # Invoked when a stacked branch activates its fallback locations and
1438
 
        # allows the transformation of the url of said location.
1439
 
        # the api signature is
1440
 
        # (branch, url) where branch is the branch having its fallback
1441
 
        # location activated and url is the url for the fallback location.
1442
 
        # The hook should return a url.
1443
 
        self['transform_fallback_location'] = []
1444
 
 
1445
 
 
1446
 
# install the default hooks into the Branch class.
1447
 
Branch.hooks = BranchHooks()
1448
 
 
1449
 
 
1450
 
class ChangeBranchTipParams(object):
1451
 
    """Object holding parameters passed to *_change_branch_tip hooks.
1452
 
 
1453
 
    There are 5 fields that hooks may wish to access:
1454
 
 
1455
 
    :ivar branch: the branch being changed
1456
 
    :ivar old_revno: revision number before the change
1457
 
    :ivar new_revno: revision number after the change
1458
 
    :ivar old_revid: revision id before the change
1459
 
    :ivar new_revid: revision id after the change
1460
 
 
1461
 
    The revid fields are strings. The revno fields are integers.
1462
 
    """
1463
 
 
1464
 
    def __init__(self, branch, old_revno, new_revno, old_revid, new_revid):
1465
 
        """Create a group of ChangeBranchTip parameters.
1466
 
 
1467
 
        :param branch: The branch being changed.
1468
 
        :param old_revno: Revision number before the change.
1469
 
        :param new_revno: Revision number after the change.
1470
 
        :param old_revid: Tip revision id before the change.
1471
 
        :param new_revid: Tip revision id after the change.
1472
 
        """
1473
 
        self.branch = branch
1474
 
        self.old_revno = old_revno
1475
 
        self.new_revno = new_revno
1476
 
        self.old_revid = old_revid
1477
 
        self.new_revid = new_revid
1478
 
 
1479
 
    def __eq__(self, other):
1480
 
        return self.__dict__ == other.__dict__
1481
 
 
1482
 
    def __repr__(self):
1483
 
        return "<%s of %s from (%s, %s) to (%s, %s)>" % (
1484
 
            self.__class__.__name__, self.branch,
1485
 
            self.old_revno, self.old_revid, self.new_revno, self.new_revid)
1486
 
 
1487
 
 
1488
 
class BzrBranchFormat4(BranchFormat):
1489
 
    """Bzr branch format 4.
1490
 
 
1491
 
    This format has:
1492
 
     - a revision-history file.
1493
 
     - a branch-lock lock file [ to be shared with the bzrdir ]
1494
 
    """
1495
 
 
1496
 
    def get_format_description(self):
1497
 
        """See BranchFormat.get_format_description()."""
1498
 
        return "Branch format 4"
1499
 
 
1500
 
    def initialize(self, a_bzrdir):
1501
 
        """Create a branch of this format in a_bzrdir."""
1502
 
        utf8_files = [('revision-history', ''),
1503
 
                      ('branch-name', ''),
1504
 
                      ]
1505
 
        return self._initialize_helper(a_bzrdir, utf8_files,
1506
 
                                       lock_type='branch4', set_format=False)
1507
 
 
1508
 
    def __init__(self):
1509
 
        super(BzrBranchFormat4, self).__init__()
1510
 
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
1511
 
 
1512
 
    def network_name(self):
1513
 
        """The network name for this format is the control dirs disk label."""
1514
 
        return self._matchingbzrdir.get_format_string()
1515
 
 
1516
 
    def open(self, a_bzrdir, _found=False):
1517
 
        """Return the branch object for a_bzrdir
1518
 
 
1519
 
        _found is a private parameter, do not use it. It is used to indicate
1520
 
               if format probing has already be done.
1521
 
        """
1522
 
        if not _found:
1523
 
            # we are being called directly and must probe.
1524
 
            raise NotImplementedError
1525
 
        return BzrBranch(_format=self,
1526
 
                         _control_files=a_bzrdir._control_files,
1527
 
                         a_bzrdir=a_bzrdir,
1528
 
                         _repository=a_bzrdir.open_repository())
1529
 
 
1530
 
    def __str__(self):
1531
 
        return "Bazaar-NG branch format 4"
1532
 
 
1533
 
 
1534
 
class BranchFormatMetadir(BranchFormat):
1535
 
    """Common logic for meta-dir based branch formats."""
1536
 
 
1537
 
    def _branch_class(self):
1538
 
        """What class to instantiate on open calls."""
1539
 
        raise NotImplementedError(self._branch_class)
1540
 
 
1541
 
    def network_name(self):
1542
 
        """A simple byte string uniquely identifying this format for RPC calls.
1543
 
 
1544
 
        Metadir branch formats use their format string.
1545
 
        """
1546
 
        return self.get_format_string()
1547
 
 
1548
 
    def open(self, a_bzrdir, _found=False):
1549
 
        """Return the branch object for a_bzrdir.
1550
 
 
1551
 
        _found is a private parameter, do not use it. It is used to indicate
1552
 
               if format probing has already be done.
1553
 
        """
1554
 
        if not _found:
1555
 
            format = BranchFormat.find_format(a_bzrdir)
1556
 
            if format.__class__ != self.__class__:
1557
 
                raise AssertionError("wrong format %r found for %r" %
1558
 
                    (format, self))
1559
 
        try:
1560
 
            transport = a_bzrdir.get_branch_transport(None)
1561
 
            control_files = lockable_files.LockableFiles(transport, 'lock',
1562
 
                                                         lockdir.LockDir)
1563
 
            return self._branch_class()(_format=self,
1564
 
                              _control_files=control_files,
1565
 
                              a_bzrdir=a_bzrdir,
1566
 
                              _repository=a_bzrdir.find_repository())
1567
 
        except errors.NoSuchFile:
1568
 
            raise errors.NotBranchError(path=transport.base)
1569
 
 
1570
 
    def __init__(self):
1571
 
        super(BranchFormatMetadir, self).__init__()
1572
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
1573
 
        self._matchingbzrdir.set_branch_format(self)
1574
 
 
1575
 
    def supports_tags(self):
1576
 
        return True
1577
 
 
1578
 
 
1579
 
class BzrBranchFormat5(BranchFormatMetadir):
1580
 
    """Bzr branch format 5.
1581
 
 
1582
 
    This format has:
1583
 
     - a revision-history file.
1584
 
     - a format string
1585
 
     - a lock dir guarding the branch itself
1586
 
     - all of this stored in a branch/ subdirectory
1587
 
     - works with shared repositories.
1588
 
 
1589
 
    This format is new in bzr 0.8.
1590
 
    """
1591
 
 
1592
 
    def _branch_class(self):
1593
 
        return BzrBranch5
1594
 
 
1595
 
    def get_format_string(self):
1596
 
        """See BranchFormat.get_format_string()."""
1597
 
        return "Bazaar-NG branch format 5\n"
1598
 
 
1599
 
    def get_format_description(self):
1600
 
        """See BranchFormat.get_format_description()."""
1601
 
        return "Branch format 5"
1602
 
 
1603
 
    def initialize(self, a_bzrdir):
1604
 
        """Create a branch of this format in a_bzrdir."""
1605
 
        utf8_files = [('revision-history', ''),
1606
 
                      ('branch-name', ''),
1607
 
                      ]
1608
 
        return self._initialize_helper(a_bzrdir, utf8_files)
1609
 
 
1610
 
    def supports_tags(self):
1611
 
        return False
1612
 
 
1613
 
 
1614
 
class BzrBranchFormat6(BranchFormatMetadir):
1615
 
    """Branch format with last-revision and tags.
1616
 
 
1617
 
    Unlike previous formats, this has no explicit revision history. Instead,
1618
 
    this just stores the last-revision, and the left-hand history leading
1619
 
    up to there is the history.
1620
 
 
1621
 
    This format was introduced in bzr 0.15
1622
 
    and became the default in 0.91.
1623
 
    """
1624
 
 
1625
 
    def _branch_class(self):
1626
 
        return BzrBranch6
1627
 
 
1628
 
    def get_format_string(self):
1629
 
        """See BranchFormat.get_format_string()."""
1630
 
        return "Bazaar Branch Format 6 (bzr 0.15)\n"
1631
 
 
1632
 
    def get_format_description(self):
1633
 
        """See BranchFormat.get_format_description()."""
1634
 
        return "Branch format 6"
1635
 
 
1636
 
    def initialize(self, a_bzrdir):
1637
 
        """Create a branch of this format in a_bzrdir."""
1638
 
        utf8_files = [('last-revision', '0 null:\n'),
1639
 
                      ('branch.conf', ''),
1640
 
                      ('tags', ''),
1641
 
                      ]
1642
 
        return self._initialize_helper(a_bzrdir, utf8_files)
1643
 
 
1644
 
    def make_tags(self, branch):
1645
 
        """See bzrlib.branch.BranchFormat.make_tags()."""
1646
 
        return BasicTags(branch)
1647
 
 
1648
 
 
1649
 
 
1650
 
class BzrBranchFormat7(BranchFormatMetadir):
1651
 
    """Branch format with last-revision, tags, and a stacked location pointer.
1652
 
 
1653
 
    The stacked location pointer is passed down to the repository and requires
1654
 
    a repository format with supports_external_lookups = True.
1655
 
 
1656
 
    This format was introduced in bzr 1.6.
1657
 
    """
1658
 
 
1659
 
    def _branch_class(self):
1660
 
        return BzrBranch7
1661
 
 
1662
 
    def get_format_string(self):
1663
 
        """See BranchFormat.get_format_string()."""
1664
 
        return "Bazaar Branch Format 7 (needs bzr 1.6)\n"
1665
 
 
1666
 
    def get_format_description(self):
1667
 
        """See BranchFormat.get_format_description()."""
1668
 
        return "Branch format 7"
1669
 
 
1670
 
    def initialize(self, a_bzrdir):
1671
 
        """Create a branch of this format in a_bzrdir."""
1672
 
        utf8_files = [('last-revision', '0 null:\n'),
1673
 
                      ('branch.conf', ''),
1674
 
                      ('tags', ''),
1675
 
                      ]
1676
 
        return self._initialize_helper(a_bzrdir, utf8_files)
1677
 
 
1678
 
    def __init__(self):
1679
 
        super(BzrBranchFormat7, self).__init__()
1680
 
        self._matchingbzrdir.repository_format = \
1681
 
            RepositoryFormatKnitPack5RichRoot()
1682
 
 
1683
 
    def make_tags(self, branch):
1684
 
        """See bzrlib.branch.BranchFormat.make_tags()."""
1685
 
        return BasicTags(branch)
1686
 
 
1687
 
    def supports_stacking(self):
1688
 
        return True
1689
 
 
1690
 
 
1691
 
class BranchReferenceFormat(BranchFormat):
1692
 
    """Bzr branch reference format.
1693
 
 
1694
 
    Branch references are used in implementing checkouts, they
1695
 
    act as an alias to the real branch which is at some other url.
1696
 
 
1697
 
    This format has:
1698
 
     - A location file
1699
 
     - a format string
1700
 
    """
1701
 
 
1702
 
    def get_format_string(self):
1703
 
        """See BranchFormat.get_format_string()."""
1704
 
        return "Bazaar-NG Branch Reference Format 1\n"
1705
 
 
1706
 
    def get_format_description(self):
1707
 
        """See BranchFormat.get_format_description()."""
1708
 
        return "Checkout reference format 1"
1709
 
 
1710
 
    def get_reference(self, a_bzrdir):
1711
 
        """See BranchFormat.get_reference()."""
1712
 
        transport = a_bzrdir.get_branch_transport(None)
1713
 
        return transport.get('location').read()
1714
 
 
1715
 
    def set_reference(self, a_bzrdir, to_branch):
1716
 
        """See BranchFormat.set_reference()."""
1717
 
        transport = a_bzrdir.get_branch_transport(None)
1718
 
        location = transport.put_bytes('location', to_branch.base)
1719
 
 
1720
 
    def initialize(self, a_bzrdir, target_branch=None):
1721
 
        """Create a branch of this format in a_bzrdir."""
1722
 
        if target_branch is None:
1723
 
            # this format does not implement branch itself, thus the implicit
1724
 
            # creation contract must see it as uninitializable
1725
 
            raise errors.UninitializableFormat(self)
1726
 
        mutter('creating branch reference in %s', a_bzrdir.transport.base)
1727
 
        branch_transport = a_bzrdir.get_branch_transport(self)
1728
 
        branch_transport.put_bytes('location',
1729
 
            target_branch.bzrdir.root_transport.base)
1730
 
        branch_transport.put_bytes('format', self.get_format_string())
1731
 
        return self.open(
1732
 
            a_bzrdir, _found=True,
1733
 
            possible_transports=[target_branch.bzrdir.root_transport])
1734
 
 
1735
 
    def __init__(self):
1736
 
        super(BranchReferenceFormat, self).__init__()
1737
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
1738
 
        self._matchingbzrdir.set_branch_format(self)
1739
 
 
1740
 
    def _make_reference_clone_function(format, a_branch):
1741
 
        """Create a clone() routine for a branch dynamically."""
1742
 
        def clone(to_bzrdir, revision_id=None,
1743
 
            repository_policy=None):
1744
 
            """See Branch.clone()."""
1745
 
            return format.initialize(to_bzrdir, a_branch)
1746
 
            # cannot obey revision_id limits when cloning a reference ...
1747
 
            # FIXME RBC 20060210 either nuke revision_id for clone, or
1748
 
            # emit some sort of warning/error to the caller ?!
1749
 
        return clone
1750
 
 
1751
 
    def open(self, a_bzrdir, _found=False, location=None,
1752
 
             possible_transports=None):
1753
 
        """Return the branch that the branch reference in a_bzrdir points at.
1754
 
 
1755
 
        _found is a private parameter, do not use it. It is used to indicate
1756
 
               if format probing has already be done.
1757
 
        """
1758
 
        if not _found:
1759
 
            format = BranchFormat.find_format(a_bzrdir)
1760
 
            if format.__class__ != self.__class__:
1761
 
                raise AssertionError("wrong format %r found for %r" %
1762
 
                    (format, self))
1763
 
        if location is None:
1764
 
            location = self.get_reference(a_bzrdir)
1765
 
        real_bzrdir = bzrdir.BzrDir.open(
1766
 
            location, possible_transports=possible_transports)
1767
 
        result = real_bzrdir.open_branch()
1768
 
        # this changes the behaviour of result.clone to create a new reference
1769
 
        # rather than a copy of the content of the branch.
1770
 
        # I did not use a proxy object because that needs much more extensive
1771
 
        # testing, and we are only changing one behaviour at the moment.
1772
 
        # If we decide to alter more behaviours - i.e. the implicit nickname
1773
 
        # then this should be refactored to introduce a tested proxy branch
1774
 
        # and a subclass of that for use in overriding clone() and ....
1775
 
        # - RBC 20060210
1776
 
        result.clone = self._make_reference_clone_function(result)
1777
 
        return result
1778
 
 
1779
 
 
1780
 
network_format_registry = registry.FormatRegistry()
1781
 
"""Registry of formats indexed by their network name.
1782
 
 
1783
 
The network name for a branch format is an identifier that can be used when
1784
 
referring to formats with smart server operations. See
1785
 
BranchFormat.network_name() for more detail.
1786
 
"""
1787
 
 
1788
 
 
1789
 
# formats which have no format string are not discoverable
1790
 
# and not independently creatable, so are not registered.
1791
 
__format5 = BzrBranchFormat5()
1792
 
__format6 = BzrBranchFormat6()
1793
 
__format7 = BzrBranchFormat7()
1794
 
BranchFormat.register_format(__format5)
1795
 
BranchFormat.register_format(BranchReferenceFormat())
1796
 
BranchFormat.register_format(__format6)
1797
 
BranchFormat.register_format(__format7)
1798
 
BranchFormat.set_default_format(__format6)
1799
 
_legacy_formats = [BzrBranchFormat4(),
1800
 
    ]
1801
 
network_format_registry.register(
1802
 
    _legacy_formats[0].network_name(), _legacy_formats[0].__class__)
1803
 
 
1804
 
 
1805
 
class BzrBranch(Branch):
1806
 
    """A branch stored in the actual filesystem.
1807
 
 
1808
 
    Note that it's "local" in the context of the filesystem; it doesn't
1809
 
    really matter if it's on an nfs/smb/afs/coda/... share, as long as
1810
 
    it's writable, and can be accessed via the normal filesystem API.
1811
 
 
1812
 
    :ivar _transport: Transport for file operations on this branch's
1813
 
        control files, typically pointing to the .bzr/branch directory.
1814
 
    :ivar repository: Repository for this branch.
1815
 
    :ivar base: The url of the base directory for this branch; the one
1816
 
        containing the .bzr directory.
1817
 
    """
1818
 
 
1819
 
    def __init__(self, _format=None,
1820
 
                 _control_files=None, a_bzrdir=None, _repository=None):
1821
 
        """Create new branch object at a particular location."""
1822
 
        if a_bzrdir is None:
1823
 
            raise ValueError('a_bzrdir must be supplied')
1824
 
        else:
1825
 
            self.bzrdir = a_bzrdir
1826
 
        self._base = self.bzrdir.transport.clone('..').base
1827
 
        # XXX: We should be able to just do
1828
 
        #   self.base = self.bzrdir.root_transport.base
1829
 
        # but this does not quite work yet -- mbp 20080522
1830
 
        self._format = _format
1831
 
        if _control_files is None:
1832
 
            raise ValueError('BzrBranch _control_files is None')
1833
 
        self.control_files = _control_files
1834
 
        self._transport = _control_files._transport
1835
 
        self.repository = _repository
1836
 
        Branch.__init__(self)
1837
 
 
1838
 
    def __str__(self):
1839
 
        return '%s(%r)' % (self.__class__.__name__, self.base)
1840
 
 
1841
 
    __repr__ = __str__
1842
 
 
1843
 
    def _get_base(self):
1844
 
        """Returns the directory containing the control directory."""
1845
 
        return self._base
1846
 
 
1847
 
    base = property(_get_base, doc="The URL for the root of this branch.")
1848
 
 
1849
 
    def is_locked(self):
1850
 
        return self.control_files.is_locked()
1851
 
 
1852
 
    def lock_write(self, token=None):
1853
 
        repo_token = self.repository.lock_write()
1854
 
        try:
1855
 
            token = self.control_files.lock_write(token=token)
1856
 
        except:
1857
 
            self.repository.unlock()
1858
 
            raise
1859
 
        return token
1860
 
 
1861
 
    def lock_read(self):
1862
 
        self.repository.lock_read()
1863
 
        try:
1864
 
            self.control_files.lock_read()
1865
 
        except:
1866
 
            self.repository.unlock()
1867
 
            raise
1868
 
 
1869
 
    def unlock(self):
1870
 
        # TODO: test for failed two phase locks. This is known broken.
1871
 
        try:
1872
 
            self.control_files.unlock()
1873
 
        finally:
1874
 
            self.repository.unlock()
1875
 
        if not self.control_files.is_locked():
1876
 
            # we just released the lock
1877
 
            self._clear_cached_state()
1878
 
 
1879
 
    def peek_lock_mode(self):
1880
 
        if self.control_files._lock_count == 0:
1881
 
            return None
1882
 
        else:
1883
 
            return self.control_files._lock_mode
1884
 
 
1885
 
    def get_physical_lock_status(self):
1886
 
        return self.control_files.get_physical_lock_status()
1887
 
 
1888
 
    @needs_read_lock
1889
 
    def print_file(self, file, revision_id):
1890
 
        """See Branch.print_file."""
1891
 
        return self.repository.print_file(file, revision_id)
1892
 
 
1893
 
    def _write_revision_history(self, history):
1894
 
        """Factored out of set_revision_history.
1895
 
 
1896
 
        This performs the actual writing to disk.
1897
 
        It is intended to be called by BzrBranch5.set_revision_history."""
1898
 
        self._transport.put_bytes(
1899
 
            'revision-history', '\n'.join(history),
1900
 
            mode=self.bzrdir._get_file_mode())
1901
 
 
1902
 
    @needs_write_lock
1903
 
    def set_revision_history(self, rev_history):
1904
 
        """See Branch.set_revision_history."""
1905
 
        if 'evil' in debug.debug_flags:
1906
 
            mutter_callsite(3, "set_revision_history scales with history.")
1907
 
        check_not_reserved_id = _mod_revision.check_not_reserved_id
1908
 
        for rev_id in rev_history:
1909
 
            check_not_reserved_id(rev_id)
1910
 
        if Branch.hooks['post_change_branch_tip']:
1911
 
            # Don't calculate the last_revision_info() if there are no hooks
1912
 
            # that will use it.
1913
 
            old_revno, old_revid = self.last_revision_info()
1914
 
        if len(rev_history) == 0:
1915
 
            revid = _mod_revision.NULL_REVISION
1916
 
        else:
1917
 
            revid = rev_history[-1]
1918
 
        self._run_pre_change_branch_tip_hooks(len(rev_history), revid)
1919
 
        self._write_revision_history(rev_history)
1920
 
        self._clear_cached_state()
1921
 
        self._cache_revision_history(rev_history)
1922
 
        for hook in Branch.hooks['set_rh']:
1923
 
            hook(self, rev_history)
1924
 
        if Branch.hooks['post_change_branch_tip']:
1925
 
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
1926
 
 
1927
 
    def _synchronize_history(self, destination, revision_id):
1928
 
        """Synchronize last revision and revision history between branches.
1929
 
 
1930
 
        This version is most efficient when the destination is also a
1931
 
        BzrBranch5, but works for BzrBranch6 as long as the revision
1932
 
        history is the true lefthand parent history, and all of the revisions
1933
 
        are in the destination's repository.  If not, set_revision_history
1934
 
        will fail.
1935
 
 
1936
 
        :param destination: The branch to copy the history into
1937
 
        :param revision_id: The revision-id to truncate history at.  May
1938
 
          be None to copy complete history.
1939
 
        """
1940
 
        if not isinstance(destination._format, BzrBranchFormat5):
1941
 
            super(BzrBranch, self)._synchronize_history(
1942
 
                destination, revision_id)
1943
 
            return
1944
 
        if revision_id == _mod_revision.NULL_REVISION:
1945
 
            new_history = []
1946
 
        else:
1947
 
            new_history = self.revision_history()
1948
 
        if revision_id is not None and new_history != []:
1949
 
            try:
1950
 
                new_history = new_history[:new_history.index(revision_id) + 1]
1951
 
            except ValueError:
1952
 
                rev = self.repository.get_revision(revision_id)
1953
 
                new_history = rev.get_history(self.repository)[1:]
1954
 
        destination.set_revision_history(new_history)
1955
 
 
1956
 
    @needs_write_lock
1957
 
    def set_last_revision_info(self, revno, revision_id):
1958
 
        """Set the last revision of this branch.
1959
 
 
1960
 
        The caller is responsible for checking that the revno is correct
1961
 
        for this revision id.
1962
 
 
1963
 
        It may be possible to set the branch last revision to an id not
1964
 
        present in the repository.  However, branches can also be
1965
 
        configured to check constraints on history, in which case this may not
1966
 
        be permitted.
1967
 
        """
1968
 
        revision_id = _mod_revision.ensure_null(revision_id)
1969
 
        # this old format stores the full history, but this api doesn't
1970
 
        # provide it, so we must generate, and might as well check it's
1971
 
        # correct
1972
 
        history = self._lefthand_history(revision_id)
1973
 
        if len(history) != revno:
1974
 
            raise AssertionError('%d != %d' % (len(history), revno))
1975
 
        self.set_revision_history(history)
1976
 
 
1977
 
    def _gen_revision_history(self):
1978
 
        history = self._transport.get_bytes('revision-history').split('\n')
1979
 
        if history[-1:] == ['']:
1980
 
            # There shouldn't be a trailing newline, but just in case.
1981
 
            history.pop()
1982
 
        return history
1983
 
 
1984
 
    @needs_write_lock
1985
 
    def generate_revision_history(self, revision_id, last_rev=None,
1986
 
        other_branch=None):
1987
 
        """Create a new revision history that will finish with revision_id.
1988
 
 
1989
 
        :param revision_id: the new tip to use.
1990
 
        :param last_rev: The previous last_revision. If not None, then this
1991
 
            must be a ancestory of revision_id, or DivergedBranches is raised.
1992
 
        :param other_branch: The other branch that DivergedBranches should
1993
 
            raise with respect to.
1994
 
        """
1995
 
        self.set_revision_history(self._lefthand_history(revision_id,
1996
 
            last_rev, other_branch))
1997
 
 
1998
 
    def basis_tree(self):
1999
 
        """See Branch.basis_tree."""
2000
 
        return self.repository.revision_tree(self.last_revision())
2001
 
 
2002
 
    @needs_write_lock
2003
 
    def pull(self, source, overwrite=False, stop_revision=None,
2004
 
             _hook_master=None, run_hooks=True, possible_transports=None,
2005
 
             _override_hook_target=None):
2006
 
        """See Branch.pull.
2007
 
 
2008
 
        :param _hook_master: Private parameter - set the branch to
2009
 
            be supplied as the master to pull hooks.
2010
 
        :param run_hooks: Private parameter - if false, this branch
2011
 
            is being called because it's the master of the primary branch,
2012
 
            so it should not run its hooks.
2013
 
        :param _override_hook_target: Private parameter - set the branch to be
2014
 
            supplied as the target_branch to pull hooks.
2015
 
        """
2016
 
        result = PullResult()
2017
 
        result.source_branch = source
2018
 
        if _override_hook_target is None:
2019
 
            result.target_branch = self
2020
 
        else:
2021
 
            result.target_branch = _override_hook_target
2022
 
        source.lock_read()
2023
 
        try:
2024
 
            # We assume that during 'pull' the local repository is closer than
2025
 
            # the remote one.
2026
 
            graph = self.repository.get_graph(source.repository)
2027
 
            result.old_revno, result.old_revid = self.last_revision_info()
2028
 
            self.update_revisions(source, stop_revision, overwrite=overwrite,
2029
 
                                  graph=graph)
2030
 
            result.tag_conflicts = source.tags.merge_to(self.tags, overwrite)
2031
 
            result.new_revno, result.new_revid = self.last_revision_info()
2032
 
            if _hook_master:
2033
 
                result.master_branch = _hook_master
2034
 
                result.local_branch = result.target_branch
2035
 
            else:
2036
 
                result.master_branch = result.target_branch
2037
 
                result.local_branch = None
2038
 
            if run_hooks:
2039
 
                for hook in Branch.hooks['post_pull']:
2040
 
                    hook(result)
2041
 
        finally:
2042
 
            source.unlock()
2043
 
        return result
2044
 
 
2045
 
    def _get_parent_location(self):
2046
 
        _locs = ['parent', 'pull', 'x-pull']
2047
 
        for l in _locs:
2048
 
            try:
2049
 
                return self._transport.get_bytes(l).strip('\n')
2050
 
            except errors.NoSuchFile:
2051
 
                pass
2052
 
        return None
2053
 
 
2054
 
    @needs_read_lock
2055
 
    def push(self, target, overwrite=False, stop_revision=None,
2056
 
             _override_hook_source_branch=None):
2057
 
        """See Branch.push.
2058
 
 
2059
 
        This is the basic concrete implementation of push()
2060
 
 
2061
 
        :param _override_hook_source_branch: If specified, run
2062
 
        the hooks passing this Branch as the source, rather than self.
2063
 
        This is for use of RemoteBranch, where push is delegated to the
2064
 
        underlying vfs-based Branch.
2065
 
        """
2066
 
        # TODO: Public option to disable running hooks - should be trivial but
2067
 
        # needs tests.
2068
 
        return _run_with_write_locked_target(
2069
 
            target, self._push_with_bound_branches, target, overwrite,
2070
 
            stop_revision,
2071
 
            _override_hook_source_branch=_override_hook_source_branch)
2072
 
 
2073
 
    def _push_with_bound_branches(self, target, overwrite,
2074
 
            stop_revision,
2075
 
            _override_hook_source_branch=None):
2076
 
        """Push from self into target, and into target's master if any.
2077
 
 
2078
 
        This is on the base BzrBranch class even though it doesn't support
2079
 
        bound branches because the *target* might be bound.
2080
 
        """
2081
 
        def _run_hooks():
2082
 
            if _override_hook_source_branch:
2083
 
                result.source_branch = _override_hook_source_branch
2084
 
            for hook in Branch.hooks['post_push']:
2085
 
                hook(result)
2086
 
 
2087
 
        bound_location = target.get_bound_location()
2088
 
        if bound_location and target.base != bound_location:
2089
 
            # there is a master branch.
2090
 
            #
2091
 
            # XXX: Why the second check?  Is it even supported for a branch to
2092
 
            # be bound to itself? -- mbp 20070507
2093
 
            master_branch = target.get_master_branch()
2094
 
            master_branch.lock_write()
2095
 
            try:
2096
 
                # push into the master from this branch.
2097
 
                self._basic_push(master_branch, overwrite, stop_revision)
2098
 
                # and push into the target branch from this. Note that we push from
2099
 
                # this branch again, because its considered the highest bandwidth
2100
 
                # repository.
2101
 
                result = self._basic_push(target, overwrite, stop_revision)
2102
 
                result.master_branch = master_branch
2103
 
                result.local_branch = target
2104
 
                _run_hooks()
2105
 
                return result
2106
 
            finally:
2107
 
                master_branch.unlock()
2108
 
        else:
2109
 
            # no master branch
2110
 
            result = self._basic_push(target, overwrite, stop_revision)
2111
 
            # TODO: Why set master_branch and local_branch if there's no
2112
 
            # binding?  Maybe cleaner to just leave them unset? -- mbp
2113
 
            # 20070504
2114
 
            result.master_branch = target
2115
 
            result.local_branch = None
2116
 
            _run_hooks()
2117
 
            return result
2118
 
 
2119
 
    def _basic_push(self, target, overwrite, stop_revision):
2120
 
        """Basic implementation of push without bound branches or hooks.
2121
 
 
2122
 
        Must be called with self read locked and target write locked.
2123
 
        """
2124
 
        result = PushResult()
2125
 
        result.source_branch = self
2126
 
        result.target_branch = target
2127
 
        result.old_revno, result.old_revid = target.last_revision_info()
2128
 
        if result.old_revid != self.last_revision():
2129
 
            # We assume that during 'push' this repository is closer than
2130
 
            # the target.
2131
 
            graph = self.repository.get_graph(target.repository)
2132
 
            target.update_revisions(self, stop_revision, overwrite=overwrite,
2133
 
                                    graph=graph)
2134
 
        if self._push_should_merge_tags():
2135
 
            result.tag_conflicts = self.tags.merge_to(target.tags, overwrite)
2136
 
        result.new_revno, result.new_revid = target.last_revision_info()
2137
 
        return result
2138
 
 
2139
 
    def get_stacked_on_url(self):
2140
 
        raise errors.UnstackableBranchFormat(self._format, self.base)
2141
 
 
2142
 
    def set_push_location(self, location):
2143
 
        """See Branch.set_push_location."""
2144
 
        self.get_config().set_user_option(
2145
 
            'push_location', location,
2146
 
            store=_mod_config.STORE_LOCATION_NORECURSE)
2147
 
 
2148
 
    @needs_write_lock
2149
 
    def set_parent(self, url):
2150
 
        """See Branch.set_parent."""
2151
 
        # TODO: Maybe delete old location files?
2152
 
        # URLs should never be unicode, even on the local fs,
2153
 
        # FIXUP this and get_parent in a future branch format bump:
2154
 
        # read and rewrite the file. RBC 20060125
2155
 
        if url is not None:
2156
 
            if isinstance(url, unicode):
2157
 
                try:
2158
 
                    url = url.encode('ascii')
2159
 
                except UnicodeEncodeError:
2160
 
                    raise errors.InvalidURL(url,
2161
 
                        "Urls must be 7-bit ascii, "
2162
 
                        "use bzrlib.urlutils.escape")
2163
 
            url = urlutils.relative_url(self.base, url)
2164
 
        self._set_parent_location(url)
2165
 
 
2166
 
    def _set_parent_location(self, url):
2167
 
        if url is None:
2168
 
            self._transport.delete('parent')
2169
 
        else:
2170
 
            self._transport.put_bytes('parent', url + '\n',
2171
 
                mode=self.bzrdir._get_file_mode())
2172
 
 
2173
 
    def set_stacked_on_url(self, url):
2174
 
        raise errors.UnstackableBranchFormat(self._format, self.base)
2175
 
 
2176
 
 
2177
 
class BzrBranch5(BzrBranch):
2178
 
    """A format 5 branch. This supports new features over plain branches.
2179
 
 
2180
 
    It has support for a master_branch which is the data for bound branches.
2181
 
    """
2182
 
 
2183
 
    @needs_write_lock
2184
 
    def pull(self, source, overwrite=False, stop_revision=None,
2185
 
             run_hooks=True, possible_transports=None,
2186
 
             _override_hook_target=None):
2187
 
        """Pull from source into self, updating my master if any.
2188
 
 
2189
 
        :param run_hooks: Private parameter - if false, this branch
2190
 
            is being called because it's the master of the primary branch,
2191
 
            so it should not run its hooks.
2192
 
        """
2193
 
        bound_location = self.get_bound_location()
2194
 
        master_branch = None
2195
 
        if bound_location and source.base != bound_location:
2196
 
            # not pulling from master, so we need to update master.
2197
 
            master_branch = self.get_master_branch(possible_transports)
2198
 
            master_branch.lock_write()
2199
 
        try:
2200
 
            if master_branch:
2201
 
                # pull from source into master.
2202
 
                master_branch.pull(source, overwrite, stop_revision,
2203
 
                    run_hooks=False)
2204
 
            return super(BzrBranch5, self).pull(source, overwrite,
2205
 
                stop_revision, _hook_master=master_branch,
2206
 
                run_hooks=run_hooks,
2207
 
                _override_hook_target=_override_hook_target)
2208
 
        finally:
2209
 
            if master_branch:
2210
 
                master_branch.unlock()
2211
 
 
2212
 
    def get_bound_location(self):
2213
 
        try:
2214
 
            return self._transport.get_bytes('bound')[:-1]
2215
 
        except errors.NoSuchFile:
2216
 
            return None
2217
 
 
2218
 
    @needs_read_lock
2219
 
    def get_master_branch(self, possible_transports=None):
2220
 
        """Return the branch we are bound to.
2221
 
 
2222
 
        :return: Either a Branch, or None
2223
 
 
2224
 
        This could memoise the branch, but if thats done
2225
 
        it must be revalidated on each new lock.
2226
 
        So for now we just don't memoise it.
2227
 
        # RBC 20060304 review this decision.
2228
 
        """
2229
 
        bound_loc = self.get_bound_location()
2230
 
        if not bound_loc:
2231
 
            return None
2232
 
        try:
2233
 
            return Branch.open(bound_loc,
2234
 
                               possible_transports=possible_transports)
2235
 
        except (errors.NotBranchError, errors.ConnectionError), e:
2236
 
            raise errors.BoundBranchConnectionFailure(
2237
 
                    self, bound_loc, e)
2238
 
 
2239
 
    @needs_write_lock
2240
 
    def set_bound_location(self, location):
2241
 
        """Set the target where this branch is bound to.
2242
 
 
2243
 
        :param location: URL to the target branch
2244
 
        """
2245
 
        if location:
2246
 
            self._transport.put_bytes('bound', location+'\n',
2247
 
                mode=self.bzrdir._get_file_mode())
2248
 
        else:
2249
 
            try:
2250
 
                self._transport.delete('bound')
2251
 
            except errors.NoSuchFile:
2252
 
                return False
2253
 
            return True
2254
 
 
2255
 
    @needs_write_lock
2256
 
    def bind(self, other):
2257
 
        """Bind this branch to the branch other.
2258
 
 
2259
 
        This does not push or pull data between the branches, though it does
2260
 
        check for divergence to raise an error when the branches are not
2261
 
        either the same, or one a prefix of the other. That behaviour may not
2262
 
        be useful, so that check may be removed in future.
2263
 
 
2264
 
        :param other: The branch to bind to
2265
 
        :type other: Branch
2266
 
        """
2267
 
        # TODO: jam 20051230 Consider checking if the target is bound
2268
 
        #       It is debatable whether you should be able to bind to
2269
 
        #       a branch which is itself bound.
2270
 
        #       Committing is obviously forbidden,
2271
 
        #       but binding itself may not be.
2272
 
        #       Since we *have* to check at commit time, we don't
2273
 
        #       *need* to check here
2274
 
 
2275
 
        # we want to raise diverged if:
2276
 
        # last_rev is not in the other_last_rev history, AND
2277
 
        # other_last_rev is not in our history, and do it without pulling
2278
 
        # history around
2279
 
        self.set_bound_location(other.base)
2280
 
 
2281
 
    @needs_write_lock
2282
 
    def unbind(self):
2283
 
        """If bound, unbind"""
2284
 
        return self.set_bound_location(None)
2285
 
 
2286
 
    @needs_write_lock
2287
 
    def update(self, possible_transports=None):
2288
 
        """Synchronise this branch with the master branch if any.
2289
 
 
2290
 
        :return: None or the last_revision that was pivoted out during the
2291
 
                 update.
2292
 
        """
2293
 
        master = self.get_master_branch(possible_transports)
2294
 
        if master is not None:
2295
 
            old_tip = _mod_revision.ensure_null(self.last_revision())
2296
 
            self.pull(master, overwrite=True)
2297
 
            if self.repository.get_graph().is_ancestor(old_tip,
2298
 
                _mod_revision.ensure_null(self.last_revision())):
2299
 
                return None
2300
 
            return old_tip
2301
 
        return None
2302
 
 
2303
 
 
2304
 
class BzrBranch7(BzrBranch5):
2305
 
    """A branch with support for a fallback repository."""
2306
 
 
2307
 
    def _get_fallback_repository(self, url):
2308
 
        """Get the repository we fallback to at url."""
2309
 
        url = urlutils.join(self.base, url)
2310
 
        a_bzrdir = bzrdir.BzrDir.open(url,
2311
 
                                      possible_transports=[self._transport])
2312
 
        return a_bzrdir.open_branch().repository
2313
 
 
2314
 
    def _activate_fallback_location(self, url):
2315
 
        """Activate the branch/repository from url as a fallback repository."""
2316
 
        self.repository.add_fallback_repository(
2317
 
            self._get_fallback_repository(url))
2318
 
 
2319
 
    def _open_hook(self):
2320
 
        try:
2321
 
            url = self.get_stacked_on_url()
2322
 
        except (errors.UnstackableRepositoryFormat, errors.NotStacked,
2323
 
            errors.UnstackableBranchFormat):
2324
 
            pass
2325
 
        else:
2326
 
            for hook in Branch.hooks['transform_fallback_location']:
2327
 
                url = hook(self, url)
2328
 
                if url is None:
2329
 
                    hook_name = Branch.hooks.get_hook_name(hook)
2330
 
                    raise AssertionError(
2331
 
                        "'transform_fallback_location' hook %s returned "
2332
 
                        "None, not a URL." % hook_name)
2333
 
            self._activate_fallback_location(url)
2334
 
 
2335
 
    def _check_stackable_repo(self):
2336
 
        if not self.repository._format.supports_external_lookups:
2337
 
            raise errors.UnstackableRepositoryFormat(self.repository._format,
2338
 
                self.repository.base)
2339
 
 
2340
 
    def __init__(self, *args, **kwargs):
2341
 
        super(BzrBranch7, self).__init__(*args, **kwargs)
2342
 
        self._last_revision_info_cache = None
2343
 
        self._partial_revision_history_cache = []
2344
 
 
2345
 
    def _clear_cached_state(self):
2346
 
        super(BzrBranch7, self)._clear_cached_state()
2347
 
        self._last_revision_info_cache = None
2348
 
        self._partial_revision_history_cache = []
2349
 
 
2350
 
    def _last_revision_info(self):
2351
 
        revision_string = self._transport.get_bytes('last-revision')
2352
 
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
2353
 
        revision_id = cache_utf8.get_cached_utf8(revision_id)
2354
 
        revno = int(revno)
2355
 
        return revno, revision_id
2356
 
 
2357
 
    def _write_last_revision_info(self, revno, revision_id):
2358
 
        """Simply write out the revision id, with no checks.
2359
 
 
2360
 
        Use set_last_revision_info to perform this safely.
2361
 
 
2362
 
        Does not update the revision_history cache.
2363
 
        Intended to be called by set_last_revision_info and
2364
 
        _write_revision_history.
2365
 
        """
2366
 
        revision_id = _mod_revision.ensure_null(revision_id)
2367
 
        out_string = '%d %s\n' % (revno, revision_id)
2368
 
        self._transport.put_bytes('last-revision', out_string,
2369
 
            mode=self.bzrdir._get_file_mode())
2370
 
 
2371
 
    @needs_write_lock
2372
 
    def set_last_revision_info(self, revno, revision_id):
2373
 
        revision_id = _mod_revision.ensure_null(revision_id)
2374
 
        old_revno, old_revid = self.last_revision_info()
2375
 
        if self._get_append_revisions_only():
2376
 
            self._check_history_violation(revision_id)
2377
 
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
2378
 
        self._write_last_revision_info(revno, revision_id)
2379
 
        self._clear_cached_state()
2380
 
        self._last_revision_info_cache = revno, revision_id
2381
 
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2382
 
 
2383
 
    def _synchronize_history(self, destination, revision_id):
2384
 
        """Synchronize last revision and revision history between branches.
2385
 
 
2386
 
        :see: Branch._synchronize_history
2387
 
        """
2388
 
        # XXX: The base Branch has a fast implementation of this method based
2389
 
        # on set_last_revision_info, but BzrBranch/BzrBranch5 have a slower one
2390
 
        # that uses set_revision_history.  This class inherits from BzrBranch5,
2391
 
        # but wants the fast implementation, so it calls
2392
 
        # Branch._synchronize_history directly.
2393
 
        Branch._synchronize_history(self, destination, revision_id)
2394
 
 
2395
 
    def _check_history_violation(self, revision_id):
2396
 
        last_revision = _mod_revision.ensure_null(self.last_revision())
2397
 
        if _mod_revision.is_null(last_revision):
2398
 
            return
2399
 
        if last_revision not in self._lefthand_history(revision_id):
2400
 
            raise errors.AppendRevisionsOnlyViolation(self.base)
2401
 
 
2402
 
    def _gen_revision_history(self):
2403
 
        """Generate the revision history from last revision
2404
 
        """
2405
 
        last_revno, last_revision = self.last_revision_info()
2406
 
        self._extend_partial_history(stop_index=last_revno-1)
2407
 
        return list(reversed(self._partial_revision_history_cache))
2408
 
 
2409
 
    def _extend_partial_history(self, stop_index=None, stop_revision=None):
2410
 
        """Extend the partial history to include a given index
2411
 
 
2412
 
        If a stop_index is supplied, stop when that index has been reached.
2413
 
        If a stop_revision is supplied, stop when that revision is
2414
 
        encountered.  Otherwise, stop when the beginning of history is
2415
 
        reached.
2416
 
 
2417
 
        :param stop_index: The index which should be present.  When it is
2418
 
            present, history extension will stop.
2419
 
        :param revision_id: The revision id which should be present.  When
2420
 
            it is encountered, history extension will stop.
2421
 
        """
2422
 
        repo = self.repository
2423
 
        if len(self._partial_revision_history_cache) == 0:
2424
 
            iterator = repo.iter_reverse_revision_history(self.last_revision())
2425
 
        else:
2426
 
            start_revision = self._partial_revision_history_cache[-1]
2427
 
            iterator = repo.iter_reverse_revision_history(start_revision)
2428
 
            #skip the last revision in the list
2429
 
            next_revision = iterator.next()
2430
 
        for revision_id in iterator:
2431
 
            self._partial_revision_history_cache.append(revision_id)
2432
 
            if (stop_index is not None and
2433
 
                len(self._partial_revision_history_cache) > stop_index):
2434
 
                break
2435
 
            if revision_id == stop_revision:
2436
 
                break
2437
 
 
2438
 
    def _write_revision_history(self, history):
2439
 
        """Factored out of set_revision_history.
2440
 
 
2441
 
        This performs the actual writing to disk, with format-specific checks.
2442
 
        It is intended to be called by BzrBranch5.set_revision_history.
2443
 
        """
2444
 
        if len(history) == 0:
2445
 
            last_revision = 'null:'
2446
 
        else:
2447
 
            if history != self._lefthand_history(history[-1]):
2448
 
                raise errors.NotLefthandHistory(history)
2449
 
            last_revision = history[-1]
2450
 
        if self._get_append_revisions_only():
2451
 
            self._check_history_violation(last_revision)
2452
 
        self._write_last_revision_info(len(history), last_revision)
2453
 
 
2454
 
    @needs_write_lock
2455
 
    def _set_parent_location(self, url):
2456
 
        """Set the parent branch"""
2457
 
        self._set_config_location('parent_location', url, make_relative=True)
2458
 
 
2459
 
    @needs_read_lock
2460
 
    def _get_parent_location(self):
2461
 
        """Set the parent branch"""
2462
 
        return self._get_config_location('parent_location')
2463
 
 
2464
 
    def set_push_location(self, location):
2465
 
        """See Branch.set_push_location."""
2466
 
        self._set_config_location('push_location', location)
2467
 
 
2468
 
    def set_bound_location(self, location):
2469
 
        """See Branch.set_push_location."""
2470
 
        result = None
2471
 
        config = self.get_config()
2472
 
        if location is None:
2473
 
            if config.get_user_option('bound') != 'True':
2474
 
                return False
2475
 
            else:
2476
 
                config.set_user_option('bound', 'False', warn_masked=True)
2477
 
                return True
2478
 
        else:
2479
 
            self._set_config_location('bound_location', location,
2480
 
                                      config=config)
2481
 
            config.set_user_option('bound', 'True', warn_masked=True)
2482
 
        return True
2483
 
 
2484
 
    def _get_bound_location(self, bound):
2485
 
        """Return the bound location in the config file.
2486
 
 
2487
 
        Return None if the bound parameter does not match"""
2488
 
        config = self.get_config()
2489
 
        config_bound = (config.get_user_option('bound') == 'True')
2490
 
        if config_bound != bound:
2491
 
            return None
2492
 
        return self._get_config_location('bound_location', config=config)
2493
 
 
2494
 
    def get_bound_location(self):
2495
 
        """See Branch.set_push_location."""
2496
 
        return self._get_bound_location(True)
2497
 
 
2498
 
    def get_old_bound_location(self):
2499
 
        """See Branch.get_old_bound_location"""
2500
 
        return self._get_bound_location(False)
2501
 
 
2502
 
    def get_stacked_on_url(self):
2503
 
        # you can always ask for the URL; but you might not be able to use it
2504
 
        # if the repo can't support stacking.
2505
 
        ## self._check_stackable_repo()
2506
 
        stacked_url = self._get_config_location('stacked_on_location')
2507
 
        if stacked_url is None:
2508
 
            raise errors.NotStacked(self)
2509
 
        return stacked_url
2510
 
 
2511
 
    def set_append_revisions_only(self, enabled):
2512
 
        if enabled:
2513
 
            value = 'True'
2514
 
        else:
2515
 
            value = 'False'
2516
 
        self.get_config().set_user_option('append_revisions_only', value,
2517
 
            warn_masked=True)
2518
 
 
2519
 
    def set_stacked_on_url(self, url):
2520
 
        self._check_stackable_repo()
2521
 
        if not url:
2522
 
            try:
2523
 
                old_url = self.get_stacked_on_url()
2524
 
            except (errors.NotStacked, errors.UnstackableBranchFormat,
2525
 
                errors.UnstackableRepositoryFormat):
2526
 
                return
2527
 
            url = ''
2528
 
            # repositories don't offer an interface to remove fallback
2529
 
            # repositories today; take the conceptually simpler option and just
2530
 
            # reopen it.
2531
 
            self.repository = self.bzrdir.find_repository()
2532
 
            # for every revision reference the branch has, ensure it is pulled
2533
 
            # in.
2534
 
            source_repository = self._get_fallback_repository(old_url)
2535
 
            for revision_id in chain([self.last_revision()],
2536
 
                self.tags.get_reverse_tag_dict()):
2537
 
                self.repository.fetch(source_repository, revision_id,
2538
 
                    find_ghosts=True)
2539
 
        else:
2540
 
            self._activate_fallback_location(url)
2541
 
        # write this out after the repository is stacked to avoid setting a
2542
 
        # stacked config that doesn't work.
2543
 
        self._set_config_location('stacked_on_location', url)
2544
 
 
2545
 
    def _get_append_revisions_only(self):
2546
 
        value = self.get_config().get_user_option('append_revisions_only')
2547
 
        return value == 'True'
2548
 
 
2549
 
    @needs_write_lock
2550
 
    def generate_revision_history(self, revision_id, last_rev=None,
2551
 
                                  other_branch=None):
2552
 
        """See BzrBranch5.generate_revision_history"""
2553
 
        history = self._lefthand_history(revision_id, last_rev, other_branch)
2554
 
        revno = len(history)
2555
 
        self.set_last_revision_info(revno, revision_id)
2556
 
 
2557
 
    @needs_read_lock
2558
 
    def get_rev_id(self, revno, history=None):
2559
 
        """Find the revision id of the specified revno."""
2560
 
        if revno == 0:
2561
 
            return _mod_revision.NULL_REVISION
2562
 
 
2563
 
        last_revno, last_revision_id = self.last_revision_info()
2564
 
        if revno <= 0 or revno > last_revno:
2565
 
            raise errors.NoSuchRevision(self, revno)
2566
 
 
2567
 
        if history is not None:
2568
 
            return history[revno - 1]
2569
 
 
2570
 
        index = last_revno - revno
2571
 
        if len(self._partial_revision_history_cache) <= index:
2572
 
            self._extend_partial_history(stop_index=index)
2573
 
        if len(self._partial_revision_history_cache) > index:
2574
 
            return self._partial_revision_history_cache[index]
2575
 
        else:
2576
 
            raise errors.NoSuchRevision(self, revno)
2577
 
 
2578
 
    @needs_read_lock
2579
 
    def revision_id_to_revno(self, revision_id):
2580
 
        """Given a revision id, return its revno"""
2581
 
        if _mod_revision.is_null(revision_id):
2582
 
            return 0
2583
 
        try:
2584
 
            index = self._partial_revision_history_cache.index(revision_id)
2585
 
        except ValueError:
2586
 
            self._extend_partial_history(stop_revision=revision_id)
2587
 
            index = len(self._partial_revision_history_cache) - 1
2588
 
            if self._partial_revision_history_cache[index] != revision_id:
2589
 
                raise errors.NoSuchRevision(self, revision_id)
2590
 
        return self.revno() - index
2591
 
 
2592
 
 
2593
 
class BzrBranch6(BzrBranch7):
2594
 
    """See BzrBranchFormat6 for the capabilities of this branch.
2595
 
 
2596
 
    This subclass of BzrBranch7 disables the new features BzrBranch7 added,
2597
 
    i.e. stacking.
2598
 
    """
2599
 
 
2600
 
    def get_stacked_on_url(self):
2601
 
        raise errors.UnstackableBranchFormat(self._format, self.base)
2602
 
 
2603
 
    def set_stacked_on_url(self, url):
2604
 
        raise errors.UnstackableBranchFormat(self._format, self.base)
2605
 
 
2606
 
 
2607
 
######################################################################
2608
 
# results of operations
2609
 
 
2610
 
 
2611
 
class _Result(object):
2612
 
 
2613
 
    def _show_tag_conficts(self, to_file):
2614
 
        if not getattr(self, 'tag_conflicts', None):
2615
 
            return
2616
 
        to_file.write('Conflicting tags:\n')
2617
 
        for name, value1, value2 in self.tag_conflicts:
2618
 
            to_file.write('    %s\n' % (name, ))
2619
 
 
2620
 
 
2621
 
class PullResult(_Result):
2622
 
    """Result of a Branch.pull operation.
2623
 
 
2624
 
    :ivar old_revno: Revision number before pull.
2625
 
    :ivar new_revno: Revision number after pull.
2626
 
    :ivar old_revid: Tip revision id before pull.
2627
 
    :ivar new_revid: Tip revision id after pull.
2628
 
    :ivar source_branch: Source (local) branch object.
2629
 
    :ivar master_branch: Master branch of the target, or the target if no
2630
 
        Master
2631
 
    :ivar local_branch: target branch if there is a Master, else None
2632
 
    :ivar target_branch: Target/destination branch object.
2633
 
    :ivar tag_conflicts: A list of tag conflicts, see BasicTags.merge_to
2634
 
    """
2635
 
 
2636
 
    def __int__(self):
2637
 
        # DEPRECATED: pull used to return the change in revno
2638
 
        return self.new_revno - self.old_revno
2639
 
 
2640
 
    def report(self, to_file):
2641
 
        if not is_quiet():
2642
 
            if self.old_revid == self.new_revid:
2643
 
                to_file.write('No revisions to pull.\n')
2644
 
            else:
2645
 
                to_file.write('Now on revision %d.\n' % self.new_revno)
2646
 
        self._show_tag_conficts(to_file)
2647
 
 
2648
 
 
2649
 
class PushResult(_Result):
2650
 
    """Result of a Branch.push operation.
2651
 
 
2652
 
    :ivar old_revno: Revision number before push.
2653
 
    :ivar new_revno: Revision number after push.
2654
 
    :ivar old_revid: Tip revision id before push.
2655
 
    :ivar new_revid: Tip revision id after push.
2656
 
    :ivar source_branch: Source branch object.
2657
 
    :ivar master_branch: Master branch of the target, or None.
2658
 
    :ivar target_branch: Target/destination branch object.
2659
 
    """
2660
 
 
2661
 
    def __int__(self):
2662
 
        # DEPRECATED: push used to return the change in revno
2663
 
        return self.new_revno - self.old_revno
2664
 
 
2665
 
    def report(self, to_file):
2666
 
        """Write a human-readable description of the result."""
2667
 
        if self.old_revid == self.new_revid:
2668
 
            note('No new revisions to push.')
2669
 
        else:
2670
 
            note('Pushed up to revision %d.' % self.new_revno)
2671
 
        self._show_tag_conficts(to_file)
2672
 
 
2673
 
 
2674
 
class BranchCheckResult(object):
2675
 
    """Results of checking branch consistency.
2676
 
 
2677
 
    :see: Branch.check
2678
 
    """
2679
 
 
2680
 
    def __init__(self, branch):
2681
 
        self.branch = branch
2682
 
 
2683
 
    def report_results(self, verbose):
2684
 
        """Report the check results via trace.note.
2685
 
 
2686
 
        :param verbose: Requests more detailed display of what was checked,
2687
 
            if any.
2688
 
        """
2689
 
        note('checked branch %s format %s',
2690
 
             self.branch.base,
2691
 
             self.branch._format)
2692
 
 
2693
 
 
2694
 
class Converter5to6(object):
2695
 
    """Perform an in-place upgrade of format 5 to format 6"""
2696
 
 
2697
 
    def convert(self, branch):
2698
 
        # Data for 5 and 6 can peacefully coexist.
2699
 
        format = BzrBranchFormat6()
2700
 
        new_branch = format.open(branch.bzrdir, _found=True)
2701
 
 
2702
 
        # Copy source data into target
2703
 
        new_branch._write_last_revision_info(*branch.last_revision_info())
2704
 
        new_branch.set_parent(branch.get_parent())
2705
 
        new_branch.set_bound_location(branch.get_bound_location())
2706
 
        new_branch.set_push_location(branch.get_push_location())
2707
 
 
2708
 
        # New branch has no tags by default
2709
 
        new_branch.tags._set_tag_dict({})
2710
 
 
2711
 
        # Copying done; now update target format
2712
 
        new_branch._transport.put_bytes('format',
2713
 
            format.get_format_string(),
2714
 
            mode=new_branch.bzrdir._get_file_mode())
2715
 
 
2716
 
        # Clean up old files
2717
 
        new_branch._transport.delete('revision-history')
2718
 
        try:
2719
 
            branch.set_parent(None)
2720
 
        except errors.NoSuchFile:
2721
 
            pass
2722
 
        branch.set_bound_location(None)
2723
 
 
2724
 
 
2725
 
class Converter6to7(object):
2726
 
    """Perform an in-place upgrade of format 6 to format 7"""
2727
 
 
2728
 
    def convert(self, branch):
2729
 
        format = BzrBranchFormat7()
2730
 
        branch._set_config_location('stacked_on_location', '')
2731
 
        # update target format
2732
 
        branch._transport.put_bytes('format', format.get_format_string())
2733
 
 
2734
 
 
2735
 
 
2736
 
def _run_with_write_locked_target(target, callable, *args, **kwargs):
2737
 
    """Run ``callable(*args, **kwargs)``, write-locking target for the
2738
 
    duration.
2739
 
 
2740
 
    _run_with_write_locked_target will attempt to release the lock it acquires.
2741
 
 
2742
 
    If an exception is raised by callable, then that exception *will* be
2743
 
    propagated, even if the unlock attempt raises its own error.  Thus
2744
 
    _run_with_write_locked_target should be preferred to simply doing::
2745
 
 
2746
 
        target.lock_write()
2747
 
        try:
2748
 
            return callable(*args, **kwargs)
2749
 
        finally:
2750
 
            target.unlock()
2751
 
 
2752
 
    """
2753
 
    # This is very similar to bzrlib.decorators.needs_write_lock.  Perhaps they
2754
 
    # should share code?
2755
 
    target.lock_write()
2756
 
    try:
2757
 
        result = callable(*args, **kwargs)
2758
 
    except:
2759
 
        exc_info = sys.exc_info()
2760
 
        try:
2761
 
            target.unlock()
2762
 
        finally:
2763
 
            raise exc_info[0], exc_info[1], exc_info[2]
2764
 
    else:
2765
 
        target.unlock()
2766
 
        return result
2767
 
 
2768
 
 
2769
 
class InterBranch(InterObject):
2770
 
    """This class represents operations taking place between two branches.
2771
 
 
2772
 
    Its instances have methods like pull() and push() and contain
2773
 
    references to the source and target repositories these operations
2774
 
    can be carried out on.
2775
 
    """
2776
 
 
2777
 
    _optimisers = []
2778
 
    """The available optimised InterBranch types."""
2779
 
 
2780
 
    @staticmethod
2781
 
    def _get_branch_formats_to_test():
2782
 
        """Return a tuple with the Branch formats to use when testing."""
2783
 
        raise NotImplementedError(self._get_branch_formats_to_test)
2784
 
 
2785
 
    def update_revisions(self, stop_revision=None, overwrite=False,
2786
 
                         graph=None):
2787
 
        """Pull in new perfect-fit revisions.
2788
 
 
2789
 
        :param stop_revision: Updated until the given revision
2790
 
        :param overwrite: Always set the branch pointer, rather than checking
2791
 
            to see if it is a proper descendant.
2792
 
        :param graph: A Graph object that can be used to query history
2793
 
            information. This can be None.
2794
 
        :return: None
2795
 
        """
2796
 
        raise NotImplementedError(self.update_revisions)
2797
 
 
2798
 
 
2799
 
class GenericInterBranch(InterBranch):
2800
 
    """InterBranch implementation that uses public Branch functions.
2801
 
    """
2802
 
 
2803
 
    @staticmethod
2804
 
    def _get_branch_formats_to_test():
2805
 
        return BranchFormat._default_format, BranchFormat._default_format
2806
 
 
2807
 
    def update_revisions(self, stop_revision=None, overwrite=False,
2808
 
        graph=None):
2809
 
        """See InterBranch.update_revisions()."""
2810
 
        self.source.lock_read()
2811
 
        try:
2812
 
            other_revno, other_last_revision = self.source.last_revision_info()
2813
 
            stop_revno = None # unknown
2814
 
            if stop_revision is None:
2815
 
                stop_revision = other_last_revision
2816
 
                if _mod_revision.is_null(stop_revision):
2817
 
                    # if there are no commits, we're done.
2818
 
                    return
2819
 
                stop_revno = other_revno
2820
 
 
2821
 
            # what's the current last revision, before we fetch [and change it
2822
 
            # possibly]
2823
 
            last_rev = _mod_revision.ensure_null(self.target.last_revision())
2824
 
            # we fetch here so that we don't process data twice in the common
2825
 
            # case of having something to pull, and so that the check for
2826
 
            # already merged can operate on the just fetched graph, which will
2827
 
            # be cached in memory.
2828
 
            self.target.fetch(self.source, stop_revision)
2829
 
            # Check to see if one is an ancestor of the other
2830
 
            if not overwrite:
2831
 
                if graph is None:
2832
 
                    graph = self.target.repository.get_graph()
2833
 
                if self.target._check_if_descendant_or_diverged(
2834
 
                        stop_revision, last_rev, graph, self.source):
2835
 
                    # stop_revision is a descendant of last_rev, but we aren't
2836
 
                    # overwriting, so we're done.
2837
 
                    return
2838
 
            if stop_revno is None:
2839
 
                if graph is None:
2840
 
                    graph = self.target.repository.get_graph()
2841
 
                this_revno, this_last_revision = \
2842
 
                        self.target.last_revision_info()
2843
 
                stop_revno = graph.find_distance_to_null(stop_revision,
2844
 
                                [(other_last_revision, other_revno),
2845
 
                                 (this_last_revision, this_revno)])
2846
 
            self.target.set_last_revision_info(stop_revno, stop_revision)
2847
 
        finally:
2848
 
            self.source.unlock()
2849
 
 
2850
 
    @classmethod
2851
 
    def is_compatible(self, source, target):
2852
 
        # GenericBranch uses the public API, so always compatible
2853
 
        return True
2854
 
 
2855
 
 
2856
 
InterBranch.register_optimiser(GenericInterBranch)