Calculate penalty expiration based on event date instead of current date

webkom · Mar 10, 2024 · ce47a83 · ce47a83
1 parent c48fb05
commit ce47a83
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lego/apps/users/models.py b/lego/apps/users/models.py
@@ -541,16 +541,16 @@ class Penalty(BasisModel):
     objects = UserPenaltyManager()  # type: ignore
 
     def expires(self):
-        dt = Penalty.penalty_offset(self.created_at) - (
-            timezone.now() - self.created_at
+        dt = Penalty.penalty_offset(self.source_event.end_time) - (
+            timezone.now() - self.source_event.end_time
         )
         return dt.days
 
     @property
     def exact_expiration(self):
         """Returns the exact time of expiration"""
-        dt = Penalty.penalty_offset(self.created_at) - (
-            timezone.now() - self.created_at
+        dt = Penalty.penalty_offset(self.source_event.end_time) - (
+            timezone.now() - self.source_event.end_time
         )
         return timezone.now() + dt