From 79ea59d60ca3e41c4d0a114066640244920270ce Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Fri, 3 Nov 2023 22:39:23 +0800
Subject: [PATCH] perf: improve cafe reward stability

---
 tasks/cafe/cafe.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/tasks/cafe/cafe.py b/tasks/cafe/cafe.py
index 5f27acb..1f9225a 100644
--- a/tasks/cafe/cafe.py
+++ b/tasks/cafe/cafe.py
@@ -133,13 +133,14 @@ class Cafe(UI):
                 if self.appear_then_click(CHECK_REWARD):
                     return CafeStatus.REWARD
             case CafeStatus.REWARD:
+                if not self.appear(GET_REWARD):
+                    self.click_with_interval(CHECK_REWARD, interval=6)
+                    return status
                 if self.match_color(GOT_REWARD):
                     self.device.click(GET_REWARD_CLOSE)
                     return CafeStatus.GOT
-                if not self.appear(GET_REWARD):
-                    return CafeStatus.OCR
                 if self.match_color(GET_REWARD):
-                    self.device.click(GET_REWARD)
+                    self.click_with_interval(GET_REWARD)
             case CafeStatus.GOT:
                 logger.info('Cafe reward have been got')
                 self.appear_then_click(GET_REWARD_CLOSE)