chore: add Junior status

one-zero-eight · Feb 24, 2024 · 7070aee · 7070aee
1 parent 9bedfe1
commit 7070aee
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/schemas/participant.py b/src/schemas/participant.py
@@ -5,16 +5,17 @@
 
 
 class ParticipantStatus(StrEnum):
-    # 12 * 7 = 84 hours per week
-
     FREE = "free"
+    JUNIOR = "junior"
     MIDDLE = "middle"
     SENIOR = "senior"
     LORD = "lord"
 
     def max_hours_to_book_per_day(self) -> Optional[int]:
         if self == ParticipantStatus.FREE:
             return 2
+        elif self == ParticipantStatus.JUNIOR:
+            return 2
         elif self == ParticipantStatus.MIDDLE:
             return 3
         elif self == ParticipantStatus.SENIOR:
@@ -26,6 +27,8 @@ def max_hours_to_book_per_day(self) -> Optional[int]:
     def max_hours_to_book_per_week(self) -> Optional[int]:
         if self == ParticipantStatus.FREE:
             return 4
+        elif self == ParticipantStatus.JUNIOR:
+            return 5
         elif self == ParticipantStatus.MIDDLE:
             return 8
         elif self == ParticipantStatus.SENIOR: