/brz/remove-bazaar

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

« back to all changes in this revision

Viewing changes to bzrlib/remote.py

  • Committer: A. S. Budden
  • Date: 2011-05-04 13:26:14 UTC
  • mto: (5816.6.16 bazaar_source)
  • mto: This revision was merged to the branch mainline in revision 5835.
  • Revision ID: abudden@gmail.com-20110504132614-3ghb1ajucl6plc5z
Set the parent location of the bound branch to that of the master branch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remote.py
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006-2011 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
19
19
from bzrlib import (
20
20
    bencode,
21
21
    branch,
22
 
    bzrdir,
 
22
    bzrdir as _mod_bzrdir,
23
23
    config,
 
24
    controldir,
24
25
    debug,
25
26
    errors,
26
27
    graph,
27
28
    lock,
28
29
    lockdir,
29
 
    repository,
30
30
    repository as _mod_repository,
31
 
    revision,
32
31
    revision as _mod_revision,
33
32
    static_tuple,
34
33
    symbol_versioning,
 
34
    urlutils,
35
35
)
36
 
from bzrlib.branch import BranchReferenceFormat
37
 
from bzrlib.bzrdir import BzrDir, RemoteBzrDirFormat
 
36
from bzrlib.branch import BranchReferenceFormat, BranchWriteLockResult
38
37
from bzrlib.decorators import needs_read_lock, needs_write_lock, only_raises
39
38
from bzrlib.errors import (
40
39
    NoSuchRevision,
42
41
    )
43
42
from bzrlib.lockable_files import LockableFiles
44
43
from bzrlib.smart import client, vfs, repository as smart_repo
45
 
from bzrlib.revision import ensure_null, NULL_REVISION
 
44
from bzrlib.smart.client import _SmartClient
 
45
from bzrlib.revision import NULL_REVISION
 
46
from bzrlib.repository import RepositoryWriteLockResult
46
47
from bzrlib.trace import mutter, note, warning
47
48
 
48
49
 
86
87
    return format
87
88
 
88
89
 
89
 
# Note: RemoteBzrDirFormat is in bzrdir.py
90
 
 
91
 
class RemoteBzrDir(BzrDir, _RpcHelper):
 
90
# Note that RemoteBzrDirProber lives in bzrlib.bzrdir so bzrlib.remote
 
91
# does not have to be imported unless a remote format is involved.
 
92
 
 
93
class RemoteBzrDirFormat(_mod_bzrdir.BzrDirMetaFormat1):
 
94
    """Format representing bzrdirs accessed via a smart server"""
 
95
 
 
96
    supports_workingtrees = False
 
97
 
 
98
    def __init__(self):
 
99
        _mod_bzrdir.BzrDirMetaFormat1.__init__(self)
 
100
        # XXX: It's a bit ugly that the network name is here, because we'd
 
101
        # like to believe that format objects are stateless or at least
 
102
        # immutable,  However, we do at least avoid mutating the name after
 
103
        # it's returned.  See <https://bugs.launchpad.net/bzr/+bug/504102>
 
104
        self._network_name = None
 
105
 
 
106
    def __repr__(self):
 
107
        return "%s(_network_name=%r)" % (self.__class__.__name__,
 
108
            self._network_name)
 
109
 
 
110
    def get_format_description(self):
 
111
        if self._network_name:
 
112
            real_format = controldir.network_format_registry.get(self._network_name)
 
113
            return 'Remote: ' + real_format.get_format_description()
 
114
        return 'bzr remote bzrdir'
 
115
 
 
116
    def get_format_string(self):
 
117
        raise NotImplementedError(self.get_format_string)
 
118
 
 
119
    def network_name(self):
 
120
        if self._network_name:
 
121
            return self._network_name
 
122
        else:
 
123
            raise AssertionError("No network name set.")
 
124
 
 
125
    def initialize_on_transport(self, transport):
 
126
        try:
 
127
            # hand off the request to the smart server
 
128
            client_medium = transport.get_smart_medium()
 
129
        except errors.NoSmartMedium:
 
130
            # TODO: lookup the local format from a server hint.
 
131
            local_dir_format = _mod_bzrdir.BzrDirMetaFormat1()
 
132
            return local_dir_format.initialize_on_transport(transport)
 
133
        client = _SmartClient(client_medium)
 
134
        path = client.remote_path_from_transport(transport)
 
135
        try:
 
136
            response = client.call('BzrDirFormat.initialize', path)
 
137
        except errors.ErrorFromSmartServer, err:
 
138
            _translate_error(err, path=path)
 
139
        if response[0] != 'ok':
 
140
            raise errors.SmartProtocolError('unexpected response code %s' % (response,))
 
141
        format = RemoteBzrDirFormat()
 
142
        self._supply_sub_formats_to(format)
 
143
        return RemoteBzrDir(transport, format)
 
144
 
 
145
    def parse_NoneTrueFalse(self, arg):
 
146
        if not arg:
 
147
            return None
 
148
        if arg == 'False':
 
149
            return False
 
150
        if arg == 'True':
 
151
            return True
 
152
        raise AssertionError("invalid arg %r" % arg)
 
153
 
 
154
    def _serialize_NoneTrueFalse(self, arg):
 
155
        if arg is False:
 
156
            return 'False'
 
157
        if arg:
 
158
            return 'True'
 
159
        return ''
 
160
 
 
161
    def _serialize_NoneString(self, arg):
 
162
        return arg or ''
 
163
 
 
164
    def initialize_on_transport_ex(self, transport, use_existing_dir=False,
 
165
        create_prefix=False, force_new_repo=False, stacked_on=None,
 
166
        stack_on_pwd=None, repo_format_name=None, make_working_trees=None,
 
167
        shared_repo=False):
 
168
        try:
 
169
            # hand off the request to the smart server
 
170
            client_medium = transport.get_smart_medium()
 
171
        except errors.NoSmartMedium:
 
172
            do_vfs = True
 
173
        else:
 
174
            # Decline to open it if the server doesn't support our required
 
175
            # version (3) so that the VFS-based transport will do it.
 
176
            if client_medium.should_probe():
 
177
                try:
 
178
                    server_version = client_medium.protocol_version()
 
179
                    if server_version != '2':
 
180
                        do_vfs = True
 
181
                    else:
 
182
                        do_vfs = False
 
183
                except errors.SmartProtocolError:
 
184
                    # Apparently there's no usable smart server there, even though
 
185
                    # the medium supports the smart protocol.
 
186
                    do_vfs = True
 
187
            else:
 
188
                do_vfs = False
 
189
        if not do_vfs:
 
190
            client = _SmartClient(client_medium)
 
191
            path = client.remote_path_from_transport(transport)
 
192
            if client_medium._is_remote_before((1, 16)):
 
