Merge pull request #55 from RocketPy-Team/enh/environment-models

DEV: custom atmosphere schema changes

Author: aasitvora99

src/models/environment.py

@@ -9,6 +9,10 @@ class EnvironmentModel(ApiBaseModel):
     latitude: float
     longitude: float
     elevation: Optional[float] = 0.0
+    pressure: Optional[float | list[tuple[float, float]]] = None
+    temperature: Optional[float | list[tuple[float, float]]] = None
+    wind_u: Optional[float | list[tuple[float, float]]] = None
+    wind_v: Optional[float | list[tuple[float, float]]] = None
 
     # Optional parameters
     atmospheric_model_type: Literal[

src/services/environment.py

@@ -31,6 +31,10 @@ def from_env_model(cls, env: EnvironmentModel) -> Self:
         rocketpy_env.set_atmospheric_model(
             type=env.atmospheric_model_type,
             file=env.atmospheric_model_file,
+            pressure=env.pressure,
+            temperature=env.temperature,
+            wind_u=env.wind_u,
+            wind_v=env.wind_v,
         )
         return cls(environment=rocketpy_env)
 

src/views/environment.py

@@ -22,7 +22,7 @@ class EnvironmentSimulation(ApiBaseView):
     initial_east: Optional[float] = None
     initial_hemisphere: Optional[str] = None
     initial_ew: Optional[str] = None
-    max_expected_height: Optional[int] = None
+    max_expected_height: Optional[float] = None
     date: Optional[datetime] = datetime.today() + timedelta(days=1)
     local_date: Optional[datetime] = datetime.today() + timedelta(days=1)
     datetime_date: Optional[datetime] = datetime.today() + timedelta(days=1)