[hlsnative] Correct handling when remaining_bytes is None

This commit is contained in:
Philipp Hagemeister 2014-09-25 09:21:45 +02:00
parent c6e90caaa6
commit fec02bcc90

View file

@ -81,16 +81,16 @@ def real_download(self, filename, info_dict):
'[hlsnative] %s: Downloading segment %d / %d' % '[hlsnative] %s: Downloading segment %d / %d' %
(info_dict['id'], i + 1, len(segment_urls))) (info_dict['id'], i + 1, len(segment_urls)))
seg_req = compat_urllib_request.Request(segurl) seg_req = compat_urllib_request.Request(segurl)
if remaining_bytes: if remaining_bytes is not None:
seg_req.add_header('Range', 'bytes=0-%d' % (remaining_bytes - 1)) seg_req.add_header('Range', 'bytes=0-%d' % (remaining_bytes - 1))
segment = self.ydl.urlopen(seg_req).read() segment = self.ydl.urlopen(seg_req).read()
if remaining_bytes: if remaining_bytes is not None:
segment = segment[:remaining_bytes] segment = segment[:remaining_bytes]
remaining_bytes -= len(segment) remaining_bytes -= len(segment)
outf.write(segment) outf.write(segment)
byte_counter += len(segment) byte_counter += len(segment)
if remaining_bytes <= 0: if remaining_bytes is not None and remaining_bytes <= 0:
break break
self._hook_progress({ self._hook_progress({