junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-30 13:10:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Expose get_plan_raw method on MagicPlanClient 🟩

Browse source
This commit is contained in:
Daniel Roth 2026-05-08 14:09:33 +00:00
parent c3aae8fd51
commit f6c17be70a
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
backend/magic_plan/magic_plan_client.py
View file

@ -17,11 +17,14 @@ class MagicPlanClient:
return PlansListResponse.model_validate(r.json()["data"]) return PlansListResponse.model_validate(r.json()["data"])
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"])
def get_plan_raw(self, plan_id: str) -> bytes:
return self._fetch_plan(plan_id).content
def _fetch_plan(self, plan_id: str) -> requests.Response:
r = self._session.get( r = self._session.get(
f"{_BASE_URL}/plans/{plan_id}", params={"key": self._api_key} f"{_BASE_URL}/plans/{plan_id}", params={"key": self._api_key}
) )
r.raise_for_status() r.raise_for_status()
return MagicPlanPlan.model_validate(r.json()["data"]) return r
def get_plan_raw(self, plan_id: str) -> bytes:
raise NotImplementedError