kumi/yt-dlp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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 @@ class FC2LiveFD(FileDownloader):
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 @@ class FC2LiveFD(FileDownloader):
'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