forked from sagnik/Project_Velocity
sayan
eeb684b46c
#38 Ipad app production readiness, Colony orchestration, Social posting Co-authored-by: Sayan Datta <sayan@Sayans-MacBook-Air.local> Reviewed-on: sagnik/Project_Velocity#44
51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from typing import Any
|
|
|
|
import httpx
|
|
|
|
|
|
class ColonyConfigurationError(RuntimeError):
|
|
pass
|
|
|
|
|
|
class ColonyGatewayError(RuntimeError):
|
|
pass
|
|
|
|
|
|
class ColonyGateway:
|
|
"""HTTP client from the Python root to the internal colony orchestrator."""
|
|
|
|
def __init__(self, *, base_url: str | None = None, timeout_s: float | None = None) -> None:
|
|
resolved_base_url = (base_url or os.getenv("COLONY_SERVICE_URL", "")).strip().rstrip("/")
|
|
if not resolved_base_url:
|
|
raise ColonyConfigurationError("COLONY_SERVICE_URL is not configured.")
|
|
self.base_url = resolved_base_url
|
|
self.timeout = httpx.Timeout(timeout_s or float(os.getenv("COLONY_TIMEOUT_SECONDS", "30")))
|
|
|
|
async def health(self) -> dict[str, Any]:
|
|
async with httpx.AsyncClient(timeout=httpx.Timeout(5.0)) as client:
|
|
response = await client.get(f"{self.base_url}/health")
|
|
if response.status_code >= 400:
|
|
raise ColonyGatewayError(f"Colony service health check failed with HTTP {response.status_code}.")
|
|
return response.json()
|
|
|
|
async def dispatch_mission(self, mission: dict[str, Any]) -> dict[str, Any]:
|
|
async with httpx.AsyncClient(timeout=self.timeout) as client:
|
|
response = await client.post(f"{self.base_url}/missions", json=mission)
|
|
if response.status_code >= 400:
|
|
raise ColonyGatewayError(
|
|
f"Colony service rejected mission with HTTP {response.status_code}: {response.text}"
|
|
)
|
|
return response.json()
|
|
|
|
async def mission_status(self, mission_id: str) -> dict[str, Any]:
|
|
async with httpx.AsyncClient(timeout=self.timeout) as client:
|
|
response = await client.get(f"{self.base_url}/missions/{mission_id}/status")
|
|
if response.status_code >= 400:
|
|
raise ColonyGatewayError(
|
|
f"Colony service status check failed with HTTP {response.status_code}: {response.text}"
|
|
)
|
|
return response.json()
|