style: format

tasks/momotalk/momotalk.py

@@ -2,9 +2,9 @@ from enum import Enum
 
 from module.base.timer import Timer
 from module.logger import logger
-from tasks.tactical_challenge.assets.assets_tactical_challenge import *
 from tasks.momotalk.ui import MomoTalkUI
 
+
 class MomoTalkStatus(Enum):
     OPEN = 0
     SORT = 1
@@ -13,6 +13,7 @@ class MomoTalkStatus(Enum):
     STORY = 4
     FINISHED = -1
 
+
 class MomoTalk(MomoTalkUI):
     def handle_momotalk(self, status):
         match status:
@@ -26,16 +27,16 @@ class MomoTalk(MomoTalkUI):
             case MomoTalkStatus.CHECK:
                 if self.check_first_student():
                     return MomoTalkStatus.CHAT
-                return MomoTalkStatus.FINISHED                
+                return MomoTalkStatus.FINISHED
             case MomoTalkStatus.CHAT:
                 if self.chat():
                     return MomoTalkStatus.STORY
                 return MomoTalkStatus.OPEN
             case MomoTalkStatus.STORY:
                 if self.skip_story():
-                    return MomoTalkStatus.CHAT    
+                    return MomoTalkStatus.CHAT
             case MomoTalkStatus.FINISHED:
-                return status                
+                return status
             case _:
                 logger.warning(f'Invalid status: {status}')
         return status
@@ -58,4 +59,3 @@ class MomoTalk(MomoTalkUI):
                 break
 
         self.config.task_delay(server_update=True)
-

tasks/momotalk/ui.py

@@ -28,6 +28,7 @@ REPLY_TEMPLATE = REPLY.matched_button.image
 STORY_TEMPLATE = STORY.matched_button.image
 CHATTING_TEMPLATE = CHATTING.matched_button.image
 
+
 class MomoTalkUI(UI):
     def __init__(self, config, device):
         super().__init__(config, device)
@@ -81,7 +82,7 @@ class MomoTalkUI(UI):
         switch.set(state, main=self)
 
         return True
-    
+
     def click_all(self, template, x_add=0, y_add=0):
         """
         Find the all the locations of the template adding an offset if specified and click them. 
@@ -116,7 +117,7 @@ class MomoTalkUI(UI):
             return True
         logger.warning("No students available for interaction")
         return False
-    
+
     def sort_messages(self):
         """
         Switch from newest to unread and sort the messages in descending order
@@ -140,7 +141,7 @@ class MomoTalkUI(UI):
             return True
         logger.warning("No students available for interaction")
         return False
-    
+
     def chat(self):
         """
         Waits for the chat area to be stable and then 
@@ -164,8 +165,8 @@ class MomoTalkUI(UI):
                 timer.reset()
             elif timer.reached():
                 logger.info("No new message detected")
-                return False        
-                
+                return False
+
     def skip_story(self):
         """
         Skip story by executing a series of steps. Returns True if the confirm skip
@@ -184,5 +185,3 @@ class MomoTalkUI(UI):
                     while not timer.reached_and_reset():
                         pass
                     break
-
-