This commit is contained in:
ischmidt20 2024-05-03 12:59:15 -07:00 committed by GitHub
commit 3c15a1d4b9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 3 deletions

View File

@ -498,7 +498,11 @@ class FragmentFD(FileDownloader):
'fragment_filename_sanitized': frag_filename, 'fragment_filename_sanitized': frag_filename,
'fragment_index': frag_index, 'fragment_index': frag_index,
}) })
if not append_fragment(decrypt_fragment(fragment, self._read_fragment(ctx)), frag_index, ctx): decrypted_fragment = decrypt_fragment(fragment, self._read_fragment(ctx))
if self.params.get('keep_fragments', False):
f, _ = self.sanitize_open(ctx.get('fragment_filename_sanitized'), 'wb')
f.write(decrypted_fragment)
if not append_fragment(decrypted_fragment, frag_index, ctx):
return False return False
except KeyboardInterrupt: except KeyboardInterrupt:
self._finish_multiline_status() self._finish_multiline_status()
@ -512,8 +516,11 @@ class FragmentFD(FileDownloader):
break break
try: try:
download_fragment(fragment, ctx) download_fragment(fragment, ctx)
result = append_fragment( decrypted_fragment = decrypt_fragment(fragment, self._read_fragment(ctx))
decrypt_fragment(fragment, self._read_fragment(ctx)), fragment['frag_index'], ctx) if self.params.get('keep_fragments', False):
f, _ = self.sanitize_open(ctx.get('fragment_filename_sanitized'), 'wb')
f.write(decrypted_fragment)
result = append_fragment(decrypted_fragment, fragment['frag_index'], ctx)
except KeyboardInterrupt: except KeyboardInterrupt:
if info_dict.get('is_live'): if info_dict.get('is_live'):
break break