From e57109caf4e645400240d246eafff3af38d20f8e Mon Sep 17 00:00:00 2001
From: Christoph Urlacher <christoph.urlacher@protonmail.com>
Date: Sat, 9 Mar 2024 20:56:58 +0100
Subject: [PATCH] Add quali_date + has_sprint fields to race

---
 formula10/database/model/db_race.py | 6 ++++--
 formula10/domain/model/race.py      | 6 ++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/formula10/database/model/db_race.py b/formula10/database/model/db_race.py
index cd0160c..b99bcb9 100644
--- a/formula10/database/model/db_race.py
+++ b/formula10/database/model/db_race.py
@@ -1,5 +1,5 @@
 from datetime import datetime
-from sqlalchemy import DateTime, Integer, String
+from sqlalchemy import Boolean, DateTime, Integer, String
 from sqlalchemy.orm import Mapped, mapped_column
 
 from formula10 import db
@@ -19,4 +19,6 @@ class DbRace(db.Model):
     name: Mapped[str] = mapped_column(String(64), nullable=False, unique=True)
     number: Mapped[int] = mapped_column(Integer, nullable=False, unique=True)
     date: Mapped[datetime] = mapped_column(DateTime, nullable=False, unique=True)
-    pxx: Mapped[int] = mapped_column(Integer, nullable=False)  # This is the place to guess
\ No newline at end of file
+    pxx: Mapped[int] = mapped_column(Integer, nullable=False)  # This is the place to guess
+    quali_date: Mapped[datetime] = mapped_column(DateTime, nullable=False, unique=True)
+    has_sprint: Mapped[bool] = mapped_column(Boolean, nullable=False)
\ No newline at end of file
diff --git a/formula10/domain/model/race.py b/formula10/domain/model/race.py
index a74e20e..6518cad 100644
--- a/formula10/domain/model/race.py
+++ b/formula10/domain/model/race.py
@@ -13,6 +13,8 @@ class Race():
         race.number = db_race.number
         race.date = db_race.date
         race.place_to_guess = db_race.pxx
+        race.quali_date = db_race.quali_date
+        race.has_sprint = db_race.has_sprint
         return race
 
     def to_db_race(self) -> DbRace:
@@ -21,6 +23,8 @@ class Race():
         db_race.number = self.number
         db_race.date = self.date
         db_race.pxx = self.place_to_guess
+        db_race.quali_date = self.date
+        db_race.has_sprint = self.has_sprint
         return db_race
 
     def __eq__(self, __value: object) -> bool:
@@ -37,6 +41,8 @@ class Race():
     number: int
     date: datetime
     place_to_guess: int
+    quali_date: datetime
+    has_sprint: bool
 
     @property
     def name_sanitized(self) -> str: