from pydantic import BaseModel

from ..idfm_interface import IdfmLineState, IdfmState, StopAreaType, TransportMode


class Stop(BaseModel):
    id: int
    name: str
    town: str
    lat: float
    lon: float
    # xepsg2154: int
    # yepsg2154: int
    lines: list[str]


class StopArea(BaseModel):
    id: int
    name: str
    town: str
    # xepsg2154: int
    # yepsg2154: int
    type: StopAreaType
    lines: list[str]  # SNCF lines are linked to stop areas and not stops.
    stops: list[Stop]