data.py

"""Data category client for planetary positions and related data."""

from __future__ import annotations

from astroapi.categories.base import BaseCategoryClient
from astroapi.types.requests import DataRequest, DateTimeLocation, PlanetaryPositionsRequest
from astroapi.types.responses import GenericResponse, PlanetaryPositionsResponse
from astroapi.utils.validators import validate_datetime_location, validate_subject


class DataClient(BaseCategoryClient):
    """Client for data endpoints.

    Provides access to planetary positions, house cusps, aspects, and other
    astrological data calculations.
    """

    API_PREFIX = "/api/v3/data"

    def get_positions(self, request: PlanetaryPositionsRequest) -> PlanetaryPositionsResponse:
        """Get planetary positions for a given date/time/location.

        Args:
            request: Planetary positions request parameters

        Returns:
            Planetary positions response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        dtl = DateTimeLocation(
            year=request.year,
            month=request.month,
            day=request.day,
            hour=request.hour,
            minute=request.minute,
            second=request.second,
            latitude=request.latitude,
            longitude=request.longitude,
        )
        validate_datetime_location(dtl)

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

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

        Args:
            request: Planetary positions request parameters

        Returns:
            Enhanced planetary positions response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        dtl = DateTimeLocation(
            year=request.year,
            month=request.month,
            day=request.day,
            hour=request.hour,
            minute=request.minute,
            second=request.second,
            latitude=request.latitude,
            longitude=request.longitude,
        )
        validate_datetime_location(dtl)

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

    def get_houses(self, request: DataRequest) -> GenericResponse:
        """Get house cusps for a subject.

        Args:
            request: Data request with subject information

        Returns:
            House cusps response

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

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

    def get_aspects(self, request: DataRequest) -> GenericResponse:
        """Get aspects for a subject.

        Args:
            request: Data request with subject information

        Returns:
            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_midpoints(self, request: DataRequest) -> GenericResponse:
        """Get midpoints for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Midpoints response

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

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

    def get_arabic_parts(self, request: DataRequest) -> GenericResponse:
        """Get Arabic parts for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Arabic parts response

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

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

    def get_angles(self, request: DataRequest) -> GenericResponse:
        """Get chart angles (ASC, MC, etc.) for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Angles response

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

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

    def get_declinations(self, request: DataRequest) -> GenericResponse:
        """Get planetary declinations for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Declinations response

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

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

    def get_asteroids(self, request: DataRequest) -> GenericResponse:
        """Get asteroid positions for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Asteroid positions response

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

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

    def get_lunar_nodes(self, request: DataRequest) -> GenericResponse:
        """Get lunar nodes for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Lunar nodes response

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

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

    def get_vertex(self, request: DataRequest) -> GenericResponse:
        """Get vertex point for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Vertex response

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

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

    def get_lilith(self, request: DataRequest) -> GenericResponse:
        """Get Lilith positions for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Lilith positions response

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

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

    def get_chiron(self, request: DataRequest) -> GenericResponse:
        """Get Chiron position for a subject.

        Args:
            request: Data request with subject information

        Returns:
            Chiron position response

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

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