Merge branch 'main' into feature/11-rate-limit-handling

Author: bukkybyte

src/shade/__init__.py

@@ -1,4 +1,11 @@
 from .errors import HTTPError, RateLimitError, ShadeError
+from .errors import (
+    AuthenticationError,
+    InvalidRequestError,
+    NetworkError,
+    NotFoundError,
+    ShadeError,
+)
 from .gateway import Gateway
 from .http import AsyncHTTPClient, SyncHTTPClient
 
@@ -11,4 +18,11 @@
     "ShadeError",
     "HTTPError",
     "RateLimitError",
-]
+]
+    "AuthenticationError",
+    "Gateway",
+    "InvalidRequestError",
+    "NetworkError",
+    "NotFoundError",
+    "ShadeError",
+]

src/shade/errors.py

@@ -39,4 +39,35 @@ def __str__(self) -> str:  # pragma: no cover
         base = super().__str__()
         if self.retry_after is not None:
             return f"{base} (retry after {self.retry_after}s)"
-        return base
+        return base
+    def __init__(
+        self,
+        message: str,
+        status_code: Optional[int] = None,
+        response_body: Optional[str] = None,
+    ) -> None:
+        self.message = message
+        self.status_code = status_code
+        self.response_body = response_body
+        super().__init__(message)
+
+    def __str__(self) -> str:
+        if self.status_code is None:
+            return self.message
+        return f"{self.message} (status code: {self.status_code})"
+
+
+class AuthenticationError(ShadeError):
+    """Raised when authentication fails or credentials are invalid."""
+
+
+class InvalidRequestError(ShadeError):
+    """Raised when a request is malformed or rejected by validation."""
+
+
+class NotFoundError(ShadeError):
+    """Raised when an API resource cannot be found."""
+
+
+class NetworkError(ShadeError):
+    """Raised when the SDK cannot complete a network request."""

tests/test_errors.py

@@ -0,0 +1,51 @@
+import shade
+from shade import (
+    AuthenticationError,
+    InvalidRequestError,
+    NetworkError,
+    NotFoundError,
+    ShadeError,
+)
+
+
+def test_shade_error_can_be_raised_standalone():
+    error = ShadeError("contract rejected payment")
+
+    assert str(error) == "contract rejected payment"
+    assert error.message == "contract rejected payment"
+    assert error.status_code is None
+    assert error.response_body is None
+
+
+def test_shade_error_includes_http_context():
+    error = ShadeError(
+        "invalid request",
+        status_code=422,
+        response_body='{"error":"missing amount"}',
+    )
+
+    assert str(error) == "invalid request (status code: 422)"
+    assert error.status_code == 422
+    assert error.response_body == '{"error":"missing amount"}'
+
+
+def test_specific_errors_inherit_from_shade_error():
+    for error_type in (
+        AuthenticationError,
+        InvalidRequestError,
+        NetworkError,
+        NotFoundError,
+    ):
+        error = error_type("request failed", status_code=400, response_body="raw")
+
+        assert isinstance(error, ShadeError)
+        assert str(error) == "request failed (status code: 400)"
+        assert error.response_body == "raw"
+
+
+def test_package_root_exports_error_classes():
+    assert shade.ShadeError is ShadeError
+    assert shade.AuthenticationError is AuthenticationError
+    assert shade.InvalidRequestError is InvalidRequestError
+    assert shade.NetworkError is NetworkError
+    assert shade.NotFoundError is NotFoundError