[hls] Properly encode fragment filename

This commit is contained in:
Sergey M․ 2015-08-31 02:28:36 +06:00
parent db8f2bfd99
commit fcd9e423ec

View File

@ -12,6 +12,7 @@
from ..utils import ( from ..utils import (
encodeArgument, encodeArgument,
encodeFilename, encodeFilename,
sanitize_open,
) )
@ -89,13 +90,13 @@ def real_download(self, filename, info_dict):
success = ctx['dl'].download(frag_filename, {'url': frag_url}) success = ctx['dl'].download(frag_filename, {'url': frag_url})
if not success: if not success:
return False return False
with open(frag_filename, 'rb') as down: down, frag_sanitized = sanitize_open(frag_filename, 'rb')
ctx['dest_stream'].write(down.read()) ctx['dest_stream'].write(down.read())
frags_filenames.append(frag_filename) frags_filenames.append(frag_sanitized)
self._finish_frag_download(ctx) self._finish_frag_download(ctx)
for frag_file in frags_filenames: for frag_file in frags_filenames:
os.remove(frag_file) os.remove(encodeFilename(frag_file))
return True return True