[hls] Implement hlsnative fd in terms of fragment fd

This commit is contained in:
Sergey M․ 2015-07-29 02:28:30 +06:00
parent ab81ef8fa7
commit f9a5affad9

View file

@ -4,12 +4,11 @@ import os
import re import re
import subprocess import subprocess
from ..postprocessor.ffmpeg import FFmpegPostProcessor
from .common import FileDownloader from .common import FileDownloader
from ..compat import ( from .fragment import FragmentFD
compat_urlparse,
compat_urllib_request, from ..compat import compat_urlparse
) from ..postprocessor.ffmpeg import FFmpegPostProcessor
from ..utils import ( from ..utils import (
encodeArgument, encodeArgument,
encodeFilename, encodeFilename,
@ -51,54 +50,50 @@ class HlsFD(FileDownloader):
return False return False
class NativeHlsFD(FileDownloader): class NativeHlsFD(FragmentFD):
""" A more limited implementation that does not require ffmpeg """ """ A more limited implementation that does not require ffmpeg """
def real_download(self, filename, info_dict): FD_NAME = 'hlsnative'
url = info_dict['url']
self.report_destination(filename)
tmpfilename = self.temp_name(filename)
self.to_screen( def real_download(self, filename, info_dict):
'[hlsnative] %s: Downloading m3u8 manifest' % info_dict['id']) man_url = info_dict['url']
data = self.ydl.urlopen(url).read() self.to_screen('[%s] Downloading m3u8 manifest' % self.FD_NAME)
s = data.decode('utf-8', 'ignore') manifest = self.ydl.urlopen(man_url).read()
segment_urls = []
s = manifest.decode('utf-8', 'ignore')
fragment_urls = []
for line in s.splitlines(): for line in s.splitlines():
line = line.strip() line = line.strip()
if line and not line.startswith('#'): if line and not line.startswith('#'):
segment_url = ( segment_url = (
line line
if re.match(r'^https?://', line) if re.match(r'^https?://', line)
else compat_urlparse.urljoin(url, line)) else compat_urlparse.urljoin(man_url, line))
segment_urls.append(segment_url) fragment_urls.append(segment_url)
# We only download the first fragment during the test
is_test = self.params.get('test', False) if self.params.get('test', False):
remaining_bytes = self._TEST_FILE_SIZE if is_test else None
byte_counter = 0
with open(tmpfilename, 'wb') as outf:
for i, segurl in enumerate(segment_urls):
self.to_screen(
'[hlsnative] %s: Downloading segment %d / %d' %
(info_dict['id'], i + 1, len(segment_urls)))
seg_req = compat_urllib_request.Request(segurl)
if remaining_bytes is not None:
seg_req.add_header('Range', 'bytes=0-%d' % (remaining_bytes - 1))
segment = self.ydl.urlopen(seg_req).read()
if remaining_bytes is not None:
segment = segment[:remaining_bytes]
remaining_bytes -= len(segment)
outf.write(segment)
byte_counter += len(segment)
if remaining_bytes is not None and remaining_bytes <= 0:
break break
self._hook_progress({ ctx = {
'downloaded_bytes': byte_counter,
'total_bytes': byte_counter,
'filename': filename, 'filename': filename,
'status': 'finished', 'total_frags': len(fragment_urls),
}) }
self.try_rename(tmpfilename, filename)
self._prepare_and_start_frag_download(ctx)
frags_filenames = []
for i, frag_url in enumerate(fragment_urls):
frag_filename = '%s-Frag%d' % (ctx['tmpfilename'], i)
success = ctx['dl'].download(frag_filename, {'url': frag_url})
if not success:
return False
with open(frag_filename, 'rb') as down:
ctx['dest_stream'].write(down.read())
frags_filenames.append(frag_filename)
self._finish_frag_download(ctx)
for frag_file in frags_filenames:
os.remove(frag_file)
return True return True