Pass in the proxy URL to the websocket connection

This commit is contained in:
DevilXD committed 2022-04-08 22:13:18 +02:00
1 parent fa59cde737
commit 5b51fb1602
1 file changed
+5 -1
+5 -1
View File
@@ -112,9 +112,13 @@ class Websocket:
) -> abc.AsyncGenerator[aiohttp.ClientWebSocketResponse, None]: ) -> abc.AsyncGenerator[aiohttp.ClientWebSocketResponse, None]:
session = await self._twitch.get_session() session = await self._twitch.get_session()
backoff = ExponentialBackoff(**kwargs) backoff = ExponentialBackoff(**kwargs)
if self._twitch.settings.proxy:
proxy = self._twitch.settings.proxy
else:
proxy = None
for delay in backoff: for delay in backoff:
try: try:
async with session.ws_connect(ws_url, ssl=True) as websocket: async with session.ws_connect(ws_url, ssl=True, proxy=proxy) as websocket:
backoff.reset() backoff.reset()
yield websocket yield websocket
except aiohttp.ClientConnectionError: except aiohttp.ClientConnectionError: