youtube-dl/youtube_dl/downloader/__init__.py

38 lines
910 B
Python
Raw Normal View History

2014-01-27 04:47:30 +00:00
from __future__ import unicode_literals
from .common import FileDownloader
from .hls import HlsFD
from .hls import NativeHlsFD
from .http import HttpFD
from .mplayer import MplayerFD
from .rtmp import RtmpFD
2013-12-23 15:39:49 +00:00
from .f4m import F4mFD
from ..utils import (
determine_ext,
)
2014-01-25 11:02:43 +00:00
def get_suitable_downloader(info_dict):
"""Get the downloader class that can handle the info dict."""
url = info_dict['url']
2014-01-27 04:47:30 +00:00
protocol = info_dict.get('protocol')
if url.startswith('rtmp'):
return RtmpFD
if protocol == 'm3u8_native':
return NativeHlsFD
2014-01-27 04:47:30 +00:00
if (protocol == 'm3u8') or (protocol is None and determine_ext(url) == 'm3u8'):
return HlsFD
if url.startswith('mms') or url.startswith('rtsp'):
return MplayerFD
2013-12-23 15:39:49 +00:00
if determine_ext(url) == 'f4m':
return F4mFD
else:
return HttpFD
__all__ = [
'get_suitable_downloader',
'FileDownloader',
]