"""Exception hierarchy for the Diffbot SDK."""

import json

from typing import Optional

class DiffbotError(Exception):

"""Base class for all Diffbot SDK errors."""

class ValidationError(DiffbotError):

"""Client-side validation failed (e.g. missing token, malformed argument)."""

class APIError(DiffbotError):

"""The Diffbot API returned an error response."""

def __init__(self, status_code: int, body: str):

self.status_code = status_code

self.body = body

self.message: Optional[str] = None

self.request_id: Optional[str] = None

try:

data = json.loads(body)

self.message = data.get("message")

self.request_id = data.get("requestId")

except (ValueError, AttributeError):

pass

display = self.message or (body[:200] + ("..." if len(body) > 200 else ""))

super().__init__(f"Diffbot API error {status_code}: {display}")

class AuthError(APIError):

"""Authentication failed (401, 403)."""

class RateLimitError(APIError):

"""Rate limit exceeded (429)."""

def __init__(self, status_code: int, body: str, retry_after: Optional[str] = None):

super().__init__(status_code, body)

self.retry_after = retry_after

class ExtractionError(DiffbotError):

"""The Diffbot API returned a 200 but reported an extraction failure."""

def __init__(self, error_code: int, error: str):

self.error_code = error_code

self.error = error

super().__init__(f"Diffbot extraction error {error_code}: {error}")