Add seasonguessresult table

formula10/database/model/db_season_guess_result.py

@@ -0,0 +1,44 @@
+from typing import List
+
+from sqlalchemy import Boolean, ForeignKey
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from formula10 import db
+from formula10.database.model.db_user import DbUser
+
+class DbSeasonGuessResult(db.Model):
+    """
+    Manually entered results for the season bonus guesses.
+    """
+
+    __tablename__ = "seasonguessresult"
+    __csv_header__ = ["user_name", "hot_take_correct", "overtakes_correct"]
+
+    def __init__(self, *, user_name: str, hot_take_correct: bool, overtakes_correct: bool):
+        self.user_name = user_name  # Primary key
+
+        self.hot_take_correct = hot_take_correct
+        self.overtakes_correct = overtakes_correct
+
+    @classmethod
+    def from_csv(cls, row: List[str]):
+        db_season_guess_result: DbSeasonGuessResult = cls(user_name=str(row[0]),
+                                                          hot_take_correct=True if str(row[1])=="True" else False,
+                                                          overtakes_correct=True if str(row[2])=="True" else False)
+
+        return db_season_guess_result
+
+    # This object can't be edited from the page context
+    # def to_csv(self) -> List[Any]:
+    #     return [
+    #         self.user_name,
+    #         self.hot_take_correct,
+    #         self.overtakes_correct
+    #     ]
+
+    user_name: Mapped[str] = mapped_column(ForeignKey("user.name"), primary_key=True)
+    hot_take_correct: Mapped[bool] = mapped_column(Boolean)
+    overtakes_correct: Mapped[bool] = mapped_column(Boolean)
+
+    # Relationships
+    user: Mapped[DbUser] = relationship("DbUser", foreign_keys=[user_name])

formula10/database/model/db_user.py

@@ -19,7 +19,8 @@ class DbUser(db.Model):
 
     @classmethod
     def from_csv(cls, row: List[str]):
-        db_user: DbUser = cls(name=str(row[0]), enabled=True if str(row[1])=="True" else False)
+        db_user: DbUser = cls(name=str(row[0]),
+                              enabled=True if str(row[1])=="True" else False)
         return db_user
 
     def to_csv(self) -> List[Any]: