DB: Delegace v db.py

mo/db.py

@@ -157,6 +157,7 @@ class RoundState(MOEnum):
     running = auto()
     grading = auto()
     closed = auto()
+    delegate = auto()
 
     def friendly_name(self) -> str:
         return round_state_names[self]
@@ -167,6 +168,7 @@ round_state_names = {
     RoundState.running: 'běží',
     RoundState.grading: 'opravuje se',
     RoundState.closed: 'ukončeno',
+    RoundState.delegate: 'po oblastech',
 }
 
 
@@ -305,6 +307,7 @@ class Contest(Base):
     master_contest_id = Column(Integer, ForeignKey('contests.contest_id'))
     round_id = Column(Integer, ForeignKey('rounds.round_id'), nullable=False)
     place_id = Column(Integer, ForeignKey('places.place_id'), nullable=False)
+    state = Column(Enum(RoundState, name='round_state'), nullable=False, server_default=text("'preparing'::round_state"))
 
     master = relationship('Contest', primaryjoin='Contest.master_contest_id == Contest.contest_id', remote_side='Contest.contest_id', post_update=True)
     place = relationship('Place')
@@ -320,6 +323,13 @@ class Contest(Base):
             add_self or Contest.contest_id != self.contest_id,
         ).options(joinedload(Contest.round)).all()
 
+    def get_state(self):
+        if self.round.state != RoundState.delegate:
+            return self.round.state
+        else:
+            return self.state
+
+
 class LogType(MOEnum):
     general = auto()
     user = auto()