193
                do_vfs = True
 
194
        if do_vfs:
 
195
            # TODO: lookup the local format from a server hint.
 
196
            local_dir_format = _mod_bzrdir.BzrDirMetaFormat1()
 
197
            self._supply_sub_formats_to(local_dir_format)
 
198
            return local_dir_format.initialize_on_transport_ex(transport,
 
199
                use_existing_dir=use_existing_dir, create_prefix=create_prefix,
 
200
                force_new_repo=force_new_repo, stacked_on=stacked_on,
 
201
                stack_on_pwd=stack_on_pwd, repo_format_name=repo_format_name,
 
202
                make_working_trees=make_working_trees, shared_repo=shared_repo,
 
203
                vfs_only=True)
 
204
        return self._initialize_on_transport_ex_rpc(client, path, transport,
 
205
            use_existing_dir, create_prefix, force_new_repo, stacked_on,
 
206
            stack_on_pwd, repo_format_name, make_working_trees, shared_repo)
 
207
 
 
208
    def _initialize_on_transport_ex_rpc(self, client, path, transport,
 
209
        use_existing_dir, create_prefix, force_new_repo, stacked_on,
 
210
        stack_on_pwd, repo_format_name, make_working_trees, shared_repo):
 
211
        args = []
 
212
        args.append(self._serialize_NoneTrueFalse(use_existing_dir))
 
213
        args.append(self._serialize_NoneTrueFalse(create_prefix))
 
214
        args.append(self._serialize_NoneTrueFalse(force_new_repo))
 
215
        args.append(self._serialize_NoneString(stacked_on))
 
216
        # stack_on_pwd is often/usually our transport
 
217
        if stack_on_pwd:
 
218
            try:
 
219
                stack_on_pwd = transport.relpath(stack_on_pwd)
 
220
                if not stack_on_pwd:
 
221
                    stack_on_pwd = '.'
 
222
            except errors.PathNotChild:
 
223
                pass
 
224
        args.append(self._serialize_NoneString(stack_on_pwd))
 
225
        args.append(self._serialize_NoneString(repo_format_name))
 
226
        args.append(self._serialize_NoneTrueFalse(make_working_trees))
 
227
        args.append(self._serialize_NoneTrueFalse(shared_repo))
 
228
        request_network_name = self._network_name or \
 
229
            _mod_bzrdir.BzrDirFormat.get_default_format().network_name()
 
230
        try:
 
231
            response = client.call('BzrDirFormat.initialize_ex_1.16',
 
232
                request_network_name, path, *args)
 
233
        except errors.UnknownSmartMethod:
 
234
            client._medium._remember_remote_is_before((1,16))
 
235
            local_dir_format = _mod_bzrdir.BzrDirMetaFormat1()
 
236
            self._supply_sub_formats_to(local_dir_format)
 
237
            return local_dir_format.initialize_on_transport_ex(transport,
 
238
                use_existing_dir=use_existing_dir, create_prefix=create_prefix,
 
239
                force_new_repo=force_new_repo, stacked_on=stacked_on,
 
240
                stack_on_pwd=stack_on_pwd, repo_format_name=repo_format_name,
 
241
                make_working_trees=make_working_trees, shared_repo=shared_repo,
 
242
                vfs_only=True)
 
243
        except errors.ErrorFromSmartServer, err:
 
244
            _translate_error(err, path=path)
 
245
        repo_path = response[0]
 
246
        bzrdir_name = response[6]
 
247
        require_stacking = response[7]
 
248
        require_stacking = self.parse_NoneTrueFalse(require_stacking)
 
249
        format = RemoteBzrDirFormat()
 
250
        format._network_name = bzrdir_name
 
251
        self._supply_sub_formats_to(format)
 
252
        bzrdir = RemoteBzrDir(transport, format, _client=client)
 
253
        if repo_path:
 
254
            repo_format = response_tuple_to_repo_format(response[1:])
 
255
            if repo_path == '.':
 
256
                repo_path = ''
 
257
            if repo_path:
 
258
                repo_bzrdir_format = RemoteBzrDirFormat()
 
259
                repo_bzrdir_format._network_name = response[5]
 
260
                repo_bzr = RemoteBzrDir(transport.clone(repo_path),
 
261
                    repo_bzrdir_format)
 
262
            else:
 
263
                repo_bzr = bzrdir
 
264
            final_stack = response[8] or None
 
265
            final_stack_pwd = response[9] or None
 
266
            if final_stack_pwd:
 
267
                final_stack_pwd = urlutils.join(
 
268
                    transport.base, final_stack_pwd)
 
269
            remote_repo = RemoteRepository(repo_bzr, repo_format)
 
270
            if len(response) > 10:
 
271
                # Updated server verb that locks remotely.
 
272
                repo_lock_token = response[10] or None
 
273
                remote_repo.lock_write(repo_lock_token, _skip_rpc=True)
 
274
                if repo_lock_token:
 
275
                    remote_repo.dont_leave_lock_in_place()
 
276
            else:
 
277
                remote_repo.lock_write()
 
278
            policy = _mod_bzrdir.UseExistingRepository(remote_repo, final_stack,
 
279
                final_stack_pwd, require_stacking)
 
280
            policy.acquire_repository()
 
281
        else:
 
282
            remote_repo = None
 
283
            policy = None
 
284
        bzrdir._format.set_branch_format(self.get_branch_format())
 
285
        if require_stacking:
 
286
            # The repo has already been created, but we need to make sure that
 
287
            # we'll make a stackable branch.
 
288
            bzrdir._format.require_stacking(_skip_repo=True)
 
289
        return remote_repo, bzrdir, require_stacking, policy
 
290
 
 
291
    def _open(self, transport):
 
292
        return RemoteBzrDir(transport, self)
 
293
 
 
294
    def __eq__(self, other):
 
295
        if not isinstance(other, RemoteBzrDirFormat):
 
296
            return False
 
297
        return self.get_format_description() == other.get_format_description()
 
298
 
 
299
    def __return_repository_format(self):
 
300
        # Always return a RemoteRepositoryFormat object, but if a specific bzr
 
301
        # repository format has been asked for, tell the RemoteRepositoryFormat
 
302
        # that it should use that for init() etc.
 
303
        result = RemoteRepositoryFormat()
 
304
        custom_format = getattr(self, '_repository_format', None)
 
305
        if custom_format:
 
306
            if isinstance(custom_format, RemoteRepositoryFormat):
 
307
                return custom_format
 
308
            else:
 
309
                # We will use the custom format to create repositories over the
 
310
                # wire; expose its details like rich_root_data for code to
 
311
                # query
 
312
                result._custom_format = custom_format
 
313
        return result
 
314
 
 
315
    def get_branch_format(self):
 
