feat: add proxy support to ClobClient initialization

py_clob_client/client.py

@@ -89,6 +89,7 @@
     delete,
     get,
     post,
+    set_proxy,
     drop_notifications_query_params,
     add_balance_allowance_params_to_url,
     add_order_scoring_params_to_url,
@@ -124,6 +125,7 @@ def __init__(
         funder: str = None,
         builder_config: BuilderConfig = None,
         tick_size_ttl: float = 300.0,
+        proxy: Optional[str] = None,
     ):
         """
         Initializes the clob client
@@ -159,6 +161,9 @@ def __init__(
         self.__neg_risk = {}
         self.__fee_rates = {}
 
+        # proxy
+        set_proxy(proxy)
+
         # RFQ client
         self.rfq = RfqClient(self)
 

py_clob_client/http_helpers/helpers.py

@@ -19,6 +19,22 @@
 _http_client = httpx.Client(http2=True)
 
 
+def set_proxy(proxy: str = None):
+    """
+    Reconfigures the shared HTTP client to use the given proxy.
+    Pass None or empty string to disable proxy.
+    """
+    global _http_client
+
+    old_client = _http_client
+    kwargs = {"http2": True}
+    if proxy:
+        kwargs["proxy"] = proxy
+
+    _http_client = httpx.Client(**kwargs)
+    old_client.close()
+
+
 def overloadHeaders(method: str, headers: dict) -> dict:
     if headers is None:
         headers = dict()

tests/http_helpers/test_proxy.py

@@ -0,0 +1,39 @@
+from unittest import TestCase
+from unittest.mock import patch, MagicMock
+
+from py_clob_client.http_helpers.helpers import set_proxy
+
+
+class TestProxy(TestCase):
+    @patch("py_clob_client.http_helpers.helpers.httpx.Client")
+    def test_set_proxy_creates_client_with_proxy(self, mock_client_cls):
+        """set_proxy should create a new httpx.Client with the proxy param."""
+        old_client = MagicMock()
+        mock_client_cls.side_effect = [old_client, MagicMock()]
+
+        set_proxy("http://localhost:8080")
+
+        mock_client_cls.assert_called_with(http2=True, proxy="http://localhost:8080")
+        old_client.close.assert_called_once()
+
+    @patch("py_clob_client.http_helpers.helpers.httpx.Client")
+    def test_set_proxy_none_creates_client_without_proxy(self, mock_client_cls):
+        """set_proxy(None) should create a client without the proxy param."""
+        old_client = MagicMock()
+        mock_client_cls.side_effect = [old_client, MagicMock()]
+
+        set_proxy(None)
+
+        mock_client_cls.assert_called_with(http2=True)
+        old_client.close.assert_called_once()
+
+    @patch("py_clob_client.http_helpers.helpers.httpx.Client")
+    def test_set_proxy_empty_string_creates_client_without_proxy(self, mock_client_cls):
+        """set_proxy('') should create a client without the proxy param."""
+        old_client = MagicMock()
+        mock_client_cls.side_effect = [old_client, MagicMock()]
+
+        set_proxy("")
+
+        mock_client_cls.assert_called_with(http2=True)
+        old_client.close.assert_called_once()