Add altitude calculations to LunarTwoResponse.

src/lct_web/models/lunar_two_response.py

@@ -14,7 +14,7 @@
 
 from pydantic import BaseModel
 
-from ..pkg_types import FeatureDict
+from ..pkg_types import AltitudeDict, FeatureDict
 
 __all__ = ["LunarTwoResponse"]
 
@@ -24,3 +24,4 @@ class LunarTwoResponse(BaseModel):
 
     features: FeatureDict
     landing_sites: FeatureDict
+    altitudes: AltitudeDict

src/lct_web/pkg_types.py

@@ -14,8 +14,9 @@
 
 from pylunar.pkg_types import DateTimeTuple, LunarFeatureList
 
-__all__ = ["NextFourPhases", "RiseSetTimes"]
+__all__ = ["AltitudeDict", "FeatureDict", "NextFourPhases", "RiseSetTimes"]
 
+AltitudeDict = dict[str, float]
 FeatureDict = dict[str, LunarFeatureList]
 NextFourPhases = dict[str, dict[str, str | DateTimeTuple]]
 RiseSetTimes = dict[str, dict[str, str | DateTimeTuple]]

src/lct_web/routers/lunar_two.py

@@ -13,7 +13,7 @@
 from __future__ import annotations
 
 from fastapi import APIRouter
-from pylunar import LunarFeatureContainer, MoonInfo
+from pylunar import AltitudeDict, LunarFeatureContainer, MoonInfo
 
 from ..dependencies import DateLocDeps
 from ..helpers import format_date_and_location
@@ -38,6 +38,9 @@ def lunar_two(params: DateLocDeps) -> LunarTwoResponse:
     lfc = LunarFeatureContainer("LunarII")
     lfc.load(moon_info)
 
+    ad = AltitudeDict()
+    ad.load(moon_info)
+
     features = {}
     landing_sites = {}
     for i, feature in enumerate(lfc):
@@ -46,4 +49,4 @@ def lunar_two(params: DateLocDeps) -> LunarTwoResponse:
         else:
             features[str(i)] = feature.list_from_feature()
 
-    return LunarTwoResponse(features=features, landing_sites=landing_sites)
+    return LunarTwoResponse(features=features, landing_sites=landing_sites, altitudes=ad)

tests/test_lunar_two.py

@@ -27,3 +27,4 @@ def test_lunar_two() -> None:
     lunar_two_info = response.json()
     assert len(lunar_two_info["features"]) == 11
     assert len(lunar_two_info["landing_sites"]) == 16
+    assert len(lunar_two_info["altitudes"]) == 4