316
        result = _mod_bzrdir.BzrDirMetaFormat1.get_branch_format(self)
 
317
        if not isinstance(result, RemoteBranchFormat):
 
318
            new_result = RemoteBranchFormat()
 
319
            new_result._custom_format = result
 
320
            # cache the result
 
321
            self.set_branch_format(new_result)
 
322
            result = new_result
 
323
        return result
 
324
 
 
325
    repository_format = property(__return_repository_format,
 
326
        _mod_bzrdir.BzrDirMetaFormat1._set_repository_format) #.im_func)
 
327
 
 
328
 
 
329
class RemoteBzrDir(_mod_bzrdir.BzrDir, _RpcHelper):
92
330
    """Control directory on a remote server, accessed via bzr:// or similar."""
93
331
 
94
332
    def __init__(self, transport, format, _client=None, _force_probe=False):
97
335
        :param _client: Private parameter for testing. Disables probing and the
98
336
            use of a real bzrdir.
99
337
        """
100
 
        BzrDir.__init__(self, transport, format)
 
338
        _mod_bzrdir.BzrDir.__init__(self, transport, format)
101
339
        # this object holds a delegated bzrdir that uses file-level operations
102
340
        # to talk to the other side
103
341
        self._real_bzrdir = None
163
401
                import traceback
164
402
                warning('VFS BzrDir access triggered\n%s',
165
403
                    ''.join(traceback.format_stack()))
166
 
            self._real_bzrdir = BzrDir.open_from_transport(
 
404
            self._real_bzrdir = _mod_bzrdir.BzrDir.open_from_transport(
167
405
                self.root_transport, _server_formats=False)
168
406
            self._format._network_name = \
169
407
                self._real_bzrdir._format.network_name()
175
413
        # Prevent aliasing problems in the next_open_branch_result cache.
176
414
        # See create_branch for rationale.
177
415
        self._next_open_branch_result = None
178
 
        return BzrDir.break_lock(self)
 
416
        return _mod_bzrdir.BzrDir.break_lock(self)
179
417
 
180
418
    def _vfs_cloning_metadir(self, require_stacking=False):
181
419
        self._ensure_real()
212
450
        if len(branch_info) != 2:
213
451
            raise errors.UnexpectedSmartServerResponse(response)
214
452
        branch_ref, branch_name = branch_info
215
 
        format = bzrdir.network_format_registry.get(control_name)
 
453
        format = controldir.network_format_registry.get(control_name)
216
454
        if repo_name:
217
 
            format.repository_format = repository.network_format_registry.get(
 
455
            format.repository_format = _mod_repository.network_format_registry.get(
218
456
                repo_name)
219
457
        if branch_ref == 'ref':
220
458
            # XXX: we need possible_transports here to avoid reopening the
221
459
            # connection to the referenced location
222
 
            ref_bzrdir = BzrDir.open(branch_name)
 
460
            ref_bzrdir = _mod_bzrdir.BzrDir.open(branch_name)
223
461
            branch_format = ref_bzrdir.cloning_metadir().get_branch_format()
224
462
            format.set_branch_format(branch_format)
225
463
        elif branch_ref == 'branch':
244
482
        self._ensure_real()
245
483
        self._real_bzrdir.destroy_repository()
246
484
 
247
 
    def create_branch(self, name=None):
 
485
    def create_branch(self, name=None, repository=None):
248
486
        # as per meta1 formats - just delegate to the format object which may
249
487
        # be parameterised.
250
488
        real_branch = self._format.get_branch_format().initialize(self,
251
 
            name=name)
 
489
            name=name, repository=repository)
252
490
        if not isinstance(real_branch, RemoteBranch):
253
 
            result = RemoteBranch(self, self.find_repository(), real_branch,
254
 
                                  name=name)
 
491
            if not isinstance(repository, RemoteRepository):
 
492
                raise AssertionError(
 
493
                    'need a RemoteRepository to use with RemoteBranch, got %r'
 
494
                    % (repository,))
 
495
            result = RemoteBranch(self, repository, real_branch, name=name)
255
496
        else:
256
497
            result = real_branch
257
498
        # BzrDir.clone_on_transport() uses the result of create_branch but does
269
510
        self._real_bzrdir.destroy_branch(name=name)
270
511
        self._next_open_branch_result = None
271
512
 
272
 
    def create_workingtree(self, revision_id=None, from_branch=None):
 
513
    def create_workingtree(self, revision_id=None, from_branch=None,
 
514
        accelerator_tree=None, hardlink=False):
273
515
        raise errors.NotLocalUrl(self.transport.base)
274
516
 
275
 
    def find_branch_format(self):
 
517
    def find_branch_format(self, name=None):
276
518
        """Find the branch 'format' for this bzrdir.
277
519
 
278
520
        This might be a synthetic object for e.g. RemoteBranch and SVN.
279
521
        """
280
 
        b = self.open_branch()
 
522
        b = self.open_branch(name=name)
281
523
        return b._format
282
524
 
283
 
    def get_branch_reference(self):
 
525
    def get_branch_reference(self, name=None):
284
526
        """See BzrDir.get_branch_reference()."""
 
527
        if name is not None:
 
528
            # XXX JRV20100304: Support opening colocated branches
 
529
            raise errors.NoColocatedBranchSupport(self)
285
530
        response = self._get_branch_reference()
286
531
        if response[0] == 'ref':
287
532
            return response[1]
318
563
            raise errors.UnexpectedSmartServerResponse(response)
319
564
        return response
320
565
 
321
 
    def _get_tree_branch(self):
 
566
    def _get_tree_branch(self, name=None):
322
567
        """See BzrDir._get_tree_branch()."""
323
 
        return None, self.open_branch()
 
568
        return None, self.open_branch(name=name)
324
569
 
325
570
    def open_branch(self, name=None, unsupported=False,
326
571
                    ignore_fallbacks=False):
441
686
        """Upgrading of remote bzrdirs is not supported yet."""
442
687
        return False
443
688
 
444
 
    def needs_format_conversion(self, format=None):
 
689
    def needs_format_conversion(self, format):
445
690
        """Upgrading of remote bzrdirs is not supported yet."""
446
 
        if format is None:
447
 
            symbol_versioning.warn(symbol_versioning.deprecated_in((1, 13, 0))
448
 
                % 'needs_format_conversion(format=None)')
449
691
        return False
450
692
 
451
693
    def clone(self, url, revision_id=None, force_new_repo=False,
458
700
        return RemoteBzrDirConfig(self)
459
701
 
460
702
 
461
 
class RemoteRepositoryFormat(repository.RepositoryFormat):
 
703
class RemoteRepositoryFormat(_mod_repository.RepositoryFormat):
462
704
    """Format for repositories accessed over a _SmartClient.
