pyclann 0.1.0__py3-none-any.whl

pyclann/__init__.py

@@ -0,0 +1,57 @@
+"""pyclann — Python client library for the Clann family-tree API."""
+
+from .client import ClannClient
+from .exceptions import (
+    ClannAuthError,
+    ClannError,
+    ClannNotFoundError,
+    ClannServerError,
+    ClannValidationError,
+)
+from .models import (
+    ChatMessage,
+    ChatSession,
+    EventType,
+    FamilyTree,
+    FamilyTreeNode,
+    LifeEvent,
+    LoginInfo,
+    Person,
+    RelationshipsResponse,
+    RelationshipType,
+    ResearchFolder,
+    ResearchNote,
+    Sex,
+    SiblingType,
+    SpouseInfo,
+    UserAiSettings,
+)
+
+__all__ = [
+    # Client
+    "ClannClient",
+    # Exceptions
+    "ClannError",
+    "ClannAuthError",
+    "ClannNotFoundError",
+    "ClannValidationError",
+    "ClannServerError",
+    # Enumerations
+    "Sex",
+    "RelationshipType",
+    "SiblingType",
+    "EventType",
+    # Models
+    "LoginInfo",
+    "FamilyTree",
+    "Person",
+    "LifeEvent",
+    "SpouseInfo",
+    "RelationshipsResponse",
+    "FamilyTreeNode",
+    "ResearchNote",
+    "ResearchFolder",
+    "ChatSession",
+    "ChatMessage",
+    "UserAiSettings",
+]

pyclann/_http.py

@@ -0,0 +1,145 @@
+"""Internal HTTP session: request dispatch and error mapping."""
+
+from __future__ import annotations
+
+from typing import Any
+
+import requests
+
+from .exceptions import (
+    ClannAuthError,
+    ClannError,
+    ClannNotFoundError,
+    ClannServerError,
+    ClannValidationError,
+)
+
+
+def _compact(d: dict[str, Any]) -> dict[str, Any]:
+    """Return a copy of *d* with all ``None``-valued keys removed."""
+    return {k: v for k, v in d.items() if v is not None}
+
+
+def _bare_id(record_id: str) -> str:
+    """Strip the SurrealDB table prefix from a record ID for use in URL paths.
+
+    ``"person:01jd4a8xyz"`` → ``"01jd4a8xyz"``.  Bare IDs pass through unchanged.
+    """
+    return record_id.split(":", 1)[-1] if ":" in record_id else record_id
+
+
+class _HttpSession:
+    """Thin wrapper around :class:`requests.Session` that handles auth headers and
+    maps non-2xx responses to typed exceptions."""
+
+    def __init__(self, api_url: str, auth_url: str) -> None:
+        self._api_url = api_url.rstrip("/")
+        self._auth_url = auth_url.rstrip("/")
+        self._session = requests.Session()
+        self._session.headers.update({"Content-Type": "application/json"})
+        self._token: str | None = None
+
+    def set_token(self, token: str) -> None:
+        """Store the JWT and attach it to all subsequent requests."""
+        self._token = token
+        self._session.headers["Authorization"] = f"Bearer {token}"
+
+    def _require_auth(self) -> None:
+        if self._token is None:
+            raise ClannAuthError("Not authenticated — call ClannClient.login() first.")
+
+    def _raise_for_status(self, response: requests.Response) -> None:
+        if response.ok:
+            return
+        try:
+            body = response.json()
+            message = body.get("error") or body.get("message") or response.text
+        except Exception:
+            message = response.text
+        if response.status_code == 401:
+            raise ClannAuthError(message)
+        if response.status_code == 403:
+            raise ClannAuthError(f"Forbidden: {message}")
+        if response.status_code == 404:
+            raise ClannNotFoundError(message)
+        if response.status_code == 400:
+            raise ClannValidationError(message)
+        if response.status_code >= 500:
+            raise ClannServerError(f"Server error {response.status_code}: {message}")
+        raise ClannError(f"HTTP {response.status_code}: {message}")
+
+    def _json_or_none(self, response: requests.Response) -> Any:
+        if not response.content:
+            return None
+        return response.json()
+
+    # ── auth service ──────────────────────────────────────────────────────────
+
+    def post_auth(self, path: str, json: Any) -> Any:
+        """POST to the authentication service (no JWT header required)."""
+        response = self._session.post(f"{self._auth_url}{path}", json=json)
+        self._raise_for_status(response)
+        return response.json()
+
+    # ── Clann API ─────────────────────────────────────────────────────────────
+
+    def get(self, path: str, params: dict[str, Any] | None = None) -> Any:
+        self._require_auth()
+        response = self._session.get(f"{self._api_url}{path}", params=params)
+        self._raise_for_status(response)
+        return response.json()
+
+    def get_bytes(self, path: str) -> bytes:
+        """Fetch a binary resource (e.g. an image).  No auth required."""
+        response = self._session.get(f"{self._api_url}{path}")
+        self._raise_for_status(response)
+        return response.content
+
+    def post(self, path: str, json: Any = None) -> Any:
+        self._require_auth()
+        response = self._session.post(f"{self._api_url}{path}", json=json)
+        self._raise_for_status(response)
+        return self._json_or_none(response)
+
+    def post_multipart(
+        self,
+        path: str,
+        field: str,
+        data: bytes,
+        filename: str,
+        content_type: str,
+    ) -> Any:
+        """Upload a file using ``multipart/form-data``.
+
+        The Content-Type header is removed from the session for this request so
+        that *requests* can set the correct ``multipart/form-data; boundary=…``
+        header automatically.
+        """
+        self._require_auth()
+        files: dict[str, tuple[str, bytes, str]] = {field: (filename, data, content_type)}
+        saved = self._session.headers.pop("Content-Type", None)
+        try:
+            response = self._session.post(f"{self._api_url}{path}", files=files)
+            self._raise_for_status(response)
+            return self._json_or_none(response)
+        finally:
+            if saved is not None:
+                self._session.headers["Content-Type"] = saved
+
+    def put(self, path: str, json: Any = None) -> Any:
+        self._require_auth()
+        response = self._session.put(f"{self._api_url}{path}", json=json)
+        self._raise_for_status(response)
+        return self._json_or_none(response)
+
+    def patch(self, path: str, json: Any = None) -> Any:
+        self._require_auth()
+        response = self._session.patch(f"{self._api_url}{path}", json=json)
+        self._raise_for_status(response)
+        return self._json_or_none(response)
+
+    def delete(self, path: str, json: Any = None) -> Any:
+        self._require_auth()
+        response = self._session.delete(f"{self._api_url}{path}", json=json)
+        self._raise_for_status(response)
+        return self._json_or_none(response)