DB: Pomocné metody na db.Round pro práci se stavy kol s více částmi + úprava check_deadline

mo/db.py

@@ -15,7 +15,7 @@ from sqlalchemy.orm.attributes import get_history
 from sqlalchemy.dialects.postgresql import JSONB
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.sql.expression import CTE
-from typing import Optional, List, Tuple
+from typing import Optional, List, Tuple, Union
 
 import mo
 from mo.util_format import timedelta, time_and_timedelta
@@ -205,6 +205,30 @@ class Round(Base):
     def round_code(self):
         return f"{self.year}-{self.category}-{self.seq}"
 
+    def single(self) -> bool:
+        return len(self.parts) == 1
+
+    def multipart(self) -> bool:
+        return len(self.parts) > 1
+
+    def state_all(self, states: Union[RoundState, List[RoundState]]) -> bool:
+        """Test jestli jsou všechny části kola v daném stavu. Lze zadat i množinu stavů."""
+        if not isinstance(states, list):
+            states = [states]
+        for part in self.parts:
+            if part.state not in states:
+                return False
+        return True
+
+    def state_any(self, states: Union[RoundState, List[RoundState]]) -> bool:
+        """Test jestli existuje alespoň jedna část kola v daném stavu. Lze zadat i množinu stavů."""
+        if not isinstance(states, list):
+            states = [states]
+        for part in self.parts:
+            if part.state in states:
+                return True
+        return False
+
 
 class RoundPart(Base):
     __tablename__ = 'round_parts'
@@ -460,13 +484,13 @@ class Paper(Base):
     for_user_obj = relationship('User', primaryjoin='Paper.for_user == User.user_id')
     uploaded_by_obj = relationship('User', primaryjoin='Paper.uploaded_by == User.user_id')
 
-    def check_deadline(self, round: Round) -> Optional[str]:
-        # K round se dá dostat přes self.task.round, ale nejspíš to není nakešované.
+    def check_deadline(self, round_part: RoundPart) -> Optional[str]:
+        # K round_part se dá dostat přes self.task.round, ale nejspíš to není nakešované.
         if self.uploaded_by == self.for_user:
-            deadline = round.ct_submit_end
+            deadline = round_part.ct_submit_end
             suffix = ' účastnickém termínu'
         else:
-            deadline = round.pr_submit_end
+            deadline = round_part.pr_submit_end
             suffix = ' dozorovém termínu'
         if deadline is not None and self.uploaded_at > deadline:
             return timedelta(self.uploaded_at, deadline, descriptive=True) + suffix