From: Lesmiscore Date: Sun, 24 Apr 2022 15:45:19 +0000 (+0900) Subject: [downloader/fc2] Stop heatbeating once FFmpeg finishes X-Git-Tag: 2022.05.18~93 X-Git-Url: https://jfr.im/git/yt-dlp.git/commitdiff_plain/69b59b4b4b52e496df980d8d21ad5ff670089c0b [downloader/fc2] Stop heatbeating once FFmpeg finishes Authored by: Lesmiscore --- diff --git a/yt_dlp/downloader/fc2.py b/yt_dlp/downloader/fc2.py index d503aac04..f9763debb 100644 --- a/yt_dlp/downloader/fc2.py +++ b/yt_dlp/downloader/fc2.py @@ -18,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]) @@ -36,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