/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 breezy/url_policy_open.py

  • Committer: Jelmer Vernooij
  • Date: 2020-03-22 01:35:14 UTC
  • mfrom: (7490.7.6 work)
  • mto: This revision was merged to the branch mainline in revision 7499.
  • Revision ID: jelmer@jelmer.uk-20200322013514-7vw1ntwho04rcuj3
merge lp:brz/3.1.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/url_policy_open.py
16
16
 
17
17
"""Branch opening with URL-based restrictions."""
18
18
 
19
 
from __future__ import absolute_import
20
 
 
21
19
import threading
22
20
 
23
21
from . import (
24
22
    errors,
 
23
    trace,
25
24
    urlutils,
26
25
    )
27
26
from .branch import Branch
28
27
from .controldir import (
29
28
    ControlDir,
 
29
    ControlDirFormat,
 
30
    )
 
31
from .transport import (
 
32
    do_catching_redirections,
 
33
    get_transport,
30
34
    )
31
35
 
32
36
 
184
188
    checked against a policy object.
185
189
 
186
190
    The policy object is expected to have the following methods:
187
 
    * check_one_url 
 
191
    * check_one_url
188
192
    * should_follow_references
189
193
    * transform_fallback_location
190
194
    """
200
204
        """
201
205
        self.policy = policy
202
206
        self._seen_urls = set()
 
207
        if probers is None:
 
208
            probers = ControlDirFormat.all_probers()
203
209
        self.probers = probers
204
210
 
205
211
    @classmethod
248
254
    def transform_fallback_locationHook(cls, branch, url):
249
255
        """Installed as the 'transform_fallback_location' Branch hook.
250
256
 
251
 
        This method calls `transform_fallback_location` on the policy object and
252
 
        either returns the url it provides or passes it back to
 
257
        This method calls `transform_fallback_location` on the policy object
 
258
        and either returns the url it provides or passes it back to
253
259
        check_and_follow_branch_reference.
254
260
        """
255
261
        try:
276
282
            # spurious loop exceptions.
277
283
            self._seen_urls = set()
278
284
 
 
285
    def _open_dir(self, url):
 
286
        """Simple BzrDir.open clone that only uses specific probers.
 
287
 
 
288
        :param url: URL to open
 
289
        :return: ControlDir instance
 
290
        """
 
291
        def redirected(transport, e, redirection_notice):
 
292
            self.policy.check_one_url(e.target)
 
293
            redirected_transport = transport._redirected_to(
 
294
                e.source, e.target)
 
295
            if redirected_transport is None:
 
296
                raise errors.NotBranchError(e.source)
 
297
            trace.note(
 
298
                '%s is%s redirected to %s',
 
299
                transport.base, e.permanently, redirected_transport.base)
 
300
            return redirected_transport
 
301
 
 
302
        def find_format(transport):
 
303
            last_error = errors.NotBranchError(transport.base)
 
304
            for prober_kls in self.probers:
 
305
                prober = prober_kls()
 
306
                try:
 
307
                    return transport, prober.probe_transport(transport)
 
308
                except errors.NotBranchError as e:
 
309
                    last_error = e
 
310
            else:
 
311
                raise last_error
 
312
        transport = get_transport(url)
 
313
        transport, format = do_catching_redirections(
 
314
            find_format, transport, redirected)
 
315
        return format.open(transport)
 
316
 
279
317
    def follow_reference(self, url):
280
318
        """Get the branch-reference value at the specified url.
281
319
 
282
320
        This exists as a separate method only to be overriden in unit tests.
283
321
        """
284
 
        controldir = ControlDir.open(url, probers=self.probers)
 
322
        controldir = self._open_dir(url)
285
323
        return controldir.get_branch_reference()
286
324
 
287
 
    def open(self, url):
 
325
    def open(self, url, ignore_fallbacks=False):
288
326
        """Open the Bazaar branch at url, first checking it.
289
327
 
290
 
        What is acceptable means is defined by the policy's `follow_reference` and
291
 
        `check_one_url` methods.
 
328
        What is acceptable means is defined by the policy's `follow_reference`
 
329
        and `check_one_url` methods.
292
330
        """
293
331
        if not isinstance(url, str):
294
332
            raise TypeError
295
333
 
296
334
        url = self.check_and_follow_branch_reference(url)
297
335
 
298
 
        def open_branch(url):
299
 
            dir = ControlDir.open(url, probers=self.probers)
300
 
            return dir.open_branch()
 
336
        def open_branch(url, ignore_fallbacks):
 
337
            dir = self._open_dir(url)
 
338
            return dir.open_branch(ignore_fallbacks=ignore_fallbacks)
301
339
        return self.run_with_transform_fallback_location_hook_installed(
302
 
            open_branch, url)
 
340
            open_branch, url, ignore_fallbacks)
303
341
 
304
342
 
305
343
def open_only_scheme(allowed_scheme, url):