environment.py

from typing import Optional, Any
from datetime import datetime, timezone, timedelta
from pydantic import ConfigDict, Field
from src.views.interface import ApiBaseView
from src.models.environment import EnvironmentModel


def _default_future_datetime() -> datetime:
    """Factory function to create timezone-aware datetime one day in the future."""
    return datetime.now(timezone.utc) + timedelta(days=1)


class EnvironmentSimulation(ApiBaseView):
    """
    Environment simulation view that handles dynamically
    encoded RocketPy Environment attributes.

    Uses the new rocketpy_encoder which may return
    different attributes based on the actual RocketPy Environment object.
    The model allows extra fields to accommodate
    any new attributes that might be encoded.
    """

    model_config = ConfigDict(
        ser_json_exclude_none=True, extra='allow', arbitrary_types_allowed=True
    )

    message: str = "Environment successfully simulated"

    # Core Environment attributes (always present)
    latitude: Optional[float] = None
    longitude: Optional[float] = None
    elevation: Optional[float] = 1
    atmospheric_model_type: Optional[str] = None
    air_gas_constant: Optional[float] = None
    standard_g: Optional[float] = None
    earth_radius: Optional[float] = None
    datum: Optional[str] = None
    timezone: Optional[str] = None
    initial_utm_zone: Optional[int] = None
    initial_utm_letter: Optional[str] = None
    initial_north: Optional[float] = None
    initial_east: Optional[float] = None
    initial_hemisphere: Optional[str] = None
    initial_ew: Optional[str] = None
    max_expected_height: Optional[float] = None
    date: Optional[datetime] = Field(default_factory=_default_future_datetime)
    local_date: Optional[datetime] = Field(
        default_factory=_default_future_datetime
    )
    datetime_date: Optional[datetime] = Field(
        default_factory=_default_future_datetime
    )

    # Function attributes
    # discretized by rocketpy_encoder
    # serialized by RocketPyEncoder
    ellipsoid: Optional[Any] = None
    barometric_height: Optional[Any] = None
    barometric_height_ISA: Optional[Any] = None
    pressure: Optional[Any] = None
    pressure_ISA: Optional[Any] = None
    temperature: Optional[Any] = None
    temperature_ISA: Optional[Any] = None
    density: Optional[Any] = None
    speed_of_sound: Optional[Any] = None
    dynamic_viscosity: Optional[Any] = None
    gravity: Optional[Any] = None
    somigliana_gravity: Optional[Any] = None
    wind_speed: Optional[Any] = None
    wind_direction: Optional[Any] = None
    wind_heading: Optional[Any] = None
    wind_velocity_x: Optional[Any] = None
    wind_velocity_y: Optional[Any] = None
    calculate_earth_radius: Optional[Any] = None
    decimal_degrees_to_arc_seconds: Optional[Any] = None
    geodesic_to_utm: Optional[Any] = None
    utm_to_geodesic: Optional[Any] = None


class EnvironmentView(EnvironmentModel):
    environment_id: Optional[str] = None


class EnvironmentCreated(ApiBaseView):
    message: str = "Environment successfully created"
    environment_id: str


class EnvironmentRetrieved(ApiBaseView):
    message: str = "Environment successfully retrieved"
    environment: EnvironmentView