render.py

"""Render category client for chart image rendering."""

from __future__ import annotations

from astroapi.categories.base import BaseCategoryClient
from astroapi.types.requests import (
    RenderCompositeRequest,
    RenderNatalRequest,
    RenderSynastryRequest,
    RenderTransitRequest,
)
from astroapi.types.responses import GenericResponse
from astroapi.utils.validators import validate_datetime_location, validate_subject


class RenderClient(BaseCategoryClient):
    """Client for chart rendering endpoints.

    Renders charts in various image formats (SVG, PNG, JPG, WebP, PDF).
    Replaces the deprecated SVG endpoints.
    """

    API_PREFIX = "/api/v3/render"

    def render_natal(self, request: RenderNatalRequest) -> GenericResponse:
        """Render natal chart."""
        validate_subject(request.subject)
        url = self._build_url("natal")
        data = self._http.post(url, json=request.model_dump(exclude_none=True))
        return GenericResponse(**data)

    def render_synastry(self, request: RenderSynastryRequest) -> GenericResponse:
        """Render synastry chart."""
        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 GenericResponse(**data)

    def render_composite(self, request: RenderCompositeRequest) -> GenericResponse:
        """Render composite chart."""
        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 GenericResponse(**data)

    def render_transit(self, request: RenderTransitRequest) -> GenericResponse:
        """Render transit chart."""
        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 GenericResponse(**data)