~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

merge only needs a lock_tree_write() on the working tree, not a full lock_write()

Show diffs side-by-side

added added

removed removed

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