170 lines
4.5 KiB
Python
170 lines
4.5 KiB
Python
import datetime
|
|
from typing import Optional, Dict, Union, List, get_args
|
|
|
|
from utils import to_objectid
|
|
from pydantic import BaseModel
|
|
|
|
from schemas.utils import PositiveFloatNullable, PositiveFloat, PositiveInt, PyObjectId
|
|
|
|
|
|
class FlightSchema(BaseModel):
|
|
date: datetime.datetime
|
|
aircraft: str
|
|
waypoint_from: Optional[str] = None
|
|
waypoint_to: Optional[str] = None
|
|
route: Optional[str] = None
|
|
|
|
hobbs_start: Optional[PositiveFloatNullable] = None
|
|
hobbs_end: Optional[PositiveFloatNullable] = None
|
|
|
|
time_start: Optional[datetime.datetime] = None
|
|
time_off: Optional[datetime.datetime] = None
|
|
time_down: Optional[datetime.datetime] = None
|
|
time_stop: Optional[datetime.datetime] = None
|
|
|
|
time_total: PositiveFloat
|
|
time_pic: PositiveFloat
|
|
time_sic: PositiveFloat
|
|
time_night: PositiveFloat
|
|
time_solo: PositiveFloat
|
|
|
|
time_xc: PositiveFloat
|
|
dist_xc: PositiveFloat
|
|
|
|
landings_day: PositiveInt
|
|
landings_night: PositiveInt
|
|
|
|
time_instrument: PositiveFloat
|
|
time_sim_instrument: PositiveFloat
|
|
holds_instrument: PositiveInt
|
|
|
|
dual_given: PositiveFloat
|
|
dual_recvd: PositiveFloat
|
|
time_sim: PositiveFloat
|
|
time_ground: PositiveFloat
|
|
|
|
tags: List[str] = []
|
|
|
|
pax: List[str] = []
|
|
crew: List[str] = []
|
|
|
|
comments: Optional[str] = None
|
|
|
|
|
|
class FlightCreateSchema(FlightSchema):
|
|
images: List[str] = []
|
|
|
|
|
|
class FlightPatchSchema(BaseModel):
|
|
date: Optional[datetime.datetime] = None
|
|
aircraft: Optional[str] = None
|
|
waypoint_from: Optional[str] = None
|
|
waypoint_to: Optional[str] = None
|
|
route: Optional[str] = None
|
|
|
|
hobbs_start: Optional[PositiveFloatNullable] = None
|
|
hobbs_end: Optional[PositiveFloatNullable] = None
|
|
|
|
time_start: Optional[datetime.datetime] = None
|
|
time_off: Optional[datetime.datetime] = None
|
|
time_down: Optional[datetime.datetime] = None
|
|
time_stop: Optional[datetime.datetime] = None
|
|
|
|
time_total: Optional[PositiveFloat] = None
|
|
time_pic: Optional[PositiveFloat] = None
|
|
time_sic: Optional[PositiveFloat] = None
|
|
time_night: Optional[PositiveFloat] = None
|
|
time_solo: Optional[PositiveFloat] = None
|
|
|
|
time_xc: Optional[PositiveFloat] = None
|
|
dist_xc: Optional[PositiveFloat] = None
|
|
|
|
landings_day: Optional[PositiveInt] = None
|
|
landings_night: Optional[PositiveInt] = None
|
|
|
|
time_instrument: Optional[PositiveFloat] = None
|
|
time_sim_instrument: Optional[PositiveFloat] = None
|
|
holds_instrument: Optional[PositiveInt] = None
|
|
|
|
dual_given: Optional[PositiveFloat] = None
|
|
dual_recvd: Optional[PositiveFloat] = None
|
|
time_sim: Optional[PositiveFloat] = None
|
|
time_ground: Optional[PositiveFloat] = None
|
|
|
|
tags: Optional[List[str]] = None
|
|
|
|
pax: Optional[List[str]] = None
|
|
crew: Optional[List[str]] = None
|
|
|
|
images: Optional[List[str]] = None
|
|
|
|
comments: Optional[str] = None
|
|
|
|
|
|
class FlightDisplaySchema(FlightCreateSchema):
|
|
user: PyObjectId
|
|
id: PyObjectId
|
|
|
|
|
|
class FlightConciseSchema(BaseModel):
|
|
user: PyObjectId
|
|
id: PyObjectId
|
|
aircraft: str
|
|
|
|
date: datetime.date
|
|
aircraft: str
|
|
waypoint_from: Optional[str] = None
|
|
waypoint_to: Optional[str] = None
|
|
|
|
time_total: PositiveFloat
|
|
|
|
comments: Optional[str] = None
|
|
|
|
|
|
FlightByDateSchema = Dict[int, Union[Dict[int, 'FlightByDateSchema'], FlightConciseSchema]]
|
|
|
|
fs_keys = list(FlightPatchSchema.__annotations__.keys()) + list(FlightDisplaySchema.__annotations__.keys())
|
|
fs_types = {label: get_args(type_)[0] if get_args(type_) else str(type_) for label, type_ in
|
|
FlightSchema.__annotations__.items() if len(get_args(type_)) > 0}
|
|
|
|
|
|
# HELPERS #
|
|
|
|
def flight_display_helper(flight: dict) -> dict:
|
|
"""
|
|
Convert given db response to a format usable by FlightDisplaySchema
|
|
|
|
:param flight: Database response
|
|
:return: Usable dict
|
|
"""
|
|
flight["id"] = str(flight["_id"])
|
|
flight["user"] = str(flight["user"])
|
|
|
|
return flight
|
|
|
|
|
|
async def flight_concise_helper(flight: dict) -> dict:
|
|
"""
|
|
Convert given db response to a format usable by FlightConciseSchema
|
|
|
|
:param flight: Database response
|
|
:return: Usable dict
|
|
"""
|
|
flight["id"] = str(flight["_id"])
|
|
flight["user"] = str(flight["user"])
|
|
|
|
return flight
|
|
|
|
|
|
def flight_add_helper(flight: dict, user: str) -> dict:
|
|
"""
|
|
Convert given flight schema and user string to a format that can be inserted into the db
|
|
|
|
:param flight: Flight request body
|
|
:param user: User that created flight
|
|
:return: Combined dict that can be inserted into db
|
|
"""
|
|
flight["user"] = to_objectid(user)
|
|
|
|
return flight
|