From bbf3bf7c3680c42f2ab8d1645a3fb87908a5d845 Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Mon, 20 Nov 2023 17:06:08 +0800
Subject: [PATCH] perf(bounty): improve ending logic

---
 tasks/bounty/bounty.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/tasks/bounty/bounty.py b/tasks/bounty/bounty.py
index a848167..9798453 100644
--- a/tasks/bounty/bounty.py
+++ b/tasks/bounty/bounty.py
@@ -61,12 +61,11 @@ class Bounty(BountyUI):
         match status:
             case BountyStatus.OCR:
                 if self.get_ticket():
+                    if self.current_ticket == 0 or not self.task:
+                        return BountyStatus.FINISH
                     return BountyStatus.SELECT
             case BountyStatus.SELECT:
                 if not self.is_ticket_enough:
-                    if self.current_ticket == 0:
-                        logger.info('Bounty ticket empty')
-                        return BountyStatus.FINISH
                     logger.warning('Bounty ticket not enough')
                     raise RequestHumanTakeover
                 if self.select_bounty(*self.current_bounty):
@@ -77,8 +76,6 @@ class Bounty(BountyUI):
             case BountyStatus.SWEEP:
                 if self.do_sweep(self.current_count):
                     self.task.pop(0)
-                    if not self.task:
-                        return BountyStatus.FINISH
                     return BountyStatus.END
                 return BountyStatus.ENTER
             case BountyStatus.END: