Add OpenF1 model classes
This commit is contained in:
30
formula10/openf1/model/api_position.py
Normal file
30
formula10/openf1/model/api_position.py
Normal file
@ -0,0 +1,30 @@
|
||||
from datetime import datetime
|
||||
from typing import Any, Callable, Dict
|
||||
|
||||
|
||||
class ApiPosition():
|
||||
__type_conversion_map__: Dict[str, Callable[[Any], Any]] = {
|
||||
"session_key": int,
|
||||
"meeting_key": int,
|
||||
"driver_number": int,
|
||||
"date": lambda date: datetime.strptime(date, "%Y-%m-%dT%H:%M:%S.%f"),
|
||||
"position": int
|
||||
}
|
||||
|
||||
def __init__(self, response: dict[str, str]):
|
||||
for key in response:
|
||||
if not hasattr(self, key):
|
||||
raise Exception(f"Mismatch between response data and {type(self).__name__} (key={key})")
|
||||
|
||||
if not key in self.__type_conversion_map__:
|
||||
raise Exception(f"Mismatch between response data and {type(self).__name__}.__type_map__ (key={key})")
|
||||
|
||||
setattr(self, key, self.__type_conversion_map__[key](response[key]))
|
||||
|
||||
print("ApiPosition:", self.__dict__)
|
||||
|
||||
session_key: int = None # type: ignore
|
||||
meeting_key: int = None # type: ignore
|
||||
driver_number: int = None # type: ignore
|
||||
date: datetime = None # type: ignore
|
||||
position: int = None # type: ignore
|
Reference in New Issue
Block a user