]> jfr.im git - yt-dlp.git/blobdiff - yt_dlp/downloader/fc2.py
[ie/mlbtv] Fix extraction (#10296)
[yt-dlp.git] / yt_dlp / downloader / fc2.py
index 157bcf23eba7e42f3d186cd6a5b20e32ebeebf1c..f9763debbe831018da64bb780611150f67cb599a 100644 (file)
@@ -1,5 +1,3 @@
-from __future__ import division, unicode_literals
-
 import threading
 
 from .common import FileDownloader
@@ -20,6 +18,9 @@ def real_download(self, filename, info_dict):
         heartbeat_state = [None, 1]
 
         def heartbeat():
+            if heartbeat_state[1] < 0:
+                return
+
             try:
                 heartbeat_state[1] += 1
                 ws.send('{"name":"heartbeat","arguments":{},"id":%d}' % heartbeat_state[1])
@@ -38,4 +39,8 @@ def heartbeat():
             'ws': None,
             'protocol': 'live_ffmpeg',
         })
-        return FFmpegFD(self.ydl, self.params or {}).download(filename, new_info_dict)
+        try:
+            return FFmpegFD(self.ydl, self.params or {}).download(filename, new_info_dict)
+        finally:
+            # stop heartbeating
+            heartbeat_state[1] = -1