Added client cert support

lexiforest · Jan 25, 2024 · 975a790 · 975a790
1 parent 76e16af
commit 975a790
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/curl_cffi/requests/__init__.py b/curl_cffi/requests/__init__.py
@@ -66,6 +66,7 @@ def request(
     debug: bool = False,
     interface: Optional[str] = None,
     multipart: Optional[CurlMime] = None,
+    cert: Optional[Union[str, Tuple[str, str]]] = None,
 ) -> Response:
     """Send an http request.
 
@@ -126,6 +127,7 @@ def request(
             http_version=http_version,
             interface=interface,
             multipart=multipart,
+            cert=cert,
         )
 
 

diff --git a/curl_cffi/requests/session.py b/curl_cffi/requests/session.py
@@ -179,6 +179,7 @@ def __init__(
         http_version: Optional[CurlHttpVersion] = None,
         debug: bool = False,
         interface: Optional[str] = None,
+        cert: Optional[Union[str, Tuple[str, str]]] = None,
     ):
         self.headers = Headers(headers)
         self.cookies = Cookies(cookies)
@@ -196,6 +197,7 @@ def __init__(
         self.http_version = http_version
         self.debug = debug
         self.interface = interface
+        self.cert = cert
 
         if proxy and proxies:
             raise TypeError("Cannot specify both 'proxy' and 'proxies'")
@@ -230,6 +232,7 @@ def _set_curl_options(
         default_headers: Optional[bool] = None,
         http_version: Optional[CurlHttpVersion] = None,
         interface: Optional[str] = None,
+        cert: Optional[Union[str, Tuple[str, str]]] = None,
         stream: bool = False,
         max_recv_speed: int = 0,
         multipart: Optional[CurlMime] = None,
@@ -428,6 +431,16 @@ def _set_curl_options(
         if accept_encoding is not None:
             c.setopt(CurlOpt.ACCEPT_ENCODING, accept_encoding.encode())
 
+        # cert
+        cert = cert or self.cert
+        if cert:
+            if isinstance(cert, str):
+                c.setopt(CurlOpt.SSLCERT, cert)
+            else:
+                cert, key = cert
+                c.setopt(CurlOpt.SSLCERT, cert)
+                c.setopt(CurlOpt.SSLKEY, key)
+
         # impersonate
         impersonate = impersonate or self.impersonate
         default_headers = (
@@ -690,6 +703,7 @@ def request(
         default_headers: Optional[bool] = None,
         http_version: Optional[CurlHttpVersion] = None,
         interface: Optional[str] = None,
+        cert: Optional[Union[str, Tuple[str, str]]] = None,
         stream: bool = False,
         max_recv_speed: int = 0,
         multipart: Optional[CurlMime] = None,
@@ -731,6 +745,7 @@ def request(
             stream=stream,
             max_recv_speed=max_recv_speed,
             multipart=multipart,
+            cert=cert,
             queue_class=queue.Queue,
             event_class=threading.Event,
         )
@@ -953,6 +968,7 @@ async def request(
         default_headers: Optional[bool] = None,
         http_version: Optional[CurlHttpVersion] = None,
         interface: Optional[str] = None,
+        cert: Optional[Union[str, Tuple[str, str]]] = None,
         stream: bool = False,
         max_recv_speed: int = 0,
         multipart: Optional[CurlMime] = None,
@@ -987,6 +1003,7 @@ async def request(
             stream=stream,
             max_recv_speed=max_recv_speed,
             multipart=multipart,
+            cert=cert,
             queue_class=asyncio.Queue,
             event_class=asyncio.Event,
         )