100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
"""检测与跳过交流"""
|
|
import logging
|
|
|
|
from cv2.typing import MatLike
|
|
|
|
from .. import R
|
|
from ..game_ui import dialog
|
|
from kotonebot.util import Interval, Countdown
|
|
from kotonebot.kaa.game_ui import WhiteFilter
|
|
from kotonebot import device, image, user, action, use_screenshot
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
@action('获取 SKIP 按钮', screenshot_mode='manual-inherit')
|
|
def skip_button():
|
|
device.screenshot()
|
|
return image.find(
|
|
R.Common.ButtonCommuSkip,
|
|
threshold=0.6,
|
|
) or image.find(
|
|
R.Common.ButtonCommuSkip,
|
|
threshold=0.6,
|
|
preprocessors=[WhiteFilter()]
|
|
)
|
|
|
|
@action('获取 FASTFORWARD 按钮', screenshot_mode='manual-inherit')
|
|
def fastforward_button():
|
|
device.screenshot()
|
|
return image.find(
|
|
R.Common.ButtonCommuFastforward,
|
|
threshold=0.6,
|
|
) or image.find(
|
|
R.Common.ButtonCommuFastforward,
|
|
threshold=0.6,
|
|
preprocessors=[WhiteFilter()]
|
|
)
|
|
|
|
@action('检查是否处于交流')
|
|
def is_at_commu():
|
|
return skip_button() is not None
|
|
|
|
@action('检查未读交流', screenshot_mode='manual')
|
|
def handle_unread_commu(img: MatLike | None = None) -> bool:
|
|
"""
|
|
检查当前是否处在未读交流，并自动跳过。
|
|
|
|
:param img: 截图。
|
|
:return: 是否跳过了交流。
|
|
"""
|
|
logger.debug('Check and skip commu')
|
|
img = use_screenshot(img)
|
|
|
|
if skip := skip_button():
|
|
# SKIP 按钮至少出现 3s 才处理
|
|
hit_cd = Countdown(sec=3).start()
|
|
while not hit_cd.expired():
|
|
device.screenshot()
|
|
if skip_button() is None:
|
|
logger.info('Commu disappeared.')
|
|
return False
|
|
device.click(skip)
|
|
logger.debug('Clicked skip button.')
|
|
return True
|
|
# 有时会碰见只有快进按钮的交流
|
|
# [screenshots/produce/in_produce/pre_final_exam_commu.png]
|
|
if fastforward := fastforward_button():
|
|
device.click(fastforward)
|
|
logger.debug('Clicked fastforward button.')
|
|
# 即使点了跳过，画面上也有可能还有其他东西需要处理
|
|
# 因此返回 False 而不是 True
|
|
return False
|
|
if image.find_multi([
|
|
R.Common.TextSkipCommuComfirmation,
|
|
R.Common.TextFastforwardCommuDialogTitle
|
|
]):
|
|
logger.info('Unread commu found.')
|
|
if dialog.yes():
|
|
logger.debug('Clicked confirm button.')
|
|
logger.debug('Pushing notification...')
|
|
user.info('发现未读交流', images=[img])
|
|
return True
|
|
else:
|
|
return False
|
|
return False
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import logging
|
|
logging.basicConfig(level=logging.INFO, format='[%(asctime)s] [%(levelname)s] [%(name)s] [%(funcName)s] [%(lineno)d] %(message)s')
|
|
logger.setLevel(logging.DEBUG)
|
|
from kotonebot.backend.context import manual_context, inject_context
|
|
from kotonebot.backend.debug.mock import MockDevice
|
|
manual_context().begin()
|
|
_md = MockDevice()
|
|
_md.load_image(r"D:\a.png")
|
|
inject_context(device=_md)
|
|
print(is_at_commu())
|
|
# while True:
|
|
# print(handle_unread_commu())
|