kabbalah.py

"""Kabbalah category client."""

from __future__ import annotations

from astroapi.categories.base import BaseCategoryClient
from astroapi.types.requests import (
    BirthAngelsRequest,
    GematriaRequest,
    TikkunRequest,
    TreeOfLifeChartRequest,
)
from astroapi.types.responses import GenericResponse
from astroapi.utils.validators import validate_datetime_location


class KabbalahClient(BaseCategoryClient):
    """Client for Kabbalah endpoints."""

    API_PREFIX = "/api/v3/kabbalah"

    def get_tree_of_life_chart(self, request: TreeOfLifeChartRequest) -> GenericResponse:
        """Get Tree of Life chart."""
        validate_datetime_location(request.birth_data)
        url = self._build_url("tree-of-life-chart")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_birth_angels(self, request: BirthAngelsRequest) -> GenericResponse:
        """Get birth angels."""
        validate_datetime_location(request.birth_data)
        url = self._build_url("birth-angels")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_tikkun(self, request: TikkunRequest) -> GenericResponse:
        """Get Tikkun analysis."""
        validate_datetime_location(request.birth_data)
        url = self._build_url("tikkun")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_gematria(self, request: GematriaRequest) -> GenericResponse:
        """Calculate Gematria."""
        url = self._build_url("gematria")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_sephiroth_glossary(self, system: str | None = None) -> GenericResponse:
        """Get Sephiroth glossary."""
        url = self._build_url("glossary", "sephiroth")
        params: dict[str, str] = {}
        if system:
            params["system"] = system
        data = self._http.get(url, params=params or None)
        return GenericResponse(**data)

    def get_hebrew_letters_glossary(self) -> GenericResponse:
        """Get Hebrew letters glossary."""
        url = self._build_url("glossary", "hebrew-letters")
        data = self._http.get(url)
        return GenericResponse(**data)

    def get_angels_glossary(self) -> GenericResponse:
        """Get 72 angels glossary."""
        url = self._build_url("glossary", "angels-72")
        data = self._http.get(url)
        return GenericResponse(**data)