astroapi-python/src/astroapi/categories/enhanced.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
96
97
98
99
100
101
102
103
104
105
106
107
"""Enhanced category client with dual-prefix pattern."""

from __future__ import annotations

from astroapi.categories.base import BaseCategoryClient
from astroapi.types.requests import DataRequest, NatalChartRequest
from astroapi.types.responses import GenericResponse, NatalChartResponse
from astroapi.utils.validators import validate_subject


class EnhancedClient(BaseCategoryClient):
    """Client for enhanced endpoints.

    This client uses multiple API prefixes for different endpoint groups.
    """

    API_PREFIX = "/api/v3/enhanced"

    def get_enhanced_natal_chart(self, request: NatalChartRequest) -> NatalChartResponse:
        """Get enhanced natal chart with additional calculations.

        Args:
            request: Natal chart request parameters

        Returns:
            Enhanced natal chart response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        validate_subject(request.subject)

        url = self._build_url("natal")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return NatalChartResponse(**data)

    def get_enhanced_positions(self, request: DataRequest) -> GenericResponse:
        """Get enhanced planetary positions with additional details.

        Args:
            request: Data request with subject information

        Returns:
            Enhanced positions response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        validate_subject(request.subject)

        url = self._build_url("positions")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_enhanced_aspects(self, request: DataRequest) -> GenericResponse:
        """Get enhanced aspects with additional interpretations.

        Args:
            request: Data request with subject information

        Returns:
            Enhanced aspects response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        validate_subject(request.subject)

        url = self._build_url("aspects")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_patterns(self, request: DataRequest) -> GenericResponse:
        """Get chart patterns (grand trines, t-squares, etc.).

        Args:
            request: Data request with subject information

        Returns:
            Chart patterns response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        validate_subject(request.subject)

        url = self._build_url("patterns")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_dominants(self, request: DataRequest) -> GenericResponse:
        """Get dominant elements, modes, and planets.

        Args:
            request: Data request with subject information

        Returns:
            Dominants response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        validate_subject(request.subject)

        url = self._build_url("dominants")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)