Merge pull request #37 from endor-force/models

Add dataclass models

pytrafikverket/models.py

@@ -0,0 +1,111 @@
+"""Data classes for Trafikverket."""
+from __future__ import annotations
+
+from dataclasses import dataclass
+from datetime import datetime
+
+
+@dataclass
+class CameraInfoModel:
+    """Dataclass for Trafikverket Camera."""
+
+    camera_name: str | None
+    camera_id: str | None
+    active: bool
+    deleted: bool
+    description: str | None
+    direction: str | None
+    fullsizephoto: bool
+    location: str | None
+    modified: datetime | None
+    phototime: datetime | None
+    photourl: str | None
+    status: str | None
+    camera_type: str | None
+
+
+@dataclass
+class FerryRouteInfoModel:
+    """Dataclass for Trafikverket Ferry route info."""
+
+    ferry_route_id: str | None
+    name: str | None
+    short_name: str | None
+    route_type: str | None
+
+
+@dataclass
+class DeviationInfoModel:
+    """Dataclass for Trafikverket Ferry deviations."""
+
+    deviation_id: str | None
+    header: str | None
+    message: str | None
+    start_time: datetime | None
+    end_time: datetime | None
+    icon_id: str | None
+    location_desc: str | None
+
+
+@dataclass
+class FerryStopModel:
+    """Dataclass for Trafikverket Ferry stop."""
+
+    ferry_stop_id: str | None
+    deleted: bool
+    departure_time: datetime | None
+    other_information: list[str] | None
+    deviation_id: list[str] | None
+    modified_time: datetime | None
+    from_harbor_name: str | None
+    to_harbor_name: str | None
+
+
+@dataclass
+class StationInfoModel:
+    """Dataclass for Trafikverket Train Station info."""
+
+    signature: str | None
+    name: str | None
+    advertised: str | None
+
+
+@dataclass
+class TrainStopModel:
+    """Dataclass for Trafikverket Train stop."""
+
+    train_stop_id: str | None
+    canceled: bool
+    advertised_time_at_location: datetime | None
+    estimated_time_at_location: datetime | None
+    time_at_location: datetime | None
+    other_information: list[str] | None
+    deviations: list[str] | None
+    modified_time: datetime | None
+    product_description: list[str] | None
+
+
+@dataclass
+class WeatherStationInfoModel:
+    """Dataclass for Trafikverket Weather info."""
+
+    weather_station_id: str
+    station_name: str | None
+    station_id: str | None
+    road_temp: float | None
+    air_temp: float | None
+    humidity: float | None
+    precipitationtype: str | None
+    precipitationtype_translated: str | None
+    winddirection: str | None
+    winddirectiontext: str | None
+    winddirectiontext_translated: str | None
+    windforce: float | None
+    windforcemax: float | None
+    active: bool
+    deleted: bool
+    measure_time: datetime | None
+    precipitation_amount: float | None
+    precipitation_amountname: str | None
+    precipitation_amountname_translated: str | None
+    modified_time: datetime | None