mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-22 00:35:34 +00:00
128 lines
5.1 KiB
Python
128 lines
5.1 KiB
Python
from module.config.utils import get_server_next_monday_update
|
||
from module.logger import logger
|
||
from module.ocr.ocr import DigitCounter
|
||
from tasks.daily.keywords import KEYWORDS_DAILY_QUEST
|
||
from tasks.dungeon.assets.assets_dungeon_ui import OCR_DUNGEON_LIST, OCR_WEEKLY_LIMIT
|
||
from tasks.dungeon.dungeon import Dungeon
|
||
from tasks.dungeon.keywords import DungeonList, KEYWORDS_DUNGEON_NAV, KEYWORDS_DUNGEON_TAB
|
||
from tasks.dungeon.ui import DUNGEON_LIST
|
||
|
||
|
||
class OcrWeeklyLimit(DigitCounter):
|
||
def format_result(self, result) -> tuple[int, int, int]:
|
||
current, remain, total = super().format_result(result)
|
||
# [OCR_WEEKLY_LIMIT format] ３3/3 -> (33, -30, 3)
|
||
if current == 11:
|
||
current = 1
|
||
remain = total - current
|
||
if current == 22:
|
||
current = 2
|
||
remain = total - current
|
||
if current == 33:
|
||
current = 3
|
||
remain = total - current
|
||
|
||
return current, remain, total
|
||
|
||
|
||
class WeeklyDungeon(Dungeon):
|
||
def require_compulsory_support(self) -> bool:
|
||
return False
|
||
|
||
def _dungeon_run(self, dungeon: DungeonList, team: int = None, wave_limit: int = 0, support_character: str = None,
|
||
skip_ui_switch: bool = False):
|
||
if team is None:
|
||
team = self.config.Weekly_Team
|
||
|
||
skip_ui_switch = True
|
||
return super()._dungeon_run(
|
||
dungeon=dungeon, team=team, wave_limit=wave_limit,
|
||
support_character=support_character, skip_ui_switch=skip_ui_switch)
|
||
|
||
def handle_ascension_dungeon_prepare(self):
|
||
# combat_wave_cost==30 in weekly, but no handle_ascension_dungeon_prepare required
|
||
return False
|
||
|
||
def get_weekly_remain(self) -> int:
|
||
"""
|
||
Pages:
|
||
in: page_guide, Survival_Index, KEYWORDS_DUNGEON_NAV.Echo_of_War
|
||
"""
|
||
ocr = OcrWeeklyLimit(OCR_WEEKLY_LIMIT)
|
||
current, _, _ = ocr.ocr_single_line(self.device.image)
|
||
total = self.config.stored.EchoOfWar.FIXED_TOTAL
|
||
if current <= total:
|
||
logger.attr('EchoOfWar', f'{current}/{total}')
|
||
self.config.stored.EchoOfWar.value = current
|
||
return current
|
||
else:
|
||
logger.warning(f'Invalid EchoOfWar limit: {current}/{total}')
|
||
return 0
|
||
|
||
def run(self):
|
||
# self.config.update_battle_pass_quests()
|
||
self.config.update_daily_quests()
|
||
self.called_daily_support = False
|
||
self.achieved_daily_quest = False
|
||
self.achieved_weekly_quest = False
|
||
self.daily_quests = self.config.stored.DailyQuest.load_quests()
|
||
self.weekly_quests = self.config.stored.BattlePassWeeklyQuest.load_quests()
|
||
|
||
dungeon = DungeonList.find(self.config.Weekly_Name)
|
||
planner = self.planner.get_weekly()
|
||
if planner is not None:
|
||
dungeon = planner
|
||
self.is_doing_planner = True
|
||
logger.attr('DungeonWeekly', dungeon)
|
||
|
||
# UI switches
|
||
self.dungeon_tab_goto(KEYWORDS_DUNGEON_TAB.Survival_Index)
|
||
# Equivalent to self.dungeon_goto(dungeon), but check limit remains
|
||
DUNGEON_LIST.search_button = OCR_DUNGEON_LIST
|
||
self._dungeon_nav_goto(KEYWORDS_DUNGEON_NAV.Echo_of_War)
|
||
self._dungeon_wait_until_dungeon_list_loaded()
|
||
monday = get_server_next_monday_update(self.config.Scheduler_ServerUpdate)
|
||
|
||
# Check limit
|
||
remain = self.get_weekly_remain()
|
||
if remain <= 0:
|
||
if KEYWORDS_DAILY_QUEST.Complete_Echo_of_War_1_times in self.daily_quests:
|
||
logger.info('Reached the limit to get Echo_of_War rewards, continue cause daily quests require it')
|
||
remain = 1
|
||
else:
|
||
logger.info('Reached the limit to get Echo_of_War rewards, stop')
|
||
self.config.task_delay(target=monday)
|
||
self.config.task_stop()
|
||
|
||
self._dungeon_insight(dungeon)
|
||
self._dungeon_enter(dungeon)
|
||
|
||
# Combat
|
||
count = self.dungeon_run(dungeon, wave_limit=min(remain, 3))
|
||
self.is_doing_planner = False
|
||
|
||
logger.attr('achieved_daily_quest', self.achieved_daily_quest)
|
||
logger.attr('achieved_weekly_quest', self.achieved_weekly_quest)
|
||
with self.config.multi_set():
|
||
if count:
|
||
# Check battle pass
|
||
if self.achieved_weekly_quest:
|
||
self.config.task_call('BattlePass')
|
||
# Check daily
|
||
if self.achieved_daily_quest:
|
||
self.config.task_call('DailyQuest')
|
||
# Update dashboard
|
||
self.config.stored.EchoOfWar.add(-count)
|
||
if self.config.stored.EchoOfWar.value < 0:
|
||
logger.warning(f'Invalid EchoOfWar value, revise to 0')
|
||
self.config.stored.EchoOfWar.value = 0
|
||
# Finished all remains
|
||
if count >= remain:
|
||
logger.info('All Echo_of_War rewards got')
|
||
self.config.task_delay(target=monday)
|
||
self.config.task_stop()
|
||
|
||
logger.warning(f'Unexpected Echo_of_War case, count={count}, remain={remain}')
|
||
self.config.task_delay(server_update=True)
|
||
self.config.task_stop()
|