]>
jfr.im git - yt-dlp.git/blob - youtube_dl/downloader/hls.py
1 from __future__
import unicode_literals
7 from .common
import FileDownloader
8 from .fragment
import FragmentFD
10 from ..compat
import compat_urlparse
11 from ..postprocessor
.ffmpeg
import FFmpegPostProcessor
16 handle_youtubedl_headers
,
20 class HlsFD(FileDownloader
):
21 def real_download(self
, filename
, info_dict
):
22 url
= info_dict
['url']
23 self
.report_destination(filename
)
24 tmpfilename
= self
.temp_name(filename
)
26 ffpp
= FFmpegPostProcessor(downloader
=self
)
27 if not ffpp
.available
:
28 self
.report_error('m3u8 download detected but ffmpeg or avconv could not be found. Please install one.')
32 args
= [ffpp
.executable
, '-y']
34 if info_dict
['http_headers'] and re
.match(r
'^https?://', url
):
35 # Trailing \r\n after each HTTP header is important to prevent warning from ffmpeg/avconv:
36 # [http @ 00000000003d2fa0] No trailing CRLF found in HTTP header.
37 headers
= handle_youtubedl_headers(info_dict
['http_headers'])
40 ''.join('%s: %s\r\n' % (key
, val
) for key
, val
in headers
.items())]
42 args
+= ['-i', url
, '-f', 'mp4', '-c', 'copy', '-bsf:a', 'aac_adtstoasc']
44 args
= [encodeArgument(opt
) for opt
in args
]
45 args
.append(encodeFilename(ffpp
._ffmpeg
_filename
_argument
(tmpfilename
), True))
49 retval
= subprocess
.call(args
, stdin
=subprocess
.PIPE
)
51 fsize
= os
.path
.getsize(encodeFilename(tmpfilename
))
52 self
.to_screen('\r[%s] %s bytes' % (args
[0], fsize
))
53 self
.try_rename(tmpfilename
, filename
)
55 'downloaded_bytes': fsize
,
63 self
.report_error('%s exited with code %d' % (ffpp
.basename
, retval
))
67 class NativeHlsFD(FragmentFD
):
68 """ A more limited implementation that does not require ffmpeg """
72 def real_download(self
, filename
, info_dict
):
73 man_url
= info_dict
['url']
74 self
.to_screen('[%s] Downloading m3u8 manifest' % self
.FD_NAME
)
75 manifest
= self
.ydl
.urlopen(man_url
).read()
77 s
= manifest
.decode('utf-8', 'ignore')
79 for line
in s
.splitlines():
81 if line
and not line
.startswith('#'):
84 if re
.match(r
'^https?://', line
)
85 else compat_urlparse
.urljoin(man_url
, line
))
86 fragment_urls
.append(segment_url
)
87 # We only download the first fragment during the test
88 if self
.params
.get('test', False):
93 'total_frags': len(fragment_urls
),
96 self
._prepare
_and
_start
_frag
_download
(ctx
)
99 for i
, frag_url
in enumerate(fragment_urls
):
100 frag_filename
= '%s-Frag%d' % (ctx
['tmpfilename'], i
)
101 success
= ctx
['dl'].download(frag_filename
, {'url': frag_url}
)
104 down
, frag_sanitized
= sanitize_open(frag_filename
, 'rb')
105 ctx
['dest_stream'].write(down
.read())
107 frags_filenames
.append(frag_sanitized
)
109 self
._finish
_frag
_download
(ctx
)
111 for frag_file
in frags_filenames
:
112 os
.remove(encodeFilename(frag_file
))