define weight in terms of parent

DemocracyClub · Oct 30, 2024 · 2594c7a · 2594c7a
1 parent 787304a
commit 2594c7a
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/postcode_lookup/template_sorter.py b/postcode_lookup/template_sorter.py
@@ -177,12 +177,8 @@ def __init__(self, *args, **kwargs) -> None:
 
     @property
     def weight(self):
-        if self.timetable.is_before(TimetableEvent.REGISTRATION_DEADLINE):
-            return -5999
-
-        if self.timetable.is_after(TimetableEvent.REGISTRATION_DEADLINE):
-            return 1002
-        return 0
+        parent_weight = super().weight
+        return 0 if parent_weight == 0 else parent_weight + 1
 
     @property
     def context(self):