未验证 提交 acbc6422 编写于 作者: L Lesmiscore

[utils] WebSocketsWrapper: Ignore warnings at websockets instantiation

This also fixes crash caused by moving asyncio to .compat.

Authored by: Lesmiscore
Thanks: J.Chung at Discord (581418557871620106)
上级 96b49af0
...@@ -36,6 +36,7 @@ ...@@ -36,6 +36,7 @@
import time import time
import traceback import traceback
import urllib.parse import urllib.parse
import warnings
import xml.etree.ElementTree import xml.etree.ElementTree
import zlib import zlib
...@@ -5221,7 +5222,11 @@ class WebSocketsWrapper(): ...@@ -5221,7 +5222,11 @@ class WebSocketsWrapper():
pool = None pool = None
def __init__(self, url, headers=None, connect=True): def __init__(self, url, headers=None, connect=True):
self.loop = asyncio.events.new_event_loop() self.loop = asyncio.new_event_loop()
with warnings.catch_warnings():
warnings.simplefilter("ignore")
# https://github.com/aaugustin/websockets/blob/9c87d43f1d7bbf6847350087aae74fd35f73a642/src/websockets/legacy/client.py#L480
# the reason to keep giving `loop` parameter: we aren't in async function
self.conn = websockets.connect( self.conn = websockets.connect(
url, extra_headers=headers, ping_interval=None, url, extra_headers=headers, ping_interval=None,
close_timeout=float('inf'), loop=self.loop, ping_timeout=float('inf')) close_timeout=float('inf'), loop=self.loop, ping_timeout=float('inf'))
...@@ -5231,6 +5236,8 @@ def __init__(self, url, headers=None, connect=True): ...@@ -5231,6 +5236,8 @@ def __init__(self, url, headers=None, connect=True):
def __enter__(self): def __enter__(self):
if not self.pool: if not self.pool:
with warnings.catch_warnings():
warnings.simplefilter("ignore")
self.pool = self.run_with_loop(self.conn.__aenter__(), self.loop) self.pool = self.run_with_loop(self.conn.__aenter__(), self.loop)
return self return self
...@@ -5251,7 +5258,7 @@ def __exit__(self, type, value, traceback): ...@@ -5251,7 +5258,7 @@ def __exit__(self, type, value, traceback):
# for contributors: If there's any new library using asyncio needs to be run in non-async, move these function out of this class # for contributors: If there's any new library using asyncio needs to be run in non-async, move these function out of this class
@staticmethod @staticmethod
def run_with_loop(main, loop): def run_with_loop(main, loop):
if not asyncio.coroutines.iscoroutine(main): if not asyncio.iscoroutine(main):
raise ValueError(f'a coroutine was expected, got {main!r}') raise ValueError(f'a coroutine was expected, got {main!r}')
try: try:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册