463
705
 
464
706
    Instances of this repository are represented by RemoteRepository
479
721
    """
480
722
 
481
723
    _matchingbzrdir = RemoteBzrDirFormat()
 
724
    supports_full_versioned_files = True
 
725
    supports_leaving_lock = True
482
726
 
483
727
    def __init__(self):
484
 
        repository.RepositoryFormat.__init__(self)
 
728
        _mod_repository.RepositoryFormat.__init__(self)
485
729
        self._custom_format = None
486
730
        self._network_name = None
487
731
        self._creating_bzrdir = None
 
732
        self._revision_graph_can_have_wrong_parents = None
488
733
        self._supports_chks = None
489
734
        self._supports_external_lookups = None
490
735
        self._supports_tree_reference = None
 
736
        self._supports_funky_characters = None
491
737
        self._rich_root_data = None
492
738
 
493
739
    def __repr__(self):
522
768
        return self._supports_external_lookups
523
769
 
524
770
    @property
 
771
    def supports_funky_characters(self):
 
772
        if self._supports_funky_characters is None:
 
773
            self._ensure_real()
 
774
            self._supports_funky_characters = \
 
775
                self._custom_format.supports_funky_characters
 
776
        return self._supports_funky_characters
 
777
 
 
778
    @property
525
779
    def supports_tree_reference(self):
526
780
        if self._supports_tree_reference is None:
527
781
            self._ensure_real()
529
783
                self._custom_format.supports_tree_reference
530
784
        return self._supports_tree_reference
531
785
 
 
786
    @property
 
787
    def revision_graph_can_have_wrong_parents(self):
 
788
        if self._revision_graph_can_have_wrong_parents is None:
 
789
            self._ensure_real()
 
790
            self._revision_graph_can_have_wrong_parents = \
 
791
                self._custom_format.revision_graph_can_have_wrong_parents
 
792
        return self._revision_graph_can_have_wrong_parents
 
793
 
532
794
    def _vfs_initialize(self, a_bzrdir, shared):
533
795
        """Helper for common code in initialize."""
534
796
        if self._custom_format:
569
831
            network_name = self._network_name
570
832
        else:
571
833
            # Select the current bzrlib default and ask for that.
572
 
            reference_bzrdir_format = bzrdir.format_registry.get('default')()
 
834
            reference_bzrdir_format = _mod_bzrdir.format_registry.get('default')()
573
835
            reference_format = reference_bzrdir_format.repository_format
574
836
            network_name = reference_format.network_name()
575
837
        # 2) try direct creation via RPC
601
863
 
602
864
    def _ensure_real(self):
603
865
        if self._custom_format is None:
604
 
            self._custom_format = repository.network_format_registry.get(
 
866
            self._custom_format = _mod_repository.network_format_registry.get(
605
867
                self._network_name)
606
868
 
607
869
    @property
644
906
 
645
907
 
646
908
class RemoteRepository(_RpcHelper, lock._RelockDebugMixin,
647
 
    bzrdir.ControlComponent):
 
909
    controldir.ControlComponent):
648
910
    """Repository accessed over rpc.
649
911
 
650
912
    For the moment most operations are performed using local transport-backed
703
965
        # transport, but I'm not sure it's worth making this method
704
966
        # optional -- mbp 2010-04-21
705
967
        return self.bzrdir.get_repository_transport(None)
706
 
        
 
968
 
707
969
    def __str__(self):
708
970
        return "%s(%s)" % (self.__class__.__name__, self.base)
709
971
 
843
1105
        """Private method for using with old (< 1.2) servers to fallback."""
844
1106
        if revision_id is None:
845
1107
            revision_id = ''
846
 
        elif revision.is_null(revision_id):
 
1108
        elif _mod_revision.is_null(revision_id):
847
1109
            return {}
848
1110
 
849
1111
        path = self.bzrdir._path_for_remote_call(self._client)
895
1157
    def _has_same_fallbacks(self, other_repo):
896
1158
        """Returns true if the repositories have the same fallbacks."""
897
1159
        # XXX: copied from Repository; it should be unified into a base class
898
 
        # <https://bugs.edge.launchpad.net/bzr/+bug/401622>
 
1160
        # <https://bugs.launchpad.net/bzr/+bug/401622>
899
1161
        my_fb = self._fallback_repositories
900
1162
        other_fb = other_repo._fallback_repositories
901
1163
        if len(my_fb) != len(other_fb):
930
1192
        """See Repository.gather_stats()."""
931
1193
        path = self.bzrdir._path_for_remote_call(self._client)
932
1194
        # revid can be None to indicate no revisions, not just NULL_REVISION
933
 
        if revid is None or revision.is_null(revid):
 
1195
        if revid is None or _mod_revision.is_null(revid):
934
1196
            fmt_revid = ''
935
1197
        else:
936
1198
            fmt_revid = revid
997
1259
        pass
998
1260
 
999
1261
    def lock_read(self):
 
1262
        """Lock the repository for read operations.
 
1263
 
 
1264
        :return: A bzrlib.lock.LogicalLockResult.
 
1265
        """
1000
1266
        # wrong eventually - want a local lock cache context
1001
1267
        if not self._lock_mode:
1002
1268
            self._note_lock('r')
1009
1275
                repo.lock_read()
1010
1276
        else:
1011
1277
            self._lock_count += 1
 
1278
        return lock.LogicalLockResult(self.unlock)
1012
1279
 
1013
1280
    def _remote_lock_write(self, token):
1014
1281
        path = self.bzrdir._path_for_remote_call(self._client)
1054
1321
            raise errors.ReadOnlyError(self)
1055
1322
        else:
1056
1323
            self._lock_count += 1
1057
 
        return self._lock_token or None
 
1324
        return RepositoryWriteLockResult(self.unlock, self._lock_token or None)
1058
1325
 
1059
1326
    def leave_lock_in_place(self):
1060
1327
        if not self._lock_token:
1214
1481
 
1215
1482
    def get_commit_builder(self, branch, parents, config, timestamp=None,
1216
1483
                           timezone=None, committer=None, revprops=None,
1217
 
                           revision_id=None):
 
1484
                           revision_id=None, lossy=False):
1218
1485
        # FIXME: It ought to be possible to call this without immediately
1219
1486
        # triggering _ensure_real.  For now it's the easiest thing to do.
1220
1487
        self._ensure_real()
1221
1488
        real_repo = self._real_repository
1222
1489
        builder = real_repo.get_commit_builder(branch, parents,
1223
1490
                config, timestamp=timestamp, timezone=timezone,
1224
 
                committer=committer, revprops=revprops, revision_id=revision_id)
 
1491
                committer=committer, revprops=revprops,
 
1492
                revision_id=revision_id, lossy=lossy)
1225
1493
        return builder
1226
1494
 
1227
1495
    def add_fallback_repository(self, repository):
1306
1574
        return self._real_repository.make_working_trees()
1307
1575
 
1308
1576
    def refresh_data(self):
1309
 
        """Re-read any data needed to to synchronise with disk.
 
1577
        """Re-read any data needed to synchronise with disk.
1310
1578
 
1311
1579
        This method is intended to be called after another repository instance
1312
1580
        (such as one used by a smart server) has inserted data into the
1313
 
        repository. It may not be called during a write group, but may be
1314
 
        called at any other time.
 
1581
        repository. On all repositories this will work outside of write groups.
 
1582
        Some repository formats (pack and newer for bzrlib native formats)
 
1583
        support refresh_data inside write groups. If called inside a write
 
1584
        group on a repository that does not support refreshing in a write group
 
1585
        IsInWriteGroupError will be raised.
1315
1586
        """
1316
 
        if self.is_in_write_group():
1317
 
            raise errors.InternalBzrError(
1318
 
                "May not refresh_data while in a write group.")
1319
1587
        if self._real_repository is not None:
1320
1588
            self._real_repository.refresh_data()
1321
1589
 
1333
1601
        return result
1334
1602
 
1335
1603
    @needs_read_lock
1336
 
    def search_missing_revision_ids(self, other, revision_id=None, find_ghosts=True):
 
1604
    def search_missing_revision_ids(self, other,
 
1605
            revision_id=symbol_versioning.DEPRECATED_PARAMETER,
 
1606
            find_ghosts=True, revision_ids=None, if_present_ids=None):
1337
1607
        """Return the revision ids that other has that this does not.
1338
1608
 
1339
1609
        These are returned in topological order.
1340
1610
 
1341
1611
        revision_id: only return revision ids included by revision_id.
1342
1612
        """
1343
 
        return repository.InterRepository.get(
1344
 
            other, self).search_missing_revision_ids(revision_id, find_ghosts)
 
1613
        if symbol_versioning.deprecated_passed(revision_id):
 
1614
            symbol_versioning.warn(
 
1615
                'search_missing_revision_ids(revision_id=...) was '
 
1616
                'deprecated in 2.4.  Use revision_ids=[...] instead.',
 
1617
                DeprecationWarning, stacklevel=2)
 
1618
            if revision_ids is not None:
 
1619
                raise AssertionError(
 
1620
                    'revision_ids is mutually exclusive with revision_id')
 
1621
            if revision_id is not None:
 
1622
                revision_ids = [revision_id]
 
1623
        inter_repo = _mod_repository.InterRepository.get(other, self)
 
1624
        return inter_repo.search_missing_revision_ids(
 
1625
            find_ghosts=find_ghosts, revision_ids=revision_ids,
 
1626
            if_present_ids=if_present_ids)
1345
1627
 
1346
 
    def fetch(self, source, revision_id=None, pb=None, find_ghosts=False,
 
1628
    def fetch(self, source, revision_id=None, find_ghosts=False,
1347
1629
            fetch_spec=None):
1348
1630
        # No base implementation to use as RemoteRepository is not a subclass
1349
1631
        # of Repository; so this is a copy of Repository.fetch().
1360
1642
            # check that last_revision is in 'from' and then return a
1361
1643
            # no-operation.
1362
1644
            if (revision_id is not None and
1363
 
                not revision.is_null(revision_id)):
 
1645
                not _mod_revision.is_null(revision_id)):
1364
1646
                self.get_revision(revision_id)
1365
1647
            return 0, []
1366
1648
        # if there is no specific appropriate InterRepository, this will get
1367
1649
        # the InterRepository base class, which raises an
1368
1650
        # IncompatibleRepositories when asked to fetch.
1369
 
        inter = repository.InterRepository.get(source, self)
1370
 
        return inter.fetch(revision_id=revision_id, pb=pb,
 
1651
        inter = _mod_repository.InterRepository.get(source, self)
 
1652
        return inter.fetch(revision_id=revision_id,
1371
1653
            find_ghosts=find_ghosts, fetch_spec=fetch_spec)
1372
1654
 
1373
1655
    def create_bundle(self, target, base, fileobj, format=None):
1597
1879
            tmpdir = osutils.mkdtemp()
1598
1880
            try:
1599
1881
                _extract_tar(tar, tmpdir)
1600
 
                tmp_bzrdir = BzrDir.open(tmpdir)
 
1882
                tmp_bzrdir = _mod_bzrdir.BzrDir.open(tmpdir)
1601
1883
                tmp_repo = tmp_bzrdir.open_repository()
1602
1884
                tmp_repo.copy_content_into(destination, revision_id)
1603
1885
            finally:
1714
1996
        return self._real_repository.item_keys_introduced_by(revision_ids,
1715
1997
            _files_pb=_files_pb)
1716
1998
 
1717
 
    def revision_graph_can_have_wrong_parents(self):
1718
 
        # The answer depends on the remote repo format.
1719
 
        self._ensure_real()
1720
 
        return self._real_repository.revision_graph_can_have_wrong_parents()
1721
 
 
1722
1999
    def _find_inconsistent_revision_parents(self, revisions_iterator=None):
1723
2000
        self._ensure_real()
1724
2001
        return self._real_repository._find_inconsistent_revision_parents(
1748
2025
        return '\n'.join((start_keys, stop_keys, count))
1749
2026
 
1750
2027
    def _serialise_search_result(self, search_result):
1751
 
        if isinstance(search_result, graph.PendingAncestryResult):
1752
 
            parts = ['ancestry-of']
1753
 
            parts.extend(search_result.heads)
1754
 
        else:
1755
 
            recipe = search_result.get_recipe()
1756
 
            parts = [recipe[0], self._serialise_search_recipe(recipe)]
 
2028
        parts = search_result.get_network_struct()
1757
2029
        return '\n'.join(parts)
1758
2030
 
1759
2031
    def autopack(self):
1769
2041
            raise errors.UnexpectedSmartServerResponse(response)
1770
2042
 
1771
2043
 
1772
 
class RemoteStreamSink(repository.StreamSink):
 
2044
class RemoteStreamSink(_mod_repository.StreamSink):
1773
2045
 
1774
2046
    def _insert_real(self, stream, src_format, resume_tokens):
1775
2047
        self.target_repo._ensure_real()
1876
2148
        self._last_substream and self._last_stream so that the stream can be
1877
2149
        resumed by _resume_stream_with_vfs.
1878
2150
        """
1879
 
                    
 
2151
 
1880
2152
        stream_iter = iter(stream)
1881
2153
        for substream_kind, substream in stream_iter:
1882
2154
            if substream_kind == 'inventory-deltas':
1885
2157
                return
1886
2158
            else:
