Paginate get_plans to return flat list[PlanSummary] 🟩

This commit is contained in:
Daniel Roth 2026-05-12 14:46:00 +00:00
parent 62acc3ce98
commit f83ddd05a8

View file

@ -11,7 +11,9 @@ class MagicPlanClient:
self._session.headers.update({"customer": customer_id, "key": api_key}) self._session.headers.update({"customer": customer_id, "key": api_key})
def get_plans(self) -> list[PlanSummary]: def get_plans(self) -> list[PlanSummary]:
raise NotImplementedError r = self._session.get(f"{_BASE_URL}/workgroups/plans", params={"page": 1})
r.raise_for_status()
return PlansListResponse.model_validate(r.json()["data"]).plans
def get_plan(self, plan_id: str) -> MagicPlanPlan: def get_plan(self, plan_id: str) -> MagicPlanPlan:
return MagicPlanPlan.model_validate(self._fetch_plan(plan_id).json()["data"]) return MagicPlanPlan.model_validate(self._fetch_plan(plan_id).json()["data"])