Add raise_for_status parameter to Session and AsyncSession (#651)

* Add raise_for_status parameter to Session and AsyncSession

- Add raise_for_status parameter (default: False) to both Session and AsyncSession
- Automatically raises HTTPError for 4xx/5xx status codes when enabled
- Similar to aiohttp's raise_for_status parameter
- Add 4 unit tests (2 sync, 2 async) to verify functionality
- Maintains 100% backward compatibility with default False value

* Fix linting: break long docstring lines to <= 88 chars

* Fix linting: line length and assert statements

Author: RohanDisa

curl_cffi/requests/session.py

@@ -80,6 +80,7 @@ class BaseSessionParams(Generic[R], TypedDict, total=False):
         cert: Optional[Union[str, tuple[str, str]]]
         response_class: Optional[type[R]]
         discard_cookies: bool
+        raise_for_status: bool
 
     class StreamRequestParams(TypedDict, total=False):
         params: Optional[Union[dict, list, tuple]]
@@ -188,6 +189,7 @@ def __init__(
         cert: Optional[Union[str, tuple[str, str]]] = None,
         response_class: Optional[type[R]] = None,
         discard_cookies: bool = False,
+        raise_for_status: bool = False,
     ):
         self.headers = Headers(headers)
         self._cookies = Cookies(cookies)  # guarded by @property
@@ -220,6 +222,7 @@ def __init__(
             )
         self.response_class = response_class or Response
         self.discard_cookies = discard_cookies
+        self.raise_for_status = raise_for_status
 
         if proxy and proxies:
             raise TypeError("Cannot specify both 'proxy' and 'proxies'")
@@ -383,6 +386,8 @@ def __init__(
                 automatic detection.
             cert: a tuple of (cert, key) filenames for client cert.
             response_class: A customized subtype of ``Response`` to use.
+            raise_for_status: automatically raise an HTTPError for 4xx and 5xx
+                status codes.
 
         Notes:
             This class can be used as a context manager.
@@ -628,6 +633,8 @@ def cleanup(fut):
             rsp.stream_task = stream_task
             rsp.quit_now = quit_now
             rsp.queue = q
+            if self.raise_for_status:
+                rsp.raise_for_status()
             return rsp
         else:
             try:
@@ -653,6 +660,8 @@ def cleanup(fut):
                     c, buffer, header_buffer, default_encoding, discard_cookies
                 )
                 rsp.request = req
+                if self.raise_for_status:
+                    rsp.raise_for_status()
                 return rsp
             finally:
                 c.reset()
@@ -730,6 +739,8 @@ def __init__(
                 automatic detection.
             cert: a tuple of (cert, key) filenames for client cert.
             response_class: A customized subtype of ``Response`` to use.
+            raise_for_status: automatically raise an HTTPError for 4xx and 5xx
+                status codes.
 
         Notes:
             This class can be used as a context manager, and it's recommended to use via
@@ -1066,6 +1077,8 @@ def cleanup(fut):
             rsp.astream_task = stream_task
             rsp.quit_now = quit_now
             rsp.queue = q
+            if self.raise_for_status:
+                rsp.raise_for_status()
             return rsp
         else:
             try:
@@ -1083,6 +1096,8 @@ def cleanup(fut):
                     curl, buffer, header_buffer, default_encoding, discard_cookies
                 )
                 rsp.request = req
+                if self.raise_for_status:
+                    rsp.raise_for_status()
                 return rsp
             finally:
                 self.release_curl(curl)

tests/unittest/test_async_session.py

@@ -457,3 +457,25 @@ async def test_stream_atext(server):
             text = await r.atext()
             chunks = text.split("\n")
             assert len(chunks) == 20
+
+
+async def test_async_session_auto_raise_for_status_enabled(server):
+    """Test that AsyncSession automatically raises HTTPError for error status codes
+    when raise_for_status=True"""
+    from curl_cffi.requests.exceptions import HTTPError
+    
+    async with AsyncSession(raise_for_status=True) as s:
+        try:
+            await s.get(str(server.url.copy_with(path="/status/404")))
+            raise AssertionError("Should have raised HTTPError for 404")
+        except HTTPError as e:
+            assert e.response.status_code == 404  # type: ignore
+
+
+async def test_async_session_auto_raise_for_status_disabled(server):
+    """Test that AsyncSession does NOT raise HTTPError when raise_for_status=False
+    (default)"""
+    async with AsyncSession(raise_for_status=False) as s:
+        r = await s.get(str(server.url.copy_with(path="/status/404")))
+        assert r.status_code == 404
+        # Should not raise an exception

tests/unittest/test_requests.py

@@ -982,3 +982,23 @@ def test_response_ip_and_port(server):
 def test_http_version(server):
     r = requests.get(str(server.url), http_version="v1")
     assert r.status_code == 200
+
+
+def test_session_auto_raise_for_status_enabled(server):
+    """Test that Session automatically raises HTTPError for error status codes
+    when raise_for_status=True"""
+    s = requests.Session(raise_for_status=True)
+    try:
+        s.get(str(server.url.copy_with(path="/status/404")))
+        raise AssertionError("Should have raised HTTPError for 404")
+    except HTTPError as e:
+        assert e.response.status_code == 404  # type: ignore
+
+
+def test_session_auto_raise_for_status_disabled(server):
+    """Test that Session does NOT raise HTTPError when raise_for_status=False
+    (default)"""
+    s = requests.Session(raise_for_status=False)
+    r = s.get(str(server.url.copy_with(path="/status/404")))
+    assert r.status_code == 404
+    # Should not raise an exception