astroapi-python/src/astroapi/categories/human_design.py at main · astro-api/astroapi-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
"""Human Design category client."""

from __future__ import annotations

from astroapi.categories.base import BaseCategoryClient
from astroapi.types.requests import (
    HumanDesignCompatibilityRequest,
    HumanDesignRequest,
    HumanDesignTransitRequest,
)
from astroapi.types.responses import GenericResponse
from astroapi.utils.validators import validate_subject


class HumanDesignClient(BaseCategoryClient):
    """Client for Human Design endpoints."""

    API_PREFIX = "/api/v3/human-design"

    def get_bodygraph(self, request: HumanDesignRequest) -> GenericResponse:
        """Calculate bodygraph."""
        validate_subject(request.subject)
        url = self._build_url("bodygraph")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_type(self, request: HumanDesignRequest) -> GenericResponse:
        """Get Human Design type only."""
        validate_subject(request.subject)
        url = self._build_url("type")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_transits(self, request: HumanDesignTransitRequest) -> GenericResponse:
        """Calculate transit overlay."""
        validate_subject(request.subject)
        url = self._build_url("transits")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_compatibility(self, request: HumanDesignCompatibilityRequest) -> GenericResponse:
        """Calculate Human Design compatibility."""
        for subject in request.subjects:
            validate_subject(subject)
        url = self._build_url("compatibility")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_design_date(self, request: HumanDesignRequest) -> GenericResponse:
        """Calculate design date."""
        validate_subject(request.subject)
        url = self._build_url("design-date")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_gates_glossary(
        self, language: str | None = None, center: str | None = None
    ) -> GenericResponse:
        """Get gates glossary."""
        url = self._build_url("glossary", "gates")
        params: dict[str, str] = {}
        if language:
            params["language"] = language
        if center:
            params["center"] = center
        data = self._http.get(url, params=params or None)
        if isinstance(data, list):
            return GenericResponse(data=data)
        return GenericResponse(**data)

    def get_channels_glossary(
        self, language: str | None = None, circuit: str | None = None
    ) -> GenericResponse:
        """Get channels glossary."""
        url = self._build_url("glossary", "channels")
        params: dict[str, str] = {}
        if language:
            params["language"] = language
        if circuit:
            params["circuit"] = circuit
        data = self._http.get(url, params=params or None)
        if isinstance(data, list):
            return GenericResponse(data=data)
        return GenericResponse(**data)

    def get_types_glossary(self, language: str | None = None) -> GenericResponse:
        """Get types glossary."""
        url = self._build_url("glossary", "types")
        params: dict[str, str] = {}
        if language:
            params["language"] = language
        data = self._http.get(url, params=params or None)
        if isinstance(data, list):
            return GenericResponse(data=data)
        return GenericResponse(**data)