analysis.py

"""Analysis category client for interpretations and reports."""

from __future__ import annotations

from astroapi.categories.base import BaseCategoryClient
from astroapi.types.requests import (
    AnalysisRequest,
    CompositeAnalysisRequest,
    DirectionReportRequest,
    LunarAnalysisRequest,
    LunarReturnReportRequest,
    NatalTransitReportRequest,
    ProgressionReportRequest,
    RelocationAnalysisRequest,
    ReturnTransitReportRequest,
    SolarReturnReportRequest,
    SynastryAnalysisRequest,
    TransitAnalysisRequest,
    VenusReturnReportRequest,
)
from astroapi.types.responses import AnalysisResponse, GenericResponse
from astroapi.utils.validators import validate_datetime_location, validate_subject


class AnalysisClient(BaseCategoryClient):
    """Client for analysis and interpretation endpoints.

    Provides natal, synastry, composite, and transit interpretations and reports.
    """

    API_PREFIX = "/api/v3/analysis"

    def get_natal_report(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get natal chart interpretation report.

        Args:
            request: Analysis request parameters

        Returns:
            Natal report 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 AnalysisResponse(**data)

    def get_natal_planets_report(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get natal planets interpretation report.

        Args:
            request: Analysis request parameters

        Returns:
            Planets report response

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

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

    def get_natal_houses_report(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get natal houses interpretation report.

        Args:
            request: Analysis request parameters

        Returns:
            Houses report response

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

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

    def get_natal_aspects_report(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get natal aspects interpretation report.

        Args:
            request: Analysis request parameters

        Returns:
            Aspects report response

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

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

    def get_synastry_report(self, request: SynastryAnalysisRequest) -> AnalysisResponse:
        """Get synastry interpretation report.

        Args:
            request: Synastry analysis request parameters

        Returns:
            Synastry report response

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

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

    def get_synastry_aspects_report(self, request: SynastryAnalysisRequest) -> AnalysisResponse:
        """Get synastry aspects interpretation report.

        Args:
            request: Synastry analysis request parameters

        Returns:
            Synastry aspects report response

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

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

    def get_composite_report(self, request: CompositeAnalysisRequest) -> AnalysisResponse:
        """Get composite chart interpretation report.

        Args:
            request: Composite analysis request parameters

        Returns:
            Composite report response

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

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

    def get_transit_report(self, request: TransitAnalysisRequest) -> AnalysisResponse:
        """Get transit interpretation report.

        Args:
            request: Transit analysis request parameters

        Returns:
            Transit report response

        Raises:
            AstrologyError: If validation fails or API request fails
        """
        validate_subject(request.natal_subject)
        validate_datetime_location(request.transit_datetime)

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

    def get_compatibility_score(self, request: SynastryAnalysisRequest) -> AnalysisResponse:
        """Get compatibility score analysis.

        Args:
            request: Synastry analysis request parameters

        Returns:
            Compatibility score response

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

        url = self._build_url("compatibility", "score")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_compatibility_detailed(self, request: SynastryAnalysisRequest) -> AnalysisResponse:
        """Get detailed compatibility analysis.

        Args:
            request: Synastry analysis request parameters

        Returns:
            Detailed compatibility response

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

        url = self._build_url("compatibility", "detailed")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_career_analysis(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get career analysis report."""
        validate_subject(request.subject)
        url = self._build_url("career")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_health_analysis(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get health analysis report."""
        validate_subject(request.subject)
        url = self._build_url("health")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_karmic_analysis(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get karmic analysis report."""
        validate_subject(request.subject)
        url = self._build_url("karmic")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_psychological_analysis(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get psychological analysis report."""
        validate_subject(request.subject)
        url = self._build_url("psychological")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_spiritual_analysis(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get spiritual analysis report."""
        validate_subject(request.subject)
        url = self._build_url("spiritual")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_vocational_analysis(self, request: AnalysisRequest) -> AnalysisResponse:
        """Get vocational analysis report."""
        validate_subject(request.subject)
        url = self._build_url("vocational")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_relationship_analysis(self, request: SynastryAnalysisRequest) -> AnalysisResponse:
        """Get relationship analysis report."""
        validate_subject(request.subject1)
        validate_subject(request.subject2)
        url = self._build_url("relationship")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_predictive_analysis(self, request: NatalTransitReportRequest) -> AnalysisResponse:
        """Get predictive analysis report."""
        validate_subject(request.natal_subject)
        url = self._build_url("predictive")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_relocation_analysis(self, request: RelocationAnalysisRequest) -> AnalysisResponse:
        """Get relocation analysis report."""
        validate_subject(request.subject)
        url = self._build_url("relocation")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_lunar_analysis(self, request: LunarAnalysisRequest) -> AnalysisResponse:
        """Get lunar analysis report."""
        validate_datetime_location(request.datetime_location)
        url = self._build_url("lunar-analysis")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_relationship_score(self, request: SynastryAnalysisRequest) -> AnalysisResponse:
        """Get relationship score."""
        validate_subject(request.subject1)
        validate_subject(request.subject2)
        url = self._build_url("relationship-score")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_solar_return_report(self, request: SolarReturnReportRequest) -> AnalysisResponse:
        """Get solar return report."""
        validate_subject(request.subject)
        url = self._build_url("solar-return-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_lunar_return_report(self, request: LunarReturnReportRequest) -> AnalysisResponse:
        """Get lunar return report."""
        validate_subject(request.subject)
        url = self._build_url("lunar-return-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_venus_return_report(self, request: VenusReturnReportRequest) -> AnalysisResponse:
        """Get Venus return report."""
        validate_subject(request.subject)
        url = self._build_url("venus-return-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_solar_return_transit_report(
        self, request: ReturnTransitReportRequest
    ) -> AnalysisResponse:
        """Get solar return transit report."""
        validate_subject(request.subject)
        url = self._build_url("solar-return-transit-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_lunar_return_transit_report(
        self, request: ReturnTransitReportRequest
    ) -> AnalysisResponse:
        """Get lunar return transit report."""
        validate_subject(request.subject)
        url = self._build_url("lunar-return-transit-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_venus_return_transit_report(
        self, request: ReturnTransitReportRequest
    ) -> AnalysisResponse:
        """Get Venus return transit report."""
        validate_subject(request.subject)
        url = self._build_url("venus-return-transit-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_direction_report(self, request: DirectionReportRequest) -> AnalysisResponse:
        """Get direction report."""
        validate_subject(request.subject)
        url = self._build_url("direction-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_progression_report(self, request: ProgressionReportRequest) -> AnalysisResponse:
        """Get progression report."""
        validate_subject(request.subject)
        url = self._build_url("progression-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return AnalysisResponse(**data)

    def get_natal_transit_report(self, request: NatalTransitReportRequest) -> GenericResponse:
        """Get natal transit report."""
        validate_subject(request.natal_subject)
        url = self._build_url("natal-transit-report")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)