mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-30 11:19:30 +00:00
92 lines
3.6 KiB
Python
92 lines
3.6 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from module.base.timer import Timer
|
|
from module.logger import logger
|
|
from tasks.base.assets.assets_base_popup import GET_REWARD
|
|
from tasks.combat.interact import CombatInteract
|
|
from tasks.dungeon.state import DungeonState
|
|
from tasks.rogue.assets.assets_rogue_reward import REWARD_CLOSE, USE_IMMERSIFIER, USE_STAMINA
|
|
from tasks.rogue.blessing.ui import RogueUI
|
|
|
|
|
|
class RogueReward(RogueUI, CombatInteract, DungeonState):
|
|
def claim_domain_reward(
|
|
self,
|
|
use_trailblaze_power=False,
|
|
use_immersifier=True,
|
|
skip_first_screenshot=True
|
|
):
|
|
"""
|
|
Pages:
|
|
in: page_main, DUNGEON_COMBAT_INTERACT, near immersifier
|
|
"""
|
|
logger.hr('Claim domain reward', level=2)
|
|
logger.info(f'use_trailblaze_power={use_trailblaze_power}, use_immersifier={use_immersifier}')
|
|
if not use_trailblaze_power and not use_immersifier:
|
|
return
|
|
|
|
confirm = Timer(0.6, count=2).start()
|
|
exhausted = False
|
|
while 1:
|
|
if skip_first_screenshot:
|
|
skip_first_screenshot = False
|
|
else:
|
|
self.device.screenshot()
|
|
|
|
if self.is_in_main():
|
|
if confirm.reached():
|
|
break
|
|
else:
|
|
confirm.reset()
|
|
|
|
if not exhausted and self.handle_combat_interact():
|
|
self.interval_clear(USE_STAMINA)
|
|
confirm.reset()
|
|
continue
|
|
if self.handle_reward():
|
|
self.interval_clear(USE_STAMINA)
|
|
confirm.reset()
|
|
continue
|
|
if self.appear(REWARD_CLOSE, interval=2):
|
|
self.dungeon_update_stamina()
|
|
if use_trailblaze_power and self.config.stored.TrailblazePower.value >= 40:
|
|
self.device.click(USE_STAMINA)
|
|
self.interval_reset(USE_STAMINA)
|
|
self.interval_clear(GET_REWARD)
|
|
confirm.reset()
|
|
continue
|
|
elif use_immersifier and self.config.stored.Immersifier.value > 0:
|
|
self.device.click(USE_IMMERSIFIER)
|
|
self.interval_reset(USE_STAMINA)
|
|
self.interval_clear(GET_REWARD)
|
|
confirm.reset()
|
|
continue
|
|
else:
|
|
logger.info('Cannot claim more rewards')
|
|
self.device.click(REWARD_CLOSE)
|
|
self.interval_reset(USE_STAMINA)
|
|
confirm.reset()
|
|
exhausted = True
|
|
continue
|
|
|
|
def can_claim_domain_reward(
|
|
self,
|
|
use_trailblaze_power=False,
|
|
use_immersifier=True
|
|
):
|
|
if not use_trailblaze_power and not use_immersifier:
|
|
logger.info('Cannot claim domain reward, as all disabled')
|
|
return False
|
|
if use_immersifier:
|
|
if self.config.stored.Immersifier.value > 0:
|
|
logger.info(f'Can claim domain reward, got immersifiers')
|
|
return True
|
|
if datetime.now() - self.config.stored.Immersifier.time > timedelta(minutes=15):
|
|
logger.info(f'Can claim domain reward, immersifier record outdated')
|
|
return True
|
|
if use_trailblaze_power and self.config.stored.TrailblazePower.predict_current() >= 40:
|
|
logger.info(f'Can claim domain reward, got enough trailblaze power')
|
|
return True
|
|
logger.info('Cannot claim domain reward, requirements not satisfied')
|
|
return False
|