Miroirs/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp synced 2024-12-28 22:24:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

apply patch

Browse source
This commit is contained in:
Lesmiscore 2022-07-14 09:11:54 +09:00
parent df63ae4477
commit 012993fa8d
No known key found for this signature in database
GPG key ID: 0EC2B52CF86236FF
1 changed files with 9 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
yt_dlp/downloader/websocket.py
View file

@ -9,7 +9,7 @@ from ..compat import asyncio
from ..dependencies import websockets from ..dependencies import websockets
class AsyncSinkFD(FileDownloader): class _WebSocketFD(FileDownloader):
async def connect(self, stdin, info_dict): async def connect(self, stdin, info_dict):
try: try:
await self.real_connection(stdin, info_dict) await self.real_connection(stdin, info_dict)
@ -21,11 +21,15 @@ class AsyncSinkFD(FileDownloader):
stdin.close() stdin.close()
async def real_connection(self, sink, info_dict): async def real_connection(self, sink, info_dict):
""" Override this in subclasses """ async with websockets.connect(info_dict['url'], extra_headers=info_dict.get('http_headers', {})) as ws:
raise NotImplementedError('This method must be implemented by subclasses') while True:
recv = await ws.recv()
if isinstance(recv, str):
recv = recv.encode('utf8')
sink.write(recv)
class FFmpegSinkFD(AsyncSinkFD): class WebSocketFragmentFD(_WebSocketFD):
""" A sink to ffmpeg for downloading fragments in any form """ """ A sink to ffmpeg for downloading fragments in any form """
def real_download(self, filename, info_dict): def real_download(self, filename, info_dict):
@ -45,7 +49,7 @@ class FFmpegSinkFD(AsyncSinkFD):
return FFmpegStdinFD(self.ydl, self.params or {}).download(filename, info_copy) return FFmpegStdinFD(self.ydl, self.params or {}).download(filename, info_copy)
class FileSinkFD(AsyncSinkFD): class WebSocketToFileFD(_WebSocketFD):
""" A sink to a file for downloading fragments in any form """ """ A sink to a file for downloading fragments in any form """
def real_download(self, filename, info_dict): def real_download(self, filename, info_dict):
tempname = self.temp_name(filename) tempname = self.temp_name(filename)
@ -87,21 +91,3 @@ class FileSinkFD(AsyncSinkFD):
finally: finally:
os.replace(tempname, filename) os.replace(tempname, filename)
return True return True
class _WebSocketFD(AsyncSinkFD):
async def real_connection(self, sink, info_dict):
async with websockets.connect(info_dict['url'], extra_headers=info_dict.get('http_headers', {})) as ws:
while True:
recv = await ws.recv()
if isinstance(recv, str):
recv = recv.encode('utf8')
sink.write(recv)
class WebSocketFragmentFD(_WebSocketFD, FFmpegSinkFD):
pass
class WebSocketToFileFD(_WebSocketFD, FileSinkFD):
pass