horoscope.py

"""Horoscope category client."""

from __future__ import annotations

from astroapi.categories.base import BaseCategoryClient
from astroapi.types.requests import (
    ChineseBaziHoroscopeRequest,
    PersonalizedHoroscopeRequest,
    PersonalTextHoroscopeRequest,
    SignTextHoroscopeRequest,
    SunSignHoroscopeRequest,
)
from astroapi.types.responses import GenericResponse, HoroscopeResponse
from astroapi.utils.validators import validate_subject


class HoroscopeClient(BaseCategoryClient):
    """Client for horoscope endpoints.

    Provides personalized and sun sign horoscopes for various periods.
    """

    API_PREFIX = "/api/v3/horoscope"

    def get_personal_daily_horoscope(
        self, request: PersonalizedHoroscopeRequest
    ) -> HoroscopeResponse:
        """Get personalized daily horoscope.

        Args:
            request: Horoscope request parameters

        Returns:
            Daily horoscope response

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

        url = self._build_url("personal", "daily")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return HoroscopeResponse(**data)

    def get_personal_weekly_horoscope(
        self, request: PersonalizedHoroscopeRequest
    ) -> HoroscopeResponse:
        """Get personalized weekly horoscope.

        Args:
            request: Horoscope request parameters

        Returns:
            Weekly horoscope response

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

        url = self._build_url("personal", "weekly")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return HoroscopeResponse(**data)

    def get_personal_monthly_horoscope(
        self, request: PersonalizedHoroscopeRequest
    ) -> HoroscopeResponse:
        """Get personalized monthly horoscope.

        Args:
            request: Horoscope request parameters

        Returns:
            Monthly horoscope response

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

        url = self._build_url("personal", "monthly")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return HoroscopeResponse(**data)

    def get_personal_yearly_horoscope(
        self, request: PersonalizedHoroscopeRequest
    ) -> HoroscopeResponse:
        """Get personalized yearly horoscope.

        Args:
            request: Horoscope request parameters

        Returns:
            Yearly horoscope response

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

        url = self._build_url("personal", "yearly")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return HoroscopeResponse(**data)

    def get_sign_daily_horoscope(self, request: SunSignHoroscopeRequest) -> HoroscopeResponse:
        """Get sun sign daily horoscope.

        Args:
            request: Sun sign horoscope request parameters

        Returns:
            Daily horoscope response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        url = self._build_url("sign", "daily")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return HoroscopeResponse(**data)

    def get_sign_weekly_horoscope(self, request: SunSignHoroscopeRequest) -> HoroscopeResponse:
        """Get sun sign weekly horoscope.

        Args:
            request: Sun sign horoscope request parameters

        Returns:
            Weekly horoscope response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        url = self._build_url("sign", "weekly")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return HoroscopeResponse(**data)

    def get_sign_monthly_horoscope(self, request: SunSignHoroscopeRequest) -> HoroscopeResponse:
        """Get sun sign monthly horoscope.

        Args:
            request: Sun sign horoscope request parameters

        Returns:
            Monthly horoscope response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        url = self._build_url("sign", "monthly")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return HoroscopeResponse(**data)

    def get_sign_yearly_horoscope(self, request: SunSignHoroscopeRequest) -> HoroscopeResponse:
        """Get sun sign yearly horoscope.

        Args:
            request: Sun sign horoscope request parameters

        Returns:
            Yearly horoscope response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        url = self._build_url("sign", "yearly")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return HoroscopeResponse(**data)

    def get_sign_daily_text(self, request: SignTextHoroscopeRequest) -> GenericResponse:
        """Get sun sign daily horoscope as text."""
        url = self._build_url("sign", "daily", "text")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_sign_weekly_text(self, request: SignTextHoroscopeRequest) -> GenericResponse:
        """Get sun sign weekly horoscope as text."""
        url = self._build_url("sign", "weekly", "text")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_sign_monthly_text(self, request: SignTextHoroscopeRequest) -> GenericResponse:
        """Get sun sign monthly horoscope as text."""
        url = self._build_url("sign", "monthly", "text")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_sign_yearly_text(self, request: SignTextHoroscopeRequest) -> GenericResponse:
        """Get sun sign yearly horoscope as text."""
        url = self._build_url("sign", "yearly", "text")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_personal_daily_text(self, request: PersonalTextHoroscopeRequest) -> GenericResponse:
        """Get personalized daily horoscope as text."""
        validate_subject(request.subject)
        url = self._build_url("personal", "daily", "text")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_personal_weekly_text(self, request: PersonalTextHoroscopeRequest) -> GenericResponse:
        """Get personalized weekly horoscope as text."""
        validate_subject(request.subject)
        url = self._build_url("personal", "weekly", "text")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_personal_monthly_text(self, request: PersonalTextHoroscopeRequest) -> GenericResponse:
        """Get personalized monthly horoscope as text."""
        validate_subject(request.subject)
        url = self._build_url("personal", "monthly", "text")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_personal_yearly_text(self, request: PersonalTextHoroscopeRequest) -> GenericResponse:
        """Get personalized yearly horoscope as text."""
        validate_subject(request.subject)
        url = self._build_url("personal", "yearly", "text")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def get_chinese_bazi_horoscope(self, request: ChineseBaziHoroscopeRequest) -> GenericResponse:
        """Get Chinese BaZi horoscope."""
        validate_subject(request.subject)
        url = self._build_url("chinese", "bazi")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)