Merge pull request #361 from planetarium/bugfix/mileage-schema

Re-use planet into mileage schema
planetarium · Dec 17, 2024 · d63cd25 · d63cd25
2 parents f433915 + be34b41
commit d63cd25
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/iap/api/mileage.py b/iap/api/mileage.py
@@ -14,4 +14,6 @@
 
 @router.get("", response_model=MileageSchema)
 def get_mileage(agent_addr: str, planet_id: str, sess: Session = Depends(session)):
-    return get_mileage_fn(sess, PlanetID(bytes(planet_id, "utf-8")), agent_addr)
+    mileage = get_mileage_fn(sess, agent_addr)
+    schema = MileageSchema(**mileage.__dict__, planet_id=PlanetID(bytes(planet_id, "utf-8")))
+    return schema
diff --git a/iap/schemas/mileage.py b/iap/schemas/mileage.py
@@ -22,6 +22,7 @@ def __post_init__(self):
 
 class MileageSchema(BaseSchema):
     id: int
+    planet_id: PlanetID
     agent_addr: str
     mileage: int