2014-01-27 04:47:30 +00:00
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
2013-09-23 15:59:27 +00:00
|
|
|
from .common import FileDownloader
|
2015-01-24 00:38:48 +00:00
|
|
|
from .f4m import F4mFD
|
2013-09-23 15:59:27 +00:00
|
|
|
from .hls import HlsFD
|
|
|
|
from .http import HttpFD
|
|
|
|
from .rtmp import RtmpFD
|
2015-06-04 14:05:33 +00:00
|
|
|
from .dash import DashSegmentsFD
|
2016-03-13 14:24:02 +00:00
|
|
|
from .rtsp import RtspFD
|
2016-02-19 18:29:24 +00:00
|
|
|
from .external import (
|
|
|
|
get_external_downloader,
|
|
|
|
FFmpegFD,
|
|
|
|
)
|
2013-09-23 15:59:27 +00:00
|
|
|
|
|
|
|
from ..utils import (
|
2015-01-23 22:50:31 +00:00
|
|
|
determine_protocol,
|
2013-09-23 15:59:27 +00:00
|
|
|
)
|
|
|
|
|
2015-01-23 22:50:31 +00:00
|
|
|
PROTOCOL_MAP = {
|
|
|
|
'rtmp': RtmpFD,
|
2016-02-19 18:29:24 +00:00
|
|
|
'm3u8_native': HlsFD,
|
|
|
|
'm3u8': FFmpegFD,
|
2016-03-13 14:24:02 +00:00
|
|
|
'mms': RtspFD,
|
|
|
|
'rtsp': RtspFD,
|
2015-01-23 22:50:31 +00:00
|
|
|
'f4m': F4mFD,
|
2015-06-04 14:27:29 +00:00
|
|
|
'http_dash_segments': DashSegmentsFD,
|
2015-01-23 22:50:31 +00:00
|
|
|
}
|
2014-01-25 11:02:43 +00:00
|
|
|
|
2015-01-23 22:50:31 +00:00
|
|
|
|
|
|
|
def get_suitable_downloader(info_dict, params={}):
|
2013-09-23 15:59:27 +00:00
|
|
|
"""Get the downloader class that can handle the info dict."""
|
2015-01-23 22:50:31 +00:00
|
|
|
protocol = determine_protocol(info_dict)
|
|
|
|
info_dict['protocol'] = protocol
|
|
|
|
|
2016-03-13 15:16:26 +00:00
|
|
|
# if (info_dict.get('start_time') or info_dict.get('end_time')) and FFmpegFD.can_download(info_dict):
|
|
|
|
# return FFmpegFD
|
2016-02-19 18:29:24 +00:00
|
|
|
|
2015-01-24 00:38:48 +00:00
|
|
|
external_downloader = params.get('external_downloader')
|
|
|
|
if external_downloader is not None:
|
|
|
|
ed = get_external_downloader(external_downloader)
|
2016-03-13 13:53:17 +00:00
|
|
|
if ed.can_download(info_dict):
|
2015-01-24 00:38:48 +00:00
|
|
|
return ed
|
|
|
|
|
2015-02-17 11:09:12 +00:00
|
|
|
if protocol == 'm3u8' and params.get('hls_prefer_native'):
|
2016-02-19 18:29:24 +00:00
|
|
|
return HlsFD
|
2015-02-17 11:09:12 +00:00
|
|
|
|
2015-01-23 22:50:31 +00:00
|
|
|
return PROTOCOL_MAP.get(protocol, HttpFD)
|
|
|
|
|
2014-11-23 21:25:12 +00:00
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
'get_suitable_downloader',
|
|
|
|
'FileDownloader',
|
|
|
|
]
|