feat: add canteen stamina claim automation
This commit is contained in:
1 parent
f094798fc8
commit
48d5f63e36
11 files changed
+221
-4
No files matched your search
@@ -0,0 +1,66 @@
|
||||
"""领地食堂体力领取流程。"""
|
||||
import time
|
||||
|
||||
from .engine import WorkflowError
|
||||
|
||||
|
||||
class StaminaRunner:
|
||||
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)
|
||||
self.log('已执行:' + action)
|
||||
|
||||
def close_and_return(self, canteen):
|
||||
self.click('close_canteen', canteen)
|
||||
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('canteen', self.wait('territory'))
|
||||
canteen = self.wait('available', 'empty')
|
||||
if canteen[0] == 'empty':
|
||||
self.log('食堂没有可领取体力,本次跳过。')
|
||||
self.close_and_return(canteen)
|
||||
return 'no_stamina'
|
||||
|
||||
self.click('claim', canteen)
|
||||
reward = self.wait('reward')
|
||||
self.log('已确认获得体力奖励弹框。')
|
||||
self.click('dismiss_reward', reward)
|
||||
empty = self.wait('empty')
|
||||
self.log('已确认食堂进入无可领体力状态。')
|
||||
self.close_and_return(empty)
|
||||
return 'claimed'
|
||||
Reference in new issue
Block a user