73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
""" Result class """
|
|
|
|
import datetime
|
|
from dataclasses import dataclass
|
|
|
|
from ergast_py.models.constructor import Constructor
|
|
from ergast_py.models.driver import Driver
|
|
from ergast_py.models.fastest_lap import FastestLap
|
|
|
|
|
|
@dataclass
|
|
class Result():
|
|
"""
|
|
Representation of a single Result from a Formula One race
|
|
|
|
Results may contain:
|
|
number: Integer
|
|
position: Integer
|
|
position_text: String
|
|
points: Integer
|
|
driver: Driver
|
|
constructor: Constructor
|
|
grid: Integer
|
|
laps: Integer
|
|
status: Integer
|
|
time: datetime.time
|
|
fastest_lap: FastestLap
|
|
qual_1: datetime.time
|
|
qual_2: datetime.time
|
|
qual_3: datetime.time
|
|
"""
|
|
|
|
def __init__(self, number: int, position: int, position_text: str, points: float,
|
|
driver: Driver, constructor: Constructor, grid: int, laps: int, status: int,
|
|
time: datetime.time, fastest_lap: FastestLap, qual_1: datetime.time,
|
|
qual_2: datetime.time, qual_3: datetime.time) -> None:
|
|
self.number = number
|
|
self.position = position
|
|
self.position_text = position_text
|
|
self.points = points
|
|
self.driver = driver
|
|
self.constructor = constructor
|
|
self.grid = grid
|
|
self.laps = laps
|
|
self.status = status
|
|
self.time = time
|
|
self.fastest_lap = fastest_lap
|
|
self.qual_1 = qual_1
|
|
self.qual_2 = qual_2
|
|
self.qual_3 = qual_3
|
|
|
|
def __repr__(self) -> str:
|
|
members = ', '.join(f"{key}={value}" for key, value in self.__dict__.items())
|
|
return f"{type(self).__name__}({members})"
|
|
|
|
def __eq__(self, __o: object) -> bool:
|
|
return isinstance(__o, Result) and (
|
|
self.number == __o.number and
|
|
self.position == __o.position and
|
|
self.position_text == __o.position_text and
|
|
self.points == __o.points and
|
|
self.driver == __o.driver and
|
|
self.constructor == __o.constructor and
|
|
self.grid == __o.grid and
|
|
self.laps == __o.laps and
|
|
self.status == __o.status and
|
|
self.time == __o.time and
|
|
self.fastest_lap == __o.fastest_lap and
|
|
self.qual_1 == __o.qual_1 and
|
|
self.qual_2 == __o.qual_2 and
|
|
self.qual_3 == __o.qual_3
|
|
)
|