63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
"""领地巅峰决斗竞技场积分领取流程。"""
|
|
import time
|
|
|
|
from .engine import WorkflowError
|
|
|
|
|
|
class ArenaRunner:
|
|
def __init__(self, desktop, matcher, flow, region_provider, log=print, timeout=12):
|
|
self.desktop = desktop
|
|
self.matcher = matcher
|
|
self.flow = flow
|
|
self.region_provider = region_provider
|
|
self.log = log
|
|
self.timeout = timeout
|
|
|
|
def wait(self, *names):
|
|
deadline = time.monotonic() + self.timeout
|
|
previous, stable = None, 0
|
|
while time.monotonic() < deadline:
|
|
screenshot = self.desktop.screenshot()
|
|
region = self.region_provider(screenshot)
|
|
frame = self.matcher.frame(screenshot, region)
|
|
found = next((name for name in names
|
|
if self.matcher.matches(frame, self.flow['states'][name])), None)
|
|
stable = stable + 1 if found and found == previous else 1 if found else 0
|
|
previous = found
|
|
if stable >= 2:
|
|
return found, screenshot, region
|
|
time.sleep(0.25)
|
|
raise WorkflowError('未识别到预期页面 ' + '/'.join(names) + '，已停止。')
|
|
|
|
def click(self, action, observation):
|
|
_, screenshot, region = observation
|
|
if self.region_provider(screenshot) != region:
|
|
raise WorkflowError('识别后游戏窗口移动，已停止，请重新运行。')
|
|
self.desktop.click(self.flow['clicks'][action], region, screenshot.size, hold=0.8)
|
|
self.log('已执行：' + action)
|
|
|
|
def return_home(self):
|
|
self.click('arena_back', self.wait('arena_page'))
|
|
self.click('territory_back', self.wait('territory'))
|
|
self.wait('home')
|
|
self.log('已确认返回首页。')
|
|
|
|
def run(self, dry_run=False):
|
|
home = self.wait('home')
|
|
if dry_run:
|
|
self.log('首页识别通过。计划：领地 → 巅峰决斗 → 竞技场 → 领取竞技积分 → 返回首页。未点击。')
|
|
return 'preview'
|
|
|
|
self.click('territory', home)
|
|
self.click('peak_duel', self.wait('territory'))
|
|
self.click('arena', self.wait('selection'))
|
|
arena = self.wait('arena_page')
|
|
self.click('claim_points', arena)
|
|
reward = self.wait('reward')
|
|
self.log('已确认竞技积分奖励弹框。')
|
|
self.click('dismiss_reward', reward)
|
|
self.wait('arena_page')
|
|
self.log('已确认竞技积分领取完成。')
|
|
self.return_home()
|
|
return 'claimed'
|