mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-27 02:27:12 +00:00
110 lines
3.0 KiB
Python
110 lines
3.0 KiB
Python
from module.base.base import ModuleBase
|
|
from module.base.timer import Timer
|
|
from module.handler.assets import *
|
|
from module.logger import logger
|
|
|
|
|
|
class InfoHandler(ModuleBase):
|
|
"""
|
|
Class to handle all kinds of message.
|
|
"""
|
|
"""
|
|
Info bar
|
|
"""
|
|
def info_bar_count(self):
|
|
if self.appear(INFO_BAR_3):
|
|
return 3
|
|
elif self.appear(INFO_BAR_2):
|
|
return 2
|
|
elif self.appear(INFO_BAR_1):
|
|
return 1
|
|
else:
|
|
return 0
|
|
|
|
def handle_info_bar(self):
|
|
if self.info_bar_count():
|
|
self.wait_until_disappear(INFO_BAR_1)
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def ensure_no_info_bar(self, timeout=0.6):
|
|
timeout = Timer(timeout)
|
|
timeout.start()
|
|
while 1:
|
|
self.device.screenshot()
|
|
self.handle_info_bar()
|
|
|
|
if timeout.reached():
|
|
break
|
|
|
|
"""
|
|
Popup info
|
|
"""
|
|
_popup_offset = (3, 30)
|
|
|
|
def handle_popup_confirm(self):
|
|
if self.appear(POPUP_CANCEL, offset=self._popup_offset) \
|
|
and self.appear(POPUP_CONFIRM, offset=self._popup_offset, interval=2):
|
|
self.device.click(POPUP_CONFIRM)
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def handle_popup_cancel(self):
|
|
if self.appear(POPUP_CONFIRM, offset=self._popup_offset) \
|
|
and self.appear(POPUP_CANCEL, offset=self._popup_offset, interval=2):
|
|
self.device.click(POPUP_CANCEL)
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def handle_urgent_commission(self, save_get_items=None):
|
|
"""
|
|
Args:
|
|
save_get_items (bool):
|
|
|
|
Returns:
|
|
bool:
|
|
"""
|
|
if save_get_items is None:
|
|
save_get_items = self.config.ENABLE_SAVE_GET_ITEMS
|
|
|
|
appear = self.appear(GET_MISSION, offset=True, interval=2)
|
|
if appear:
|
|
logger.info('Get urgent commission')
|
|
if save_get_items:
|
|
self.handle_info_bar()
|
|
self.device.save_screenshot('get_mission')
|
|
self.device.click(GET_MISSION)
|
|
return appear
|
|
|
|
def handle_combat_low_emotion(self):
|
|
if not self.config.IGNORE_LOW_EMOTION_WARN:
|
|
return False
|
|
|
|
return self.handle_popup_confirm()
|
|
|
|
"""
|
|
Story
|
|
"""
|
|
def story_skip(self):
|
|
if self.handle_popup_confirm():
|
|
return True
|
|
if self.appear_then_click(STORY_SKIP, offset=True, interval=2):
|
|
return True
|
|
if self.appear(STORY_LETTER_BLACK) and self.appear_then_click(STORY_LETTERS_ONLY, offset=True, interval=2):
|
|
return True
|
|
if self.appear_then_click(STORY_CHOOSE, offset=True, interval=2):
|
|
return True
|
|
if self.appear_then_click(STORY_CHOOSE_2, offset=True, interval=2):
|
|
return True
|
|
|
|
return False
|
|
|
|
def handle_story_skip(self):
|
|
if not self.config.ENABLE_MAP_CLEAR_MODE:
|
|
return False
|
|
|
|
return self.story_skip()
|