Source code for amtraker_ingestion.chalicelib.constants

"""
Constants and Configuration Values
===================================

This module defines constants used throughout the Amtrak Ingestion system,
including API endpoints, S3 bucket names, file paths, and data schemas.

Attributes
----------
AMTRAKER_API : str
    URL for the Amtraker trains API endpoint.
S3_BUCKET : str
    Name of the S3 bucket for data storage.
CSV_FIELDS : list of str
    Field names for event CSV files.
EASTERN_TIME : ZoneInfo
    Eastern timezone for service date calculations.
"""

from zoneinfo import ZoneInfo
import pathlib
from enum import Enum




class Provider(str, Enum):
    """
    Enum of supported transit providers.

    Attributes
    ----------
    AMTRAK : str
        Amtrak national passenger railroad.
    VIA : str
        VIA Rail Canada.
    BRIGHTLINE : str
        Brightline Florida high-speed rail.
    """

    AMTRAK = "Amtrak"
    VIA = "VIA"
    BRIGHTLINE = "Brightline"

    def __str__(self):
        return self.value



AMTRAKER_API = "https://api-v3.amtraker.com/v3/trains"
AMTRAK_STATIC_GTFS = "https://content.amtrak.com/content/gtfs/GTFS.zip"
BRIGHTLINE_STATIC_GTFS = "http://feed.gobrightline.com/bl_gtfs.zip"
VIA_RAIL_STATIC_GTFS = (
    "https://www.viarail.ca/sites/all/files/gtfs/viarail.zip"
)
EASTERN_TIME = ZoneInfo("US/Eastern")
S3_BUCKET = "amtrak-performance"
DATA_DIR = pathlib.Path("/tmp")
LOCAL_DATA_TEMPLATE = str(
    DATA_DIR / "daily-*/*/Year={year}/Month={month}/Day={day}/events.csv"
)
S3_DATA_TEMPLATE = "Events-live/{relative_path}.gz"
CSV_FILENAME = "events.csv"
CSV_FIELDS = [
    "service_date",
    "route_id",
    "trip_id",
    "direction_id",
    "stop_id",
    "stop_sequence",
    "vehicle_id",
    "vehicle_label",
    "event_type",
    "event_time",
    "scheduled_headway",
    "scheduled_tt",
]