ergast-py/ergast_py/models/race.py

""" Race class """

import datetime
from dataclasses import dataclass

from ergast_py.models.circuit import Circuit
from ergast_py.models.lap import Lap
from ergast_py.models.pit_stop import PitStop
from ergast_py.models.result import Result


@dataclass
class Race():
    """
    Representation of a single Race from a Formula One season

    Races may contain:
        season: Integer
        round_no: Integer
        url: String
        race_name: String
        circuit: Circuit
        date: datetime.datetime
        results: Result[]
        first_practice: datetime.datetime
        second_practice: datetime.datetime
        third_practice: datetime.datetime
        sprint: datetime.datetime
        sprint_results: Result[]
        qualifying: datetime.datetime
        qualifying_results: Result[]
        pit_stops: PitStop[]
        laps: Lap[]
    """

    def __init__(self, season: int, round_no: int, url: str, race_name: str, circuit: Circuit,
                 date: datetime.datetime, results: list[Result], first_practice: datetime.datetime,
                 second_practice: datetime.datetime, third_practice: datetime.datetime,
                 sprint: datetime.datetime, sprint_results: list[Result],
                 qualifying: datetime.datetime, qualifying_results: list[Result],
                 pit_stops: list[PitStop], laps: list[Lap]) -> None:
        self.season = season
        self.round_no = round_no
        self.url = url
        self.race_name = race_name
        self.circuit = circuit
        self.date = date
        self.results = results
        self.first_practice = first_practice
        self.second_practice = second_practice
        self.third_practice = third_practice
        self.sprint = sprint
        self.sprint_results = sprint_results
        self.qualifying = qualifying
        self.qualifying_results = qualifying_results
        self.pit_stops = pit_stops
        self.laps = laps

    def __repr__(self) -> str:
        members = ', '.join(f"{key}={value}" for key, value in self.__dict__.items())
        return f"{type(self).__name__}({members})"