1887
2159
                yield substream_kind, substream
1888
 
            
1889
 
 
1890
 
class RemoteStreamSource(repository.StreamSource):
 
2160
 
 
2161
 
 
2162
class RemoteStreamSource(_mod_repository.StreamSource):
1891
2163
    """Stream data from a remote server."""
1892
2164
 
1893
2165
    def get_stream(self, search):
1953
2225
        candidate_verbs = [
1954
2226
            ('Repository.get_stream_1.19', (1, 19)),
1955
2227
            ('Repository.get_stream', (1, 13))]
 
2228
 
1956
2229
        found_verb = False
1957
2230
        for verb, version in candidate_verbs:
1958
2231
            if medium._is_remote_before(version):
1962
2235
                    verb, args, search_bytes)
1963
2236
            except errors.UnknownSmartMethod:
1964
2237
                medium._remember_remote_is_before(version)
 
2238
            except errors.UnknownErrorFromSmartServer, e:
 
2239
                if isinstance(search, graph.EverythingResult):
 
2240
                    error_verb = e.error_from_smart_server.error_verb
 
2241
                    if error_verb == 'BadSearch':
 
2242
                        # Pre-2.4 servers don't support this sort of search.
 
2243
                        # XXX: perhaps falling back to VFS on BadSearch is a
 
2244
                        # good idea in general?  It might provide a little bit
 
2245
                        # of protection against client-side bugs.
 
2246
                        medium._remember_remote_is_before((2, 4))
 
2247
                        break
 
2248
                raise
1965
2249
            else:
1966
2250
                response_tuple, response_handler = response
1967
2251
                found_verb = True
1971
2255
        if response_tuple[0] != 'ok':
1972
2256
            raise errors.UnexpectedSmartServerResponse(response_tuple)
1973
2257
        byte_stream = response_handler.read_streamed_body()
1974
 
        src_format, stream = smart_repo._byte_stream_to_stream(byte_stream)
 
2258
        src_format, stream = smart_repo._byte_stream_to_stream(byte_stream,
 
2259
            self._record_counter)
1975
2260
        if src_format.network_name() != repo._format.network_name():
