Tmp: optimize code

Author: guoh064

module/campaign/gems_farming.py

@@ -1,15 +1,15 @@
-from cached_property import cached_property
 from datetime import datetime
 
 import numpy as np
 
+from module.base.decorator import cached_property
 from module.campaign.campaign_base import CampaignBase
 from module.campaign.run import CampaignRun
 from module.combat.assets import BATTLE_PREPARATION
 from module.combat.emotion import Emotion
 from module.equipment.assets import *
 from module.equipment.fleet_equipment import FleetEquipment
-from module.exception import CampaignEnd, ScriptError
+from module.exception import CampaignEnd, ScriptEnd, ScriptError
 from module.handler.assets import AUTO_SEARCH_MAP_OPTION_OFF
 from module.logger import logger
 from module.map.assets import FLEET_PREPARATION, MAP_PREPARATION
@@ -42,29 +42,12 @@ def check_reduce(self, battle):
             CampaignEnd: Pause current task to prevent emotion control in the future.
         """
 
-        if not self.is_calculate:
-            return
-
-        method = self.config.Fleet_FleetOrder
-
-        if method == 'fleet1_all_fleet2_standby':
-            battle = (battle, 0)
-        elif method == 'fleet1_standby_fleet2_all':
-            battle = (0, battle)
-        else:
-            raise ScriptError(f'Unknown fleet order: {method}')
-
-        battle = tuple(np.array(battle) * self.reduce_per_battle_before_entering)
-        logger.info(f'Expect emotion reduce: {battle}')
-
-        self.update()
-        self.record()
-        self.show()
-        recovered = max([f.get_recovered(b) for f, b in zip(self.fleets, battle)])
-        if recovered > datetime.now():
-            logger.info('Delay current task to prevent emotion control in the future')
-            self.config.GEMS_EMOTION_TRIGGRED = True
-            raise CampaignEnd('Emotion withdraw')
+        try:
+            super().check_reduce(battle)
+        except ScriptEnd:
+            self.config.GEMS_EMOTION_TRIGGERED = True
+            self.config.task_delay(minute=-30) # to undo emotion delay
+            raise CampaignEnd('Emotion control')
 
     def wait(self, fleet_index):
         pass
@@ -362,8 +345,6 @@ def get_templates(common_dd):
             logger.error(f'Invalid CommonDD setting: {common_dd}')
             raise ScriptError(f'Invalid CommonDD setting: {common_dd}')
 
-    _new_emotion: int
-
     def flagship_change_execute(self):
         """
         Returns:
@@ -382,7 +363,7 @@ def flagship_change_execute(self):
         ship = self.get_common_rarity_cv()
         if ship:
             target_ship = min(ship, key=lambda s: (s.level, -s.emotion))
-            self._new_emotion = target_ship.emotion
+            self.set_emotion(target_ship.emotion)
             self._ship_change_confirm(target_ship.button)
 
             logger.info('Change flagship success')
@@ -423,13 +404,14 @@ def vanguard_change_execute(self):
         ship = self.get_common_rarity_dd()
         if ship:
             target_ship = max(ship, key=lambda s: s.emotion)
-            self._new_emotion = min(self._new_emotion, target_ship.emotion)
+            self.set_emotion(min(self.get_emotion(), target_ship.emotion))
             self._ship_change_confirm(target_ship.button)
 
             logger.info('Change vanguard ship success')
             return True
         else:
             logger.info('Change vanguard ship failed, no DD in common rarity.')
+            self.set_emotion(0)  # a failure in vanguard change means low emotion DD, assuming 0.
             self._dock_reset()
             self.ui_back(check_button=page_fleet.check_button)
             return False
@@ -451,6 +433,12 @@ def triggered_stop_condition(self, oil_check=True):
 
         return super().triggered_stop_condition(oil_check=oil_check)
 
+    def get_emotion(self):
+        if self.config.Fleet_FleetOrder == 'fleet1_standby_fleet2_all':
+            return self.campaign.config.Emotion_Fleet2Value
+        else:
+            return self.campaign.config.Emotion_Fleet1Value
+
     def set_emotion(self, emotion):
         if self.config.Fleet_FleetOrder == 'fleet1_standby_fleet2_all':
             self.campaign.config.set_record(Emotion_Fleet2Value=emotion)
@@ -474,23 +462,19 @@ def run(self, name, folder='campaign_main', mode='normal', total=0):
             try:
                 super().run(name=name, folder=folder, total=total)
             except CampaignEnd as e:
-                if e.args[0] == 'Emotion withdraw':
+                if e.args[0] in ['Emotion withdraw', 'Emotion control']:
                     self._trigger_emotion = True
                 else:
                     raise e
 
             # End
             if self._trigger_lv32 or self._trigger_emotion:
                 success = True
-                self._new_emotion = 200  # magic value to exceed 150
                 if self.change_flagship:
                     success = self.flagship_change()
                 if self.change_vanguard:
                     success = success and self.vanguard_change()
 
-                if self._new_emotion < 200:
-                    self.set_emotion(self._new_emotion)
-
                 if is_limit and self.config.StopCondition_RunCount <= 0:
                     logger.hr('Triggered stop condition: Run count')
                     self.config.StopCondition_RunCount = 0