from typing import Optional
from pydantic import BaseModel
from datetime import datetime
from app.models.movement import MovementType


class MovementBase(BaseModel):
    movement_type: MovementType
    aircraft_registration: str
    aircraft_type: Optional[str] = None
    callsign: Optional[str] = None
    timestamp: datetime
    entity_type: str
    entity_id: int
    to_location: Optional[str] = None
    from_location: Optional[str] = None
    runway: Optional[str] = None
    wind: Optional[str] = None
    pressure_setting: Optional[str] = None
    created_by: Optional[str] = None
    ip_address: Optional[str] = None
    notes: Optional[str] = None


class MovementCreate(MovementBase):
    pass


class Movement(MovementBase):
    id: int
    created_at: datetime

    class Config:
        from_attributes = True