fix(experience): pick valid SEP rating (still wonky...)

zyv · Apr 17, 2024 · 5f12a66 · 5f12a66
1 parent 118ce49
commit 5f12a66
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/logbook/views/experience.py b/logbook/views/experience.py
@@ -16,6 +16,7 @@
 )
 from .utils import (
     AuthenticatedTemplateView,
+    get_current_sep_rating,
 )
 
 
@@ -38,7 +39,7 @@ def get_sep_revalidation_experience(log_entries: QuerySet[LogEntry]) -> Experien
         aircraft__type__in={AircraftType.SEP, AircraftType.TMG},
         departure_time__gte=make_aware(
             datetime.combine(
-                Certificate.objects.get(name__contains="SEP").valid_until - relativedelta(months=12),
+                get_current_sep_rating().valid_until - relativedelta(months=12),
                 time.min,
             ),
             UTC,

diff --git a/logbook/views/utils.py b/logbook/views/utils.py
@@ -13,6 +13,11 @@
 EXPIRY_WARNING_THRESHOLD = relativedelta(months=3)
 
 
+def get_current_sep_rating() -> Certificate:
+    (current_sep_rating,) = filter(lambda item: item.valid, Certificate.objects.filter(name__contains="SEP"))
+    return current_sep_rating
+
+
 def check_certificates_expiry(request):
     today = datetime.now(tz=UTC).date()
     for certificate in Certificate.objects.filter(