mirrors/yt-dlp
1
1
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-22 16:31:41 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[downloader/fc2] Stop heatbeating once FFmpeg finishes

Browse Source 
Authored by: Lesmiscore
This commit is contained in:
Lesmiscore 2022-04-25 00:45:19 +09:00
parent acbc642250
commit 69b59b4b4b
No known key found for this signature in database
GPG Key ID: 0EC2B52CF86236FF
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
yt_dlp/downloader/fc2.py
View File

@ -18,6 +18,9 @@ def real_download(self, filename, info_dict):
heartbeat_state = [None, 1] heartbeat_state = [None, 1]
def heartbeat(): def heartbeat():
if heartbeat_state[1] < 0:
return
try: try:
heartbeat_state[1] += 1 heartbeat_state[1] += 1
ws.send('{"name":"heartbeat","arguments":{},"id":%d}' % heartbeat_state[1]) ws.send('{"name":"heartbeat","arguments":{},"id":%d}' % heartbeat_state[1])
@ -36,4 +39,8 @@ def heartbeat():
'ws': None, 'ws': None,
'protocol': 'live_ffmpeg', '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