63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
from dataclasses import dataclass
|
|
import datetime
|
|
|
|
from ergast_py.helpers import Helpers
|
|
from ergast_py.models.circuit import Circuit
|
|
from ergast_py.models.lap import Lap
|
|
from ergast_py.models.result import Result
|
|
from ergast_py.models.pit_stop import PitStop
|
|
|
|
@dataclass
|
|
class Race:
|
|
"""
|
|
Representation of a single Race from a Formula One season
|
|
Races may contain:
|
|
season: Integer
|
|
round: Integer
|
|
url: String
|
|
raceName: String
|
|
circuit: Circuit
|
|
date: datetime.datetime
|
|
results: Result[]
|
|
firstPractice: datetime.datetime
|
|
secondPractice: datetime.datetime
|
|
thirdPractice: datetime.datetime
|
|
sprint: datetime.datetime
|
|
sprintResults: Result[]
|
|
qualifying: datetime.datetime
|
|
qualifyingResults: Result[]
|
|
pitStops: PitStop[]
|
|
laps: Lap[]
|
|
"""
|
|
|
|
def __init__(self, season: int, round: int, url: str, raceName: str, circuit: Circuit, date: datetime.datetime,
|
|
results: list[Result], firstPractice: datetime.datetime, secondPractice: datetime.datetime,
|
|
thirdPractice: datetime.datetime, sprint: datetime.datetime, sprintResults: list[Result],
|
|
qualifying: datetime.datetime, qualifyingResults: list[Result], pitStops: list[PitStop],
|
|
laps: list[Lap]) -> None:
|
|
self.season = season
|
|
self.round = round
|
|
self.url = url
|
|
self.raceName = raceName
|
|
self.circuit = circuit
|
|
self.date = date
|
|
self.results = results
|
|
self.firstPractice = firstPractice
|
|
self.secondPractice = secondPractice
|
|
self.thirdPractice = thirdPractice
|
|
self.sprint = sprint
|
|
self.sprintResults = sprintResults
|
|
self.qualifying = qualifying
|
|
self.qualifyingResults = qualifyingResults
|
|
self.pitStops = pitStops
|
|
self.laps = laps
|
|
pass
|
|
|
|
def __str__(self) -> str:
|
|
members = ', '.join(f"{key}={value}" for key, value in self.__dict__.items())
|
|
return f"{type(self).__name__}({members})"
|
|
|
|
def __repr__(self) -> str:
|
|
members = ', '.join(f"{key}={value}" for key, value in self.__dict__.items())
|
|
return f"{type(self).__name__}({members})"
|