"""Klient HTTP do warstwy logicznej. Jedyny punkt styku prezentacji w dół. Przekazuje dane z formularza i odbiera opracowane wyniki. Prezentacja nie sięga bezpośrednio do bazy ani do silnika. """ from __future__ import annotations import os from typing import Any import httpx from app import link_crypto from app.config import settings def _auth_headers() -> dict[str, str]: """Token międzywarstwowy (LOG-32) — pusty, gdy ochrona wyłączona.""" token = os.getenv("INTERNAL_TOKEN", "") return {"X-Astrololo-Token": token} if token else {} def _link() -> link_crypto.Link | None: """Klucz łącza prezentacja↔logika. Czytany przy każdym wywołaniu, bo konfiguracja może się zmienić bez restartu procesu (testy, podmiana sekretu).""" key = link_crypto.key_from_env(link_crypto.ENV_PRESENTATION_LOGIC) return link_crypto.Link(key) if key else None class LogicClient: def __init__(self, base_url: str | None = None) -> None: self.base_url = (base_url or settings.logic_url).rstrip("/") def _post(self, path: str, payload: dict[str, Any], timeout: float) -> dict[str, Any]: """Jedyna droga w dół. Celowo JEDNA: dopóki każda metoda składała żądanie sama, dołożenie nowej znaczyło, że łatwo zapomnieć o tokenie albo kluczu łącza — i tak się już raz stało (401 wyszedł dopiero na produkcji).""" with httpx.Client(timeout=timeout) as client: return link_crypto.call_json(client, "POST", f"{self.base_url}{path}", payload=payload, headers=_auth_headers(), link=_link()) def query(self, query: str, field: str, exact: bool, limit: int) -> dict[str, Any]: payload = {"query": query, "field": field, "exact": exact, "limit": limit} return self._post("/api/query", payload, settings.http_timeout) def positions( self, when_utc_iso: str, lat: float, lon: float, objects: list[str] | None = None, house_system: str = "whole_sign", stations: bool = False, zodiac: str = "tropical", ) -> dict[str, Any]: """Pełny horoskop dla danego momentu — woła logic /chart/positions.""" payload = { "when_utc": when_utc_iso, "lat": lat, "lon": lon, "objects": objects, "house_system": house_system, "stations": stations, "zodiac": zodiac, } # stacje wymagają root-findów — dłuższy timeout timeout = max(settings.http_timeout, 60.0) if stations else settings.http_timeout return self._post("/chart/positions", payload, timeout) def report( self, when_utc_iso: str, lat: float, lon: float, limit: int = 5000, group: bool = False ) -> dict[str, Any]: """Sygnifikatory z obliczeń szukane w bazie — woła logic /chart/report.""" payload = {"when_utc": when_utc_iso, "lat": lat, "lon": lon, "limit": limit, "group": group} return self._post("/chart/report", payload, max(settings.http_timeout, 30.0)) def prompt( self, profile: str, when_utc_iso: str, lat: float, lon: float, budget: str = "medium", from_date: str | None = None, to_date: str | None = None, provider: str | None = None, model: str | None = None, ) -> dict[str, Any]: """Gotowy prompt do LLM z wyliczeń (LOG-29/30) — woła logic /chart/prompt. `provider` jest potrzebny dla budżetu „maksymalny kontekst modelu": limit znaków zależy wtedy od okna kontekstu konkretnego modelu.""" payload: dict[str, Any] = { "profile": profile, "when_utc": when_utc_iso, "lat": lat, "lon": lon, "budget": budget, "provider": provider, "model": model, } if from_date and to_date: payload["from_date"], payload["to_date"] = from_date, to_date return self._post("/chart/prompt", payload, max(settings.http_timeout, 60.0)) def horoscope( self, profile: str, when_utc_iso: str, lat: float, lon: float, budget: str = "medium", provider: str | None = None, model: str | None = None, from_date: str | None = None, to_date: str | None = None, ) -> dict[str, Any]: """Napisany horoskop (LOG-31) — woła logic /chart/horoscope. Generowanie tekstu trwa (zwłaszcza na modelu lokalnym), stąd długi timeout. """ payload: dict[str, Any] = { "profile": profile, "when_utc": when_utc_iso, "lat": lat, "lon": lon, "budget": budget, } if provider: payload["provider"] = provider if model: payload["model"] = model if from_date and to_date: payload["from_date"], payload["to_date"] = from_date, to_date return self._post("/chart/horoscope", payload, max(settings.http_timeout, 300.0)) def llm_models(self) -> dict[str, Any]: """Katalog modeli per dostawca (podpowiedzi do pola wyboru w UI).""" with httpx.Client(timeout=settings.http_timeout) as client: return link_crypto.call_json(client, "GET", f"{self.base_url}/llm/models", headers=_auth_headers(), link=_link()) def timeline( self, when_utc_iso: str, lat: float, lon: float, from_date: str, to_date: str, interpret: bool = True, ) -> dict[str, Any]: """Oś czasu z technik (+interpretacje z bazy) — woła logic /chart/timeline.""" payload = { "when_utc": when_utc_iso, "lat": lat, "lon": lon, "from_date": from_date, "to_date": to_date, "interpret": interpret, } return self._post("/chart/timeline", payload, max(settings.http_timeout, 60.0))