kumi/yt-dlp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[fragment] Ensure the file is closed on error

Browse source
This commit is contained in:
pukkandan 2021-05-04 21:07:43 +05:30
parent 06425e9621
commit 6ef6bcbd6b
No known key found for this signature in database
GPG key ID: 0F00D95A001F4698
2 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
yt_dlp/downloader/dash.py
View file

@ -113,6 +113,7 @@ class DashSegmentsFD(FragmentFD):
if count > fragment_retries:
if not fatal:
return False, frag_index
ctx['dest_stream'].close()
self.report_error('Giving up after %s fragment retries' % fragment_retries)
return False, frag_index
@ -135,6 +136,7 @@ class DashSegmentsFD(FragmentFD):
self.report_skip_fragment(frag_index)
return True
else:
ctx['dest_stream'].close()
self.report_error(
'fragment %s not found, unable to continue' % frag_index)
return False
@ -143,6 +145,7 @@ class DashSegmentsFD(FragmentFD):
self.report_skip_fragment(frag_index)
return True
else:
ctx['dest_stream'].close()
self.report_error(
'fragment %s not found, unable to continue' % frag_index)
return False

3
yt_dlp/downloader/hls.py
View file

@ -299,6 +299,7 @@ class HlsFD(FragmentFD):
if count <= fragment_retries:
self.report_retry_fragment(err, frag_index, count, fragment_retries)
if count > fragment_retries:
ctx['dest_stream'].close()
self.report_error('Giving up after %s fragment retries' % fragment_retries)
return False, frag_index
@ -403,6 +404,7 @@ class HlsFD(FragmentFD):
self.report_skip_fragment(frag_index)
return True
else:
ctx['dest_stream'].close()
self.report_error(
'fragment %s not found, unable to continue' % frag_index)
return False
@ -411,6 +413,7 @@ class HlsFD(FragmentFD):
self.report_skip_fragment(frag_index)
return True
else:
ctx['dest_stream'].close()
self.report_error(
'fragment %s not found, unable to continue' % frag_index)
return False