mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
b7439cbd17
operation_ids give us better function names when generating the typescript API from the openapi.json. BaseModel.Config.title does the same for type names. Response models (or 204 status codes) need to be explicit or the API will be declared as returning any.
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
from uuid import UUID
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from game.ato.flightstate import InFlight
|
|
from game.server.leaflet import LeafletPoint
|
|
from game.server.waypoints.models import FlightWaypointJs
|
|
from game.server.waypoints.routes import waypoints_for_flight
|
|
|
|
if TYPE_CHECKING:
|
|
from game import Game
|
|
from game.ato import Flight
|
|
|
|
|
|
class FlightJs(BaseModel):
|
|
id: UUID
|
|
blue: bool
|
|
position: LeafletPoint | None
|
|
sidc: str
|
|
waypoints: list[FlightWaypointJs] | None
|
|
|
|
class Config:
|
|
title = "Flight"
|
|
|
|
@staticmethod
|
|
def for_flight(flight: Flight, with_waypoints: bool) -> FlightJs:
|
|
# Don't provide a location for aircraft that aren't in the air. Later we can
|
|
# expand the model to include the state data for the UI so that it can make its
|
|
# own decisions about whether or not to draw the aircraft, but for now we'll
|
|
# filter here.
|
|
position = None
|
|
if isinstance(flight.state, InFlight):
|
|
position = flight.position().latlng()
|
|
waypoints = None
|
|
if with_waypoints:
|
|
waypoints = waypoints_for_flight(flight)
|
|
return FlightJs(
|
|
id=flight.id,
|
|
blue=flight.blue,
|
|
position=position,
|
|
sidc=str(flight.sidc()),
|
|
waypoints=waypoints,
|
|
)
|
|
|
|
@staticmethod
|
|
def all_in_game(game: Game, with_waypoints: bool) -> list[FlightJs]:
|
|
flights = []
|
|
for coalition in game.coalitions:
|
|
for package in coalition.ato.packages:
|
|
for flight in package.flights:
|
|
flights.append(FlightJs.for_flight(flight, with_waypoints))
|
|
return flights
|