Upload schemas.py

schemas.py

@@ -0,0 +1,35 @@
+from pydantic import BaseModel, Field, field_validator
+from typing import Optional, Literal
+
+class TakeoffRequest(BaseModel):
+    altitude_ft: Optional[float] = Field(None, description="Indicated altitude in feet")
+    qnh_hpa: Optional[float] = Field(1013.25, description="QNH pressure setting in hPa")
+    temperature_c: Optional[float] = Field(None, description="Outside air temperature in Celsius")
+    weight_kg: Optional[float] = Field(None, description="Aircraft mass in kg")
+    wind_type: Literal["Headwind", "Tailwind"] = Field("Headwind", description="Direction of wind component")
+    wind_speed_kt: Optional[float] = Field(None, description="Wind speed in knots")
+    safety_factor: float = Field(1.0, description="Safety factor multiplier (usually 1.0 to 1.5)")
+
+    @field_validator('weight_kg')
+    def check_weight(cls, v):
+        if v is not None and (v < 800 or v > 1500):
+             # Soft warning logic could go here, but for now we just validate bounds roughly
+             pass 
+        return v
+        
+    def is_complete(self) -> bool:
+        """Checks if all mandatory fields are present."""
+        return all(x is not None for x in [
+            self.altitude_ft, 
+            self.temperature_c, 
+            self.weight_kg, 
+            self.wind_speed_kt
+        ])
+    
+    def get_missing_fields(self) -> list[str]:
+        missing = []
+        if self.altitude_ft is None: missing.append("Altitude")
+        if self.temperature_c is None: missing.append("Temperature")
+        if self.weight_kg is None: missing.append("Aircraft Weight")
+        if self.wind_speed_kt is None: missing.append("Wind Speed")
+        return missing