1976
2261
            raise AssertionError(
1977
2262
                "Mismatched RemoteRepository and stream src %r, %r" % (
2081
2366
                                  name=name)
2082
2367
        return result
2083
2368
 
2084
 
    def initialize(self, a_bzrdir, name=None):
 
2369
    def initialize(self, a_bzrdir, name=None, repository=None):
2085
2370
        # 1) get the network name to use.
2086
2371
        if self._custom_format:
2087
2372
            network_name = self._custom_format.network_name()
2088
2373
        else:
2089
2374
            # Select the current bzrlib default and ask for that.
2090
 
            reference_bzrdir_format = bzrdir.format_registry.get('default')()
 
2375
            reference_bzrdir_format = _mod_bzrdir.format_registry.get('default')()
2091
2376
            reference_format = reference_bzrdir_format.get_branch_format()
2092
2377
            self._custom_format = reference_format
2093
2378
            network_name = reference_format.network_name()
2115
2400
        # Turn the response into a RemoteRepository object.
2116
2401
        format = RemoteBranchFormat(network_name=response[1])
2117
2402
        repo_format = response_tuple_to_repo_format(response[3:])
2118
 
        if response[2] == '':
2119
 
            repo_bzrdir = a_bzrdir
 
2403
        repo_path = response[2]
 
2404
        if repository is not None:
 
2405
            remote_repo_url = urlutils.join(a_bzrdir.user_url, repo_path)
 
2406
            url_diff = urlutils.relative_url(repository.user_url,
 
2407
                    remote_repo_url)
 
2408
            if url_diff != '.':
 
2409
                raise AssertionError(
 
2410
                    'repository.user_url %r does not match URL from server '
 
2411
                    'response (%r + %r)'
 
2412
                    % (repository.user_url, a_bzrdir.user_url, repo_path))
 
2413
            remote_repo = repository
2120
2414
        else:
2121
 
            repo_bzrdir = RemoteBzrDir(
2122
 
                a_bzrdir.root_transport.clone(response[2]), a_bzrdir._format,
2123
 
                a_bzrdir._client)
2124
 
        remote_repo = RemoteRepository(repo_bzrdir, repo_format)
 
2415
            if repo_path == '':
 
2416
                repo_bzrdir = a_bzrdir
 
2417
            else:
 
2418
                repo_bzrdir = RemoteBzrDir(
 
2419
                    a_bzrdir.root_transport.clone(repo_path), a_bzrdir._format,
 
2420
                    a_bzrdir._client)
 
2421
            remote_repo = RemoteRepository(repo_bzrdir, repo_format)
2125
2422
        remote_branch = RemoteBranch(a_bzrdir, remote_repo,
2126
2423
            format=format, setup_stacking=False, name=name)
2127
2424
        # XXX: We know this is a new branch, so it must have revno 0, revid
2148
2445
        self._ensure_real()
2149
2446
        return self._custom_format.supports_set_append_revisions_only()
2150
2447
 
 
2448
    def _use_default_local_heads_to_fetch(self):
 
2449
        # If the branch format is a metadir format *and* its heads_to_fetch
 
2450
        # implementation is not overridden vs the base class, we can use the
 
2451
        # base class logic rather than use the heads_to_fetch RPC.  This is
 
2452
        # usually cheaper in terms of net round trips, as the last-revision and
 
2453
        # tags info fetched is cached and would be fetched anyway.
 
2454
        self._ensure_real()
 
2455
        if isinstance(self._custom_format, branch.BranchFormatMetadir):
 
2456
            branch_class = self._custom_format._branch_class()
 
2457
            heads_to_fetch_impl = branch_class.heads_to_fetch.im_func
 
2458
            if heads_to_fetch_impl is branch.Branch.heads_to_fetch.im_func:
 
2459
                return True
 
2460
        return False
2151
2461
 
2152
2462
class RemoteBranch(branch.Branch, _RpcHelper, lock._RelockDebugMixin):
2153
2463
    """Branch stored on a server accessed by HPSS RPC.
2352
2662
            self._is_stacked = False
2353
2663
        else:
2354
2664
            self._is_stacked = True
2355
 
        
 
2665
 
2356
2666
    def _vfs_get_tags_bytes(self):
2357
2667
        self._ensure_real()
2358
2668
        return self._real_branch._get_tags_bytes()
2359
2669
 
 
2670
    @needs_read_lock
2360
2671
    def _get_tags_bytes(self):
 
2672
        if self._tags_bytes is None:
 
2673
            self._tags_bytes = self._get_tags_bytes_via_hpss()
 
2674
        return self._tags_bytes
 
2675
 
 
2676
    def _get_tags_bytes_via_hpss(self):
2361
2677
        medium = self._client._medium
2362
2678
        if medium._is_remote_before((1, 13)):
2363
2679
            return self._vfs_get_tags_bytes()
2373
2689
        return self._real_branch._set_tags_bytes(bytes)
2374
2690
 
2375
2691
    def _set_tags_bytes(self, bytes):
 
2692
        if self.is_locked():
 
2693
            self._tags_bytes = bytes
2376
2694
        medium = self._client._medium
2377
2695
        if medium._is_remote_before((1, 18)):
2378
2696
            self._vfs_set_tags_bytes(bytes)
2387
2705
            self._vfs_set_tags_bytes(bytes)
2388
2706
 
2389
2707
    def lock_read(self):
 
2708
        """Lock the branch for read operations.
 
2709
 
 
2710
        :return: A bzrlib.lock.LogicalLockResult.
 
2711
        """
2390
2712
        self.repository.lock_read()
2391
2713
        if not self._lock_mode:
2392
2714
            self._note_lock('r')
2396
2718
                self._real_branch.lock_read()
2397
2719
        else:
2398
2720
            self._lock_count += 1
 
2721
        return lock.LogicalLockResult(self.unlock)
2399
2722
 
2400
2723
    def _remote_lock_write(self, token):
2401
2724
        if token is None:
2402
2725
            branch_token = repo_token = ''
2403
2726
        else:
2404
2727
            branch_token = token
2405
 
            repo_token = self.repository.lock_write()
 
2728
            repo_token = self.repository.lock_write().repository_token
2406
2729
            self.repository.unlock()
2407
2730
        err_context = {'token': token}
2408
 
        response = self._call(
2409
 
            'Branch.lock_write', self._remote_path(), branch_token,
2410
 
            repo_token or '', **err_context)
 
2731
        try:
 
2732
            response = self._call(
 
2733
                'Branch.lock_write', self._remote_path(), branch_token,
 
2734
                repo_token or '', **err_context)
 
2735
        except errors.LockContention, e:
 
2736
            # The LockContention from the server doesn't have any
 
2737
            # information about the lock_url. We re-raise LockContention
 
2738
            # with valid lock_url.
 
2739
            raise errors.LockContention('(remote lock)',
 
2740
                self.repository.base.split('.bzr/')[0])
2411
2741
        if response[0] != 'ok':
2412
2742
            raise errors.UnexpectedSmartServerResponse(response)
2413
2743
        ok, branch_token, repo_token = response
2434
2764
            self._lock_mode = 'w'
2435
2765
            self._lock_count = 1
2436
2766
        elif self._lock_mode == 'r':
2437
 
            raise errors.ReadOnlyTransaction
 
2767
            raise errors.ReadOnlyError(self)
2438
2768
        else:
2439
2769
            if token is not None:
2440
2770
                # A token was given to lock_write, and we're relocking, so
2445
2775
            self._lock_count += 1
2446
2776
            # Re-lock the repository too.
2447
2777
            self.repository.lock_write(self._repo_lock_token)
2448
 
        return self._lock_token or None
 
2778
        return BranchWriteLockResult(self.unlock, self._lock_token or None)
2449
2779
 
2450
2780
    def _unlock(self, branch_token, repo_token):
2451
2781
        err_context = {'token': str((branch_token, repo_token))}
2589
2919
            raise errors.UnexpectedSmartServerResponse(response)
2590
2920
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2591
2921
 
 
2922
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((2, 4, 0)))
2592
2923
    @needs_write_lock
2593
2924
    def set_revision_history(self, rev_history):
 
2925
        """See Branch.set_revision_history."""
 
2926
        self._set_revision_history(rev_history)
 
2927
 
 
2928
    @needs_write_lock
 
2929
    def _set_revision_history(self, rev_history):
2594
2930
        # Send just the tip revision of the history; the server will generate
2595
2931
        # the full history from that.  If the revision doesn't exist in this
2596
2932
        # branch, NoSuchRevision will be raised.
2673
3009
        # XXX: These should be returned by the set_last_revision_info verb
2674
3010
        old_revno, old_revid = self.last_revision_info()
2675
3011
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
2676
 
        revision_id = ensure_null(revision_id)
 
3012
        if not revision_id or not isinstance(revision_id, basestring):
 
3013
            raise errors.InvalidRevisionId(revision_id=revision_id, branch=self)
2677
3014
        try:
2678
3015
            response = self._call('Branch.set_last_revision_info',
2679
3016
                self._remote_path(), self._lock_token, self._repo_lock_token,
2708
3045
            except errors.UnknownSmartMethod:
2709
3046
                medium._remember_remote_is_before((1, 6))
2710
3047
        self._clear_cached_state_of_remote_branch_only()
2711
 
        self.set_revision_history(self._lefthand_history(revision_id,
 
3048
        self._set_revision_history(self._lefthand_history(revision_id,
2712
3049
            last_rev=last_rev,other_branch=other_branch))
2713
3050
 
2714
3051
    def set_push_location(self, location):
2715
3052
        self._ensure_real()
2716
3053
        return self._real_branch.set_push_location(location)
2717
3054
 
 
3055
    def heads_to_fetch(self):
 
3056
        if self._format._use_default_local_heads_to_fetch():
 
3057
            # We recognise this format, and its heads-to-fetch implementation
 
3058
            # is the default one (tip + tags).  In this case it's cheaper to
 
3059
            # just use the default implementation rather than a special RPC as
 
3060
            # the tip and tags data is cached.
 
3061
            return branch.Branch.heads_to_fetch(self)
 
3062
        medium = self._client._medium
 
3063
        if medium._is_remote_before((2, 4)):
 
3064
            return self._vfs_heads_to_fetch()
 
3065
        try:
 
3066
            return self._rpc_heads_to_fetch()
 
3067
        except errors.UnknownSmartMethod:
 
3068
            medium._remember_remote_is_before((2, 4))
 
3069
            return self._vfs_heads_to_fetch()
 
3070
 
 
3071
    def _rpc_heads_to_fetch(self):
 
3072
        response = self._call('Branch.heads_to_fetch', self._remote_path())
 
3073
        if len(response) != 2:
 
3074
            raise errors.UnexpectedSmartServerResponse(response)
 
3075
        must_fetch, if_present_fetch = response
 
3076
        return set(must_fetch), set(if_present_fetch)
 
3077
 
 
3078
    def _vfs_heads_to_fetch(self):
 
3079
        self._ensure_real()
 
3080
        return self._real_branch.heads_to_fetch()
 
3081
 
2718
3082
 
2719
3083
class RemoteConfig(object):
2720
3084
    """A Config that reads and writes from smart verbs.
2774
3138
        medium = self._branch._client._medium
2775
3139
        if medium._is_remote_before((1, 14)):
2776
3140
            return self._vfs_set_option(value, name, section)
 
3141
        if isinstance(value, dict):
 
3142
            if medium._is_remote_before((2, 2)):
 
3143
                return self._vfs_set_option(value, name, section)
 
3144
            return self._set_config_option_dict(value, name, section)
 
3145
        else:
 
3146
            return self._set_config_option(value, name, section)
 
3147
 
 
3148
    def _set_config_option(self, value, name, section):
2777
3149
        try:
2778
3150
            path = self._branch._remote_path()
2779
3151
            response = self._branch._client.call('Branch.set_config_option',
2780
3152
                path, self._branch._lock_token, self._branch._repo_lock_token,
2781
3153
                value.encode('utf8'), name, section or '')
2782
3154
        except errors.UnknownSmartMethod:
 
3155
            medium = self._branch._client._medium
2783
3156
            medium._remember_remote_is_before((1, 14))
2784
3157
            return self._vfs_set_option(value, name, section)
2785
3158
        if response != ():
2786
3159
            raise errors.UnexpectedSmartServerResponse(response)
2787
3160
 
 
3161
    def _serialize_option_dict(self, option_dict):
 
3162
        utf8_dict = {}
 
3163
        for key, value in option_dict.items():
 
3164
            if isinstance(key, unicode):
 
3165
                key = key.encode('utf8')
 
3166
            if isinstance(value, unicode):
 
3167
                value = value.encode('utf8')
 
3168
            utf8_dict[key] = value
 
3169
        return bencode.bencode(utf8_dict)
 
3170
 
 
3171
    def _set_config_option_dict(self, value, name, section):
 
3172
        try:
 
3173
            path = self._branch._remote_path()
 
3174
            serialised_dict = self._serialize_option_dict(value)
 
3175
            response = self._branch._client.call(
 
3176
                'Branch.set_config_option_dict',
 
3177
                path, self._branch._lock_token, self._branch._repo_lock_token,
 
3178
                serialised_dict, name, section or '')
 
3179
        except errors.UnknownSmartMethod:
 
3180
            medium = self._branch._client._medium
 
3181
            medium._remember_remote_is_before((2, 2))
 
3182
            return self._vfs_set_option(value, name, section)
 
3183
        if response != ():
 
3184
            raise errors.UnexpectedSmartServerResponse(response)
 
3185
 
2788
3186
    def _real_object(self):
2789
3187
        self._branch._ensure_real()
2790
3188
        return self._branch._real_branch
2873
3271
                    'Missing key %r in context %r', key_err.args[0], context)
