extends flight init parameters

Author: GabrielBarberini

Makefile

@@ -8,3 +8,14 @@ flake8:
 
 pylint:
 	pylint --extension-pkg-whitelist='pydantic' ./lib/*
+
+dev:
+	python3 -m uvicorn lib:app --reload --port 3000
+
+clean:
+	docker stop infinity-api
+	docker rm infinity-api
+	docker system prune -fa
+
+build:
+	docker build -t infinity-api . --no-cache

lib/models/flight.py

@@ -1,11 +1,31 @@
+from enum import Enum
+from typing import Optional
 from pydantic import BaseModel
 from lib.models.rocket import Rocket
 from lib.models.environment import Env
 
 
+class EquationsOfMotion(str, Enum):
+    STANDARD = "STANDARD"
+    SOLID_PROPULSION = "SOLID_PROPULSION"
+
+
 class Flight(BaseModel):
+    name: str = "Flight"
     environment: Env = Env()
     rocket: Rocket = Rocket()
-    inclination: int = 85
-    heading: int = 0
     rail_length: float = 5.2
+    inclination: Optional[int] = 80.0
+    heading: Optional[int] = 90.0
+    # TODO: implement initial_solution
+    terminate_on_apogee: Optional[bool] = False
+    max_time: Optional[int] = 600
+    max_time_step: Optional[float] = 9999
+    min_time_step: Optional[int] = 0
+    rtol: Optional[float] = 1e-3
+    atol: Optional[float] = 1e-3
+    time_overshoot: Optional[bool] = True
+    verbose: Optional[bool] = False
+    equations_of_motion: Optional[EquationsOfMotion] = (
+        EquationsOfMotion.STANDARD
+    )

lib/services/flight.py

@@ -35,6 +35,16 @@ def from_flight_model(cls, flight: Flight) -> Self:
             heading=flight.heading,
             environment=rocketpy_env,
             rail_length=flight.rail_length,
+            #   initial_solution=flight.initial_solution,
+            terminate_on_apogee=flight.terminate_on_apogee,
+            max_time=flight.max_time,
+            max_time_step=flight.max_time_step,
+            min_time_step=flight.min_time_step,
+            rtol=flight.rtol,
+            atol=flight.atol,
+            time_overshoot=flight.time_overshoot,
+            verbose=flight.verbose,
+            equations_of_motion=flight.equations_of_motion.value.lower(),
         )
         return cls(flight=rocketpy_flight)