Add context manager support to AnticaptchaClient

python_anticaptcha/base.py

@@ -98,6 +98,16 @@ def __init__(
         )
         self.session = requests.Session()
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.session.close()
+        return False
+
+    def close(self):
+        self.session.close()
+
     @property
     def client_ip(self) -> str:
         if not hasattr(self, "_client_ip"):

tests/test_base.py

@@ -149,6 +149,38 @@ def test_timeout_raises(self, mock_sleep):
         assert "exceeded" in str(exc_info.value).lower()
 
 
+class TestContextManager:
+    def test_enter_returns_self(self):
+        client = AnticaptchaClient("key123")
+        assert client.__enter__() is client
+
+    def test_exit_closes_session(self):
+        client = AnticaptchaClient("key123")
+        with patch.object(client.session, "close") as mock_close:
+            client.__exit__(None, None, None)
+        mock_close.assert_called_once()
+
+    def test_close_closes_session(self):
+        client = AnticaptchaClient("key123")
+        with patch.object(client.session, "close") as mock_close:
+            client.close()
+        mock_close.assert_called_once()
+
+    def test_with_statement(self):
+        with AnticaptchaClient("key123") as client:
+            assert isinstance(client, AnticaptchaClient)
+            session = client.session
+        with patch.object(session, "close") as mock_close:
+            # Session was already closed by __exit__, verify it's callable
+            session.close()
+        mock_close.assert_called_once()
+
+    def test_exit_returns_false(self):
+        client = AnticaptchaClient("key123")
+        result = client.__exit__(None, None, None)
+        assert result is False
+
+
 class TestSnakeCaseAliases:
     def test_create_task_alias(self):
         assert AnticaptchaClient.create_task is AnticaptchaClient.createTask