Christoph Urlacher
d3097038a5
All checks were successful
Build Formula10 Docker Image / build-docker (push) Successful in 15s
73 lines
3.4 KiB
Python
73 lines
3.4 KiB
Python
import json
|
|
from typing import List
|
|
from formula10.database.common_queries import find_single_driver_strict
|
|
from formula10.database.model.db_season_guess import DbSeasonGuess
|
|
from formula10.domain.model.driver import Driver
|
|
from formula10.domain.model.team import Team
|
|
from formula10.domain.model.user import User
|
|
|
|
|
|
class SeasonGuess():
|
|
@classmethod
|
|
def from_db_season_guess(cls, db_season_guess: DbSeasonGuess):
|
|
season_guess: SeasonGuess = cls()
|
|
season_guess.user = User.from_db_user(db_season_guess.user)
|
|
season_guess.hot_take = db_season_guess.hot_take if db_season_guess.hot_take is not None else None
|
|
season_guess.p2_wcc = Team.from_db_team(db_season_guess.p2_team) if db_season_guess.p2_team is not None else None
|
|
season_guess.most_overtakes = Driver.from_db_driver(db_season_guess.overtake_driver) if db_season_guess.overtake_driver is not None else None
|
|
season_guess.most_dnfs = Driver.from_db_driver(db_season_guess.dnf_driver) if db_season_guess.dnf_driver is not None else None
|
|
season_guess.most_wdc_gained = Driver.from_db_driver(db_season_guess.gained_driver) if db_season_guess.gained_driver is not None else None
|
|
season_guess.most_wdc_lost = Driver.from_db_driver(db_season_guess.lost_driver) if db_season_guess.lost_driver is not None else None
|
|
|
|
# Deserialize from json
|
|
team_winners: List[str | None] = json.loads(db_season_guess.team_winners_driver_ids_json)
|
|
podiums: List[str] = json.loads(db_season_guess.podium_drivers_driver_ids_json)
|
|
|
|
# Populate relationships
|
|
season_guess.team_winners = [
|
|
Driver.from_db_driver(find_single_driver_strict(int(driver_id))) if driver_id is not None else None
|
|
for driver_id in team_winners
|
|
]
|
|
season_guess.podiums = [
|
|
Driver.from_db_driver(find_single_driver_strict(int(driver_id)))
|
|
for driver_id in podiums
|
|
]
|
|
|
|
return season_guess
|
|
|
|
def to_db_season_guess(self):
|
|
# "Unpopulate" relationships
|
|
team_winners: List[str | None] = [
|
|
driver.name if driver is not None else None
|
|
for driver in self.team_winners
|
|
]
|
|
podiums: List[str] = [
|
|
driver.name for driver in self.podiums
|
|
]
|
|
|
|
# Serialize to json
|
|
db_season_guess: DbSeasonGuess = DbSeasonGuess(user_id=self.user.id)
|
|
db_season_guess.hot_take = self.hot_take
|
|
db_season_guess.p2_team_id = self.p2_wcc.id if self.p2_wcc is not None else None
|
|
db_season_guess.overtake_driver_id = self.most_overtakes.id if self.most_overtakes is not None else None
|
|
db_season_guess.dnf_driver_id = self.most_dnfs.id if self.most_dnfs is not None else None
|
|
db_season_guess.gained_driver_id = self.most_wdc_gained.id if self.most_wdc_gained is not None else None
|
|
db_season_guess.lost_driver_id = self.most_wdc_lost.id if self.most_wdc_lost is not None else None
|
|
db_season_guess.team_winners_driver_ids_json=json.dumps(team_winners)
|
|
db_season_guess.podium_drivers_driver_ids_json=json.dumps(podiums)
|
|
|
|
return db_season_guess
|
|
|
|
user: User
|
|
hot_take: str | None
|
|
p2_wcc: Team | None
|
|
most_overtakes: Driver | None
|
|
most_dnfs: Driver | None
|
|
most_wdc_gained: Driver | None
|
|
most_wdc_lost: Driver | None
|
|
team_winners: List[Driver | None]
|
|
podiums: List[Driver]
|
|
|
|
def hot_take_string(self) -> str:
|
|
return self.hot_take if self.hot_take is not None else ""
|