From b56f25773a1dd88ca5a91f67035efe595f1e609c Mon Sep 17 00:00:00 2001
From: ZYX <chnzyx@qq.com>
Date: Sun, 30 Jun 2024 00:42:00 +0800
Subject: [PATCH] -

---
 diver.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/diver.py b/diver.py
index 69c2031..e8cd268 100644
--- a/diver.py
+++ b/diver.py
@@ -34,6 +34,7 @@ class DivergentUniverse(UniverseUtils):
     def __init__(self, debug=0, nums=-1, speed=0):
         super().__init__()
         self._stop = True
+        self.end = 0
         self.floor = 0
         self.allow_e = 1
         self.count = self.my_cnt = 0
@@ -111,6 +112,11 @@ def loop(self):
                     if static_res != '':
                         print(static_res)
                         break
+        if self.end and res == '加载界面':
+            self.press('esc')
+            time.sleep(2)
+            self.press('esc')
+            self._stop = True
         
     def do_action(self, action) -> int:
         if type(action) == str:
@@ -833,7 +839,7 @@ def end_of_uni(self):
         )
         if self.debug == 0 and self.check_bonus == 0 and self.nums <= self.my_cnt and self.nums >= 0:
             log.info('已完成上限，准备停止运行')
-            self._stop = True
+            self.end = 1
         self.floor = 0
         self.init_floor()
 
@@ -907,6 +913,7 @@ def on_key_press(self, event):
             self.stop()
 
     def start(self):
+        self.__init__()
         self._stop = False
         keyboard.on_press(self.on_key_press)
         self.keys = KeyController(self)