StarRailCopilot/module/hard/hard.py
2020-03-29 01:41:34 +08:00

51 lines
1.8 KiB
Python

import importlib
from module.base.ocr import Digit
from module.campaign.campaign_hard import Campaign
from module.campaign.run import CampaignRun
from module.hard.assets import *
from module.logger import logger
OCR_HARD_REMAIN = Digit(OCR_HARD_REMAIN, letter=(123, 227, 66), back=(24, 24, 24), limit=3, name='OCR_HARD_REMAIN')
RECORD_OPTION = ('DailyRecord', 'hard')
RECORD_SINCE = (0,)
class CampaignHard(CampaignRun):
equipment_has_take_on = False
campaign: Campaign
def run(self):
logger.hr('Campaign hard', level=1)
chapter, stage = self.config.HARD_CAMPAIGN.split('-')
name = f'campaign_{chapter}_{stage}'
# Initial
self.load_campaign(name='campaign_hard', folder='campaign_hard') # Load campaign file
module = importlib.import_module('.' + name, 'campaign.campaign_main') # Load map from normal mode.
self.campaign.MAP = module.MAP
self.campaign_name_set(name + '_HARD')
if self.equipment_has_take_on:
self.campaign.equipment_has_take_on = True
# UI ensure
self.ui_weigh_anchor()
self.campaign_ensure_mode(mode='normal')
self.campaign_ensure_chapter(index=int(chapter))
self.campaign_ensure_mode(mode='hard')
self.campaign.ENTRANCE = self.campaign_get_entrance(name=f'{chapter}-{stage}')
# Run
remain = OCR_HARD_REMAIN.ocr(self.device.image)
logger.attr('Remain', remain)
for n in range(remain):
self.campaign.run()
self.campaign.equipment_take_off_when_finished()
def record_executed_since(self):
return self.config.record_executed_since(option=RECORD_OPTION, since=RECORD_SINCE)
def record_save(self):
return self.config.record_save(option=RECORD_OPTION)