Small database migration fixes

formula10/__init__.py

@@ -35,10 +35,9 @@ import formula10.controller.error_controller
 # Large DB Update
 # - Don't use names for frontend post requests, either use IDs or post the whole object (if its possible)...
 # - For season guess calc there is missing: Fastest laps + sprint points + sprint DNFs (in race result)
-# - Fix Valtteri/Russel spelling errors
 # - Mask to allow changing usernames (easy if name is not used as ID)
 # - Maybe even masks for races + drivers + teams?
-# - DB fields for links to F1 site
+# - DB fields for links to F1 site - NO: just hardcode them in with a dictionary
 # - DB fields for qualifying dates
 
 # Leaderboards/Points

formula10/domain/model/season_guess.py

@@ -58,6 +58,15 @@ class SeasonGuess():
 
         return db_season_guess
 
+    def __eq__(self, __value: object) -> bool:
+        if isinstance(__value, SeasonGuess):
+            return self.user == __value.user
+
+        return NotImplemented
+
+    def __hash__(self) -> int:
+        return hash(self.user)
+
     user: User
     hot_take: str | None
     p2_wcc: Team | None

formula10/domain/model/season_guess_result.py

@@ -18,6 +18,15 @@ class SeasonGuessResult():
         db_season_guess_result.overtakes_correct = self.overtakes_correct
         return db_season_guess_result
 
+    def __eq__(self, __value: object) -> bool:
+        if isinstance(__value, SeasonGuessResult):
+            return self.user == __value.user
+
+        return NotImplemented
+
+    def __hash__(self) -> int:
+        return hash(self.user)
+
     user: User
     hot_take_correct: bool
     overtakes_correct: bool