StarRailCopilot/module/campaign/campaign_base.py

72 lines
2.3 KiB
Python

from module.base.button import Button
from module.logger import logger
from module.map.exception import CampaignEnd
from module.map.exception import ScriptError
from module.map.map import Map
from module.map.map_base import CampaignMap
from module.base.decorator import Config
class CampaignBase(Map):
FUNCTION_NAME_BASE = 'battle_'
ENTRANCE = Button(area=(), color=(), button=(), name='default_button')
MAP: CampaignMap
def battle_default(self):
if self.clear_enemy():
return True
logger.warning('No battle executed.')
return False
@Config.when(MAP_CLEAR_ALL_THIS_TIME=True)
def execute_a_battle(self):
logger.info('Using function: clear_all')
self.clear_all_mystery()
if self.battle_count >= 3:
self.pick_up_ammo()
if self.map.select(is_enemy=True, is_boss=False).count > 0:
return self.battle_default()
else:
if self.config.FLEET_2:
return self.fleet_2.clear_boss()
else:
return self.clear_boss()
@Config.when(MAP_CLEAR_ALL_THIS_TIME=False)
def execute_a_battle(self):
func = self.FUNCTION_NAME_BASE + 'default'
for extra_battle in range(10):
if hasattr(self, self.FUNCTION_NAME_BASE + str(self.battle_count - extra_battle)):
func = self.FUNCTION_NAME_BASE + str(self.battle_count - extra_battle)
break
logger.hr(f'{self.FUNCTION_NAME_BASE}{self.battle_count}', level=2)
logger.info(f'Using function: {func}')
func = self.__getattribute__(func)
result = func()
if not result:
logger.warning('No combat executed.')
raise ScriptError('No combat executed.')
return result
def run(self):
logger.hr(self.ENTRANCE, level=2)
self.handle_spare_fleet()
self.ENTRANCE.area = self.ENTRANCE.button
self.enter_map(self.ENTRANCE, mode=self.config.CAMPAIGN_MODE)
self.handle_map_fleet_lock()
self.handle_fleet_reverse()
self.map_init(self.MAP)
for _ in range(20):
try:
self.execute_a_battle()
except CampaignEnd:
logger.hr('Campaign end')
break