Fix: invest button may change in retries

module/event_hospital/clue.py

@@ -170,12 +170,14 @@ class HospitalClue(HospitalUI):
             if self.appear_then_click(HOSIPITAL_CLUE_CHECK, offset=(20, 20), interval=2):
                 continue
 
-    def invest_enter(self, invest, skip_first_screenshot=True):
+    def invest_enter(self, skip_first_screenshot=True):
         """
         Args:
-            invest:
             skip_first_screenshot:
 
+        Returns:
+            bool: If success to enter
+
         Pages:
             in: is_in_clue
             out: FLEET_PREPARATION
@@ -188,9 +190,13 @@ class HospitalClue(HospitalUI):
             else:
                 self.device.screenshot()
             if self.appear(FLEET_PREPARATION, offset=(20, 50)):
-                break
+                return True
 
             if self.is_in_clue(interval=2):
+                invest = next(self.iter_invest(), None)
+                if invest is None:
+                    logger.info('No more invest')
+                    return False
                 logger.info(f'is_in_clue -> {invest}')
                 self.device.click(invest)
                 continue

module/event_hospital/hospital.py

@@ -20,12 +20,9 @@ class Hospital(HospitalClue, HospitalCombat):
             # May raise ScriptEnd
             self.emotion.check_reduce(battle=1)
 
-            invest = next(self.iter_invest(), None)
+            entered = self.invest_enter()
-            if invest is None:
+            if not entered:
-                logger.info('No more invest')
                 break
-
-            self.invest_enter(invest)
             self.hospital_combat()
 
             # Scheduler