ee4d7c3d89
Instead of replacing 'https' with 'http' for every URL in 'get_downloader()', this now only happens once during downloader initialization. Also unit tests.
48 lines
1.0 KiB
Python
48 lines
1.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2015-2019 Mike Fährmann
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 2 as
|
|
# published by the Free Software Foundation.
|
|
|
|
"""Downloader modules"""
|
|
|
|
import importlib
|
|
|
|
modules = [
|
|
"http",
|
|
"text",
|
|
"ytdl",
|
|
]
|
|
|
|
|
|
def find(scheme):
|
|
"""Return downloader class suitable for handling the given scheme"""
|
|
try:
|
|
return _cache[scheme]
|
|
except KeyError:
|
|
pass
|
|
|
|
klass = None
|
|
if scheme == "https":
|
|
scheme = "http"
|
|
if scheme in modules: # prevent unwanted imports
|
|
try:
|
|
module = importlib.import_module("." + scheme, __package__)
|
|
klass = module.__downloader__
|
|
except ImportError:
|
|
pass
|
|
|
|
if scheme == "http":
|
|
_cache["http"] = _cache["https"] = klass
|
|
else:
|
|
_cache[scheme] = klass
|
|
return klass
|
|
|
|
|
|
# --------------------------------------------------------------------
|
|
# internals
|
|
|
|
_cache = {}
|