mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-23 09:01:45 +00:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
from datetime import datetime, time
|
|
|
|
from module.device.app import AppControl
|
|
from module.device.control import Control
|
|
from module.device.screenshot import Screenshot
|
|
from module.handler.assets import GET_MISSION
|
|
from module.logger import logger
|
|
|
|
|
|
class Device(Screenshot, Control, AppControl):
|
|
def handle_night_commission(self, hour=21, threshold=30):
|
|
"""
|
|
Args:
|
|
hour (int): Hour that night commission refresh.
|
|
threshold (int): Seconds around refresh time.
|
|
|
|
Returns:
|
|
bool: If handled.
|
|
"""
|
|
now = datetime.now().time()
|
|
if now < time(hour - 1, 59, 60 - threshold):
|
|
return False
|
|
if now > time(hour, 0, threshold):
|
|
return False
|
|
|
|
if GET_MISSION.match(self.image, offset=True):
|
|
logger.info('Night commission appear.')
|
|
self.click(GET_MISSION)
|
|
return True
|
|
|
|
return False
|
|
|
|
def screenshot(self):
|
|
"""
|
|
Returns:
|
|
PIL.Image.Image:
|
|
"""
|
|
super().screenshot()
|
|
if self.handle_night_commission():
|
|
super().screenshot()
|
|
|
|
return self.image
|