ITQC/pydase
24
0
Fork 0
mirror of https://github.com/tiqi-group/pydase.git synced 2025-06-06 13:30:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #228 from tiqi-group/fix/aiohttp_socks_dependency

Browse Source 
fix: using client without aiohttp_socks dependency does not raise
This commit is contained in:
Mose Müller 2025-05-21 09:49:29 +02:00 committed by GitHub
commit 0dace2a9f0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/pydase/client/client.py
View File

@ -3,11 +3,11 @@ import logging
import sys import sys
import threading import threading
import urllib.parse import urllib.parse
from builtins import ModuleNotFoundError
from types import TracebackType from types import TracebackType
from typing import TYPE_CHECKING, Any, TypedDict, cast from typing import TYPE_CHECKING, Any, TypedDict, cast
import aiohttp import aiohttp
import aiohttp_socks.connector
import socketio # type: ignore import socketio # type: ignore
from pydase.client.proxy_class import ProxyClass from pydase.client.proxy_class import ProxyClass
@ -150,6 +150,17 @@ class Client:
def _initialize_socketio_client(self) -> None: def _initialize_socketio_client(self) -> None:
if self._proxy_url is not None: if self._proxy_url is not None:
try:
import aiohttp_socks.connector
except ModuleNotFoundError:
raise ModuleNotFoundError(
"Missing dependency 'aiohttp_socks'. To use SOCKS5 proxy support, "
"install the optional 'socks' extra:\n\n"
' pip install "pydase[socks]"\n\n'
"This is required when specifying a `proxy_url` for "
"`pydase.Client`."
)
session = aiohttp.ClientSession( session = aiohttp.ClientSession(
connector=aiohttp_socks.connector.ProxyConnector.from_url( connector=aiohttp_socks.connector.ProxyConnector.from_url(
url=self._proxy_url, loop=self._loop url=self._proxy_url, loop=self._loop