mirror/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-03-09 12:50:23 -05:00
Code Activity

minor fix

Browse source
This commit is contained in:
coletdjnz 2024-05-18 17:28:25 +12:00
parent d274eb1f53
commit 4cfce3b22f
No known key found for this signature in database
GPG key ID: 91984263BB39894A
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
yt_dlp/networking/_websockets.py
View file

@ -201,9 +201,7 @@ def _send(self, request):
ssl_context = None ssl_context = None
sock = self._make_sock(proxy, request.url, timeout) sock = self._make_sock(proxy, request.url, timeout)
if parse_uri(request.url).secure: if parse_uri(request.url).secure:
ssl_context = self._make_sslcontext() ssl_context = WebsocketsSSLContext(self._make_sslcontext())
if isinstance(sock, ssl.SSLSocket) and WebsocketsSSLContext: # tls in tls
ssl_context = WebsocketsSSLContext(ssl_context)
try: try:
conn = websockets.sync.client.connect( conn = websockets.sync.client.connect(
@ -272,6 +270,6 @@ def __init__(self, ssl_context: ssl.SSLContext):
self.ssl_context = ssl_context self.ssl_context = ssl_context
def wrap_socket(self, sock, server_hostname=None): def wrap_socket(self, sock, server_hostname=None):
if WebsocketsSSLTransport: if isinstance(sock, ssl.SSLSocket) and WebsocketsSSLTransport:
return WebsocketsSSLTransport(sock, self.ssl_context, server_hostname=server_hostname) return WebsocketsSSLTransport(sock, self.ssl_context, server_hostname=server_hostname)
return self.ssl_context.wrap_socket(sock, server_hostname=server_hostname) return self.ssl_context.wrap_socket(sock, server_hostname=server_hostname)