2874
3272
                raise err
2875
3273
 
2876
 
    if err.error_verb == 'IncompatibleRepositories':
2877
 
        raise errors.IncompatibleRepositories(err.error_args[0],
2878
 
            err.error_args[1], err.error_args[2])
2879
 
    elif err.error_verb == 'NoSuchRevision':
 
3274
    if err.error_verb == 'NoSuchRevision':
2880
3275
        raise NoSuchRevision(find('branch'), err.error_args[0])
2881
3276
    elif err.error_verb == 'nosuchrevision':
2882
3277
        raise NoSuchRevision(find('repository'), err.error_args[0])
2889
3284
            detail=extra)
2890
3285
    elif err.error_verb == 'norepository':
2891
3286
        raise errors.NoRepositoryPresent(find('bzrdir'))
2892
 
    elif err.error_verb == 'LockContention':
2893
 
        raise errors.LockContention('(remote lock)')
2894
3287
    elif err.error_verb == 'UnlockableTransport':
2895
3288
        raise errors.UnlockableTransport(find('bzrdir').root_transport)
2896
 
    elif err.error_verb == 'LockFailed':
2897
 
        raise errors.LockFailed(err.error_args[0], err.error_args[1])
2898
3289
    elif err.error_verb == 'TokenMismatch':
2899
3290
        raise errors.TokenMismatch(find('token'), '(remote token)')
2900
3291
    elif err.error_verb == 'Diverged':
2901
3292
        raise errors.DivergedBranches(find('branch'), find('other_branch'))
2902
 
    elif err.error_verb == 'TipChangeRejected':
2903
 
        raise errors.TipChangeRejected(err.error_args[0].decode('utf8'))
2904
 
    elif err.error_verb == 'UnstackableBranchFormat':
2905
 
        raise errors.UnstackableBranchFormat(*err.error_args)
2906
 
    elif err.error_verb == 'UnstackableRepositoryFormat':
2907
 
        raise errors.UnstackableRepositoryFormat(*err.error_args)
2908
3293
    elif err.error_verb == 'NotStacked':
2909
3294
        raise errors.NotStacked(branch=find('branch'))
2910
3295
    elif err.error_verb == 'PermissionDenied':
2920
3305
    elif err.error_verb == 'NoSuchFile':
2921
3306
        path = get_path()
2922
3307
        raise errors.NoSuchFile(path)
 
3308
    _translate_error_without_context(err)
 
3309
 
 
3310
 
 
3311
def _translate_error_without_context(err):
 
3312
    """Translate any ErrorFromSmartServer values that don't require context"""
 
3313
    if err.error_verb == 'IncompatibleRepositories':
 
3314
        raise errors.IncompatibleRepositories(err.error_args[0],
 
3315
            err.error_args[1], err.error_args[2])
 
3316
    elif err.error_verb == 'LockContention':
 
3317
        raise errors.LockContention('(remote lock)')
 
3318
    elif err.error_verb == 'LockFailed':
 
3319
        raise errors.LockFailed(err.error_args[0], err.error_args[1])
 
3320
    elif err.error_verb == 'TipChangeRejected':
 
3321
        raise errors.TipChangeRejected(err.error_args[0].decode('utf8'))
 
3322
    elif err.error_verb == 'UnstackableBranchFormat':
 
3323
        raise errors.UnstackableBranchFormat(*err.error_args)
 
3324
    elif err.error_verb == 'UnstackableRepositoryFormat':
 
3325
        raise errors.UnstackableRepositoryFormat(*err.error_args)
2923
3326
    elif err.error_verb == 'FileExists':
2924
3327
        raise errors.FileExists(err.error_args[0])
2925
3328
    elif err.error_verb == 'DirectoryNotEmpty':
2944
3347
            raise UnicodeEncodeError(encoding, val, start, end, reason)
2945
3348
    elif err.error_verb == 'ReadOnlyError':
2946
3349
        raise errors.TransportNotPossible('readonly transport')
 
3350
    elif err.error_verb == 'MemoryError':
 
3351
        raise errors.BzrError("remote server out of memory\n"
 
3352
            "Retry non-remotely, or contact the server admin for details.")
2947
3353
    raise errors.UnknownErrorFromSmartServer(err)