46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
|
import datetime
|
||
|
|
|
||
|
|
from ergast_py.constants.status_type import StatusType
|
||
|
|
|
||
|
|
class Helpers:
|
||
|
|
def construct_datetime_str(self, date: str, time: str) -> datetime.datetime:
|
||
|
|
new_datetime = datetime.datetime.strptime(f"{date} {time}", "%Y-%m-%d %H:%M:%SZ")
|
||
|
|
new_datetime = new_datetime.replace(tzinfo=datetime.timezone.utc)
|
||
|
|
return new_datetime
|
||
|
|
|
||
|
|
def construct_datetime_dict(self, dict: dict) -> datetime.datetime:
|
||
|
|
if "date" not in dict or "time" not in dict:
|
||
|
|
return None
|
||
|
|
return self.construct_datetime_str(dict["date"], dict["time"])
|
||
|
|
|
||
|
|
def construct_date(self, date: str) -> datetime.date:
|
||
|
|
elements = date.split("-")
|
||
|
|
return datetime.date(year=int(elements[0]), month=int(elements[1]), day=int(elements[2]))
|
||
|
|
|
||
|
|
def construct_lap_time_millis(self, millis: dict) -> datetime.time:
|
||
|
|
if "millis" in millis:
|
||
|
|
value = int(millis["millis"])
|
||
|
|
return datetime.datetime.fromtimestamp(value/1000.0).time()
|
||
|
|
return None
|
||
|
|
|
||
|
|
def format_lap_time(self, time: str) -> datetime.time:
|
||
|
|
if time != "":
|
||
|
|
return datetime.datetime.strptime(time, "%M:%S.%f").time()
|
||
|
|
return None
|
||
|
|
|
||
|
|
def construct_lap_time(self, time: dict) -> datetime.time:
|
||
|
|
if "time" in time:
|
||
|
|
value = time["time"]
|
||
|
|
return self.format_lap_time(value)
|
||
|
|
return None
|
||
|
|
|
||
|
|
def construct_local_time(self, time: str) -> datetime.time:
|
||
|
|
if "time" != "":
|
||
|
|
return datetime.datetime.strptime(f"{time}", "%H:%M:%S").time()
|
||
|
|
return None
|
||
|
|
|
||
|
|
def construct_pitstop_duration(self, time: str) -> datetime.time:
|
||
|
|
if "time" != "":
|
||
|
|
return datetime.datetime.strptime(f"{time}", "%S.%f").time()
|
||
|
|
return None
|