[downloader/fragment] Make single thread download work for --live-fro…

…m-start (yt-dlp#3446) Authored by: Lesmiscore
sss333xj · Apr 16, 2022 · c854208 · c854208
1 parent e06bd88
commit c854208
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/yt_dlp/downloader/fragment.py b/yt_dlp/downloader/fragment.py
@@ -521,8 +521,13 @@ def _download_fragment(fragment):
             for fragment in fragments:
                 if not interrupt_trigger[0]:
                     break
-                download_fragment(fragment, ctx)
-                result = append_fragment(decrypt_fragment(fragment, self._read_fragment(ctx)), fragment['frag_index'], ctx)
+                try:
+                    download_fragment(fragment, ctx)
+                    result = append_fragment(decrypt_fragment(fragment, self._read_fragment(ctx)), fragment['frag_index'], ctx)
+                except KeyboardInterrupt:
+                    if info_dict.get('is_live'):
+                        break
+                    raise
                 if not result:
                     return False