fix: schedule popups

tasks/base/assets/assets_base_popup.py

@@ -135,6 +135,28 @@ ITEM_EXPIRED = ButtonWrapper(
         button=(537, 487, 742, 553),
     ),
 )
+LEVEL_UP = ButtonWrapper(
+    name='LEVEL_UP',
+    jp=None,
+    en=Button(
+        file='./assets/en/base/popup/LEVEL_UP.png',
+        area=(722, 210, 746, 238),
+        search=(702, 190, 766, 258),
+        color=(184, 202, 219),
+        button=(571, 559, 713, 601),
+    ),
+)
+LOCATION_LEVEL_UP = ButtonWrapper(
+    name='LOCATION_LEVEL_UP',
+    jp=None,
+    en=Button(
+        file='./assets/en/base/popup/LOCATION_LEVEL_UP.png',
+        area=(430, 141, 463, 180),
+        search=(410, 121, 483, 200),
+        color=(177, 194, 207),
+        button=(577, 616, 707, 647),
+    ),
+)
 NETWORK_RECONNECT = ButtonWrapper(
     name='NETWORK_RECONNECT',
     jp=Button(

tasks/base/popup.py

@@ -105,3 +105,15 @@ class PopupHandler(ModuleBase):
             return True
 
         return False
+
+    def handle_location_level_up(self, interval=5) -> bool:
+        if self.appear_then_click(LOCATION_LEVEL_UP, interval=interval):
+            return True
+
+        return False
+    
+    def handle_level_up(self, interval=5) -> bool:
+        if self.appear_then_click(LEVEL_UP, interval=interval):
+            return True
+
+        return False

tasks/schedule/ui.py

@@ -18,7 +18,8 @@ class ScheduleUI(UI):
         while 1:
             self.device.screenshot()
             self.appear_then_click(dest_enter, interval=1)
-            self.handle_affection_level_up()
+            if self.handle_affection_level_up() or self.handle_location_level_up():
+                timer.reset()
             if self.appear(dest_check):
                 return True