fix(base): 重置 ElectronFlowGoal 中的 failed 状态

SVUCTF · Oct 31, 2024 · 41bd6e4 · 41bd6e4
1 parent 4337fb1
commit 41bd6e4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/base/wireworld/app/models/goal.py b/base/wireworld/app/models/goal.py
@@ -23,6 +23,9 @@ def update(self, world: Wireworld) -> bool:
     def is_achieved(self) -> bool:
         return self.current_matches >= self.required_matches
 
+    def reset(self):
+        self.current_matches = 0
+
 
 class SpecificPeriodGoal(Goal):
     type: Literal["SpecificPeriodGoal"]
@@ -65,5 +68,9 @@ def update(self, world: Wireworld) -> bool:
     def is_achieved(self) -> bool:
         return not self.failed and self.current_matches >= self.required_matches
 
+    def reset(self):
+        self.current_matches = 0
+        self.failed = False
+
 
 GoalUnion = Union[SpecificPeriodGoal, ElectronFlowGoal]
diff --git a/base/wireworld/app/services/challenge.py b/base/wireworld/app/services/challenge.py
@@ -134,7 +134,7 @@ def reset(cls):
         )
         cls._setup_initial_state()
         for goal in cls.config.goals:
-            goal.current_matches = 0
+            goal.reset()
         cls.completed = False
 
     @classmethod