304 lines
10 KiB
Python
304 lines
10 KiB
Python
"""
|
|
此文件包含非练习/考试的行动。
|
|
|
|
具体包括：おでかけ、相談、活動支給、授業
|
|
"""
|
|
from logging import getLogger
|
|
|
|
from kotonebot.kaa.config.schema import produce_solution
|
|
from kotonebot.kaa.game_ui import dialog
|
|
|
|
from kotonebot.kaa.tasks import R
|
|
from kotonebot.kaa.config import conf
|
|
from ..produce.common import fast_acquisitions
|
|
from kotonebot.kaa.game_ui.commu_event_buttons import CommuEventButtonUI
|
|
from kotonebot.util import Countdown, Interval
|
|
from kotonebot.errors import UnrecoverableError
|
|
from kotonebot import device, image, action, sleep
|
|
from kotonebot.backend.dispatch import SimpleDispatcher
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
@action('检测是否可以执行活動支給')
|
|
def allowance_available():
|
|
"""
|
|
判断是否可以执行活動支給。
|
|
"""
|
|
return image.find(R.InPurodyuusu.ButtonTextAllowance) is not None
|
|
|
|
@action('检测是否可以执行授業')
|
|
def study_available():
|
|
"""
|
|
判断是否可以执行授業。
|
|
"""
|
|
# [screenshots/produce/action_study1.png]
|
|
return image.find(R.InPurodyuusu.ButtonIconStudy) is not None
|
|
|
|
@action('检测是否可以执行相談')
|
|
def consult_available():
|
|
"""
|
|
判断是否可以执行相談。
|
|
"""
|
|
return image.find(R.InPurodyuusu.ButtonIconConsult) is not None
|
|
|
|
# TODO: 把进入授業的逻辑和执行授業的逻辑分离
|
|
@action('执行授業')
|
|
def enter_study():
|
|
"""
|
|
执行授業。
|
|
|
|
前置条件：位于行动页面，且所有行动按钮清晰可见 \n
|
|
结束状态：选择选项后可能会出现的，比如领取奖励、加载画面等。
|
|
"""
|
|
logger.info("Executing 授業.")
|
|
# [screenshots/produce/action_study1.png]
|
|
logger.debug("Double clicking on 授業.")
|
|
device.double_click(image.expect_wait(R.InPurodyuusu.ButtonIconStudy))
|
|
# 等待进入页面。中间可能会出现未读交流
|
|
# [screenshots/produce/action_study2.png]
|
|
while not image.find(R.InPurodyuusu.IconTitleStudy):
|
|
logger.debug("Waiting for 授業 screen.")
|
|
fast_acquisitions()
|
|
# 首先需要判断是不是自习课
|
|
# [kotonebot-resource\sprites\jp\in_purodyuusu\screenshot_study_self_study.png]
|
|
if image.find_multi([
|
|
R.InPurodyuusu.TextSelfStudyDance,
|
|
R.InPurodyuusu.TextSelfStudyVisual,
|
|
R.InPurodyuusu.TextSelfStudyVocal
|
|
]):
|
|
logger.info("授業 type: Self study.")
|
|
target = produce_solution().data.self_study_lesson
|
|
if target == 'dance':
|
|
logger.debug("Clicking on lesson dance.")
|
|
device.double_click(image.expect(R.InPurodyuusu.TextSelfStudyDance))
|
|
elif target == 'visual':
|
|
logger.debug("Clicking on lesson visual.")
|
|
device.double_click(image.expect(R.InPurodyuusu.TextSelfStudyVisual))
|
|
elif target == 'vocal':
|
|
logger.debug("Clicking on lesson vocal.")
|
|
device.double_click(image.expect(R.InPurodyuusu.TextSelfStudyVocal))
|
|
from ..produce.in_purodyuusu import until_practice_scene, practice
|
|
logger.info("Entering practice scene.")
|
|
until_practice_scene()
|
|
logger.info("Executing practice.")
|
|
practice()
|
|
logger.info("Practice completed.")
|
|
# 不是自习课
|
|
else:
|
|
logger.info("授業 type: Normal.")
|
|
# 获取三个选项的内容
|
|
ui = CommuEventButtonUI()
|
|
buttons = ui.all()
|
|
if not buttons:
|
|
raise UnrecoverableError("Failed to find any buttons.")
|
|
# 选中 +30 的选项
|
|
target_btn = next((btn for btn in buttons if '+30' in btn.description), None)
|
|
if target_btn is None:
|
|
logger.error("Failed to find +30 option. Pick the second button instead.")
|
|
target_btn = buttons[1]
|
|
logger.debug('Clicking "%s".', target_btn.description)
|
|
if target_btn.selected:
|
|
device.click(target_btn)
|
|
else:
|
|
device.double_click(target_btn)
|
|
while fast_acquisitions() is None:
|
|
logger.info("Waiting for acquisitions finished.")
|
|
logger.info("授業 completed.")
|
|
|
|
|
|
@action('执行活動支給')
|
|
def enter_allowance():
|
|
"""
|
|
执行活動支給。
|
|
|
|
前置条件：位于行动页面，且所有行动按钮清晰可见 \n
|
|
结束状态：位于行动页面
|
|
"""
|
|
logger.info("Executing 活動支給.")
|
|
# 点击活動支給 [screenshots\allowance\step_1.png]
|
|
logger.info("Double clicking on 活動支給.")
|
|
device.double_click(image.expect(R.InPurodyuusu.ButtonTextAllowance), interval=1)
|
|
# 等待进入页面
|
|
while not image.find(R.InPurodyuusu.IconTitleAllowance):
|
|
logger.debug("Waiting for 活動支給 screen.")
|
|
fast_acquisitions()
|
|
# 领取奖励
|
|
it = Interval()
|
|
while True:
|
|
# TODO: 检测是否在行动页面应当单独一个函数
|
|
if image.find_multi([
|
|
R.InPurodyuusu.TextPDiary, # 普通周
|
|
R.InPurodyuusu.ButtonFinalPracticeDance # 离考试剩余一周
|
|
]):
|
|
break
|
|
if image.find(R.InPurodyuusu.LootboxSliverLock):
|
|
logger.info("Click on lootbox.")
|
|
device.click()
|
|
sleep(0.5) # 防止点击了第一个箱子后立马点击了第二个
|
|
continue
|
|
if fast_acquisitions() is not None:
|
|
continue
|
|
it.wait()
|
|
logger.info("活動支給 completed.")
|
|
|
|
# TODO: 将逻辑用循环改写
|
|
@action('执行相談', screenshot_mode='manual-inherit')
|
|
def enter_consult():
|
|
"""
|
|
执行相談。
|
|
|
|
前置条件：位于行动页面，且所有行动按钮清晰可见 \n
|
|
结束状态：位于行动页面
|
|
"""
|
|
logger.info("Executing 相談.")
|
|
logger.info("Double clicking on 相談.")
|
|
device.screenshot()
|
|
device.double_click(image.expect(R.InPurodyuusu.ButtonIconConsult), interval=1)
|
|
|
|
# 等待进入页面
|
|
while not image.find(R.InPurodyuusu.IconTitleConsult):
|
|
device.screenshot()
|
|
logger.debug("Waiting for 相談 screen.")
|
|
fast_acquisitions()
|
|
# # 尝试固定购买第一个物品
|
|
# device.click(R.InPurodyuusu.PointConsultFirstItem)
|
|
# sleep(0.5)
|
|
# device.click(image.expect(R.InPurodyuusu.ButtonIconExchange))
|
|
# # 等待弹窗
|
|
# timeout_cd = Countdown(sec=5).start()
|
|
# while not timeout_cd.expired():
|
|
# if dialog.yes():
|
|
# break
|
|
# # 结束
|
|
# while not image.find(R.InPurodyuusu.ButtonEndConsult):
|
|
# fast_acquisitions()
|
|
# device.click(image.expect_wait(R.InPurodyuusu.ButtonEndConsult))
|
|
# # 可能会弹出确认对话框
|
|
# timeout_cd.reset().start()
|
|
# while not timeout_cd.expired():
|
|
# dialog.yes()
|
|
device.click(R.InPurodyuusu.PointConsultFirstItem)
|
|
sleep(0.3)
|
|
it = Interval()
|
|
wait_purchase_cd = Countdown(sec=5)
|
|
exit_cd = Countdown(sec=5)
|
|
purchase_clicked = False
|
|
purchase_confirmed = False
|
|
exit_clicked = False
|
|
while True:
|
|
device.screenshot()
|
|
it.wait()
|
|
if wait_purchase_cd.expired():
|
|
# 等待购买确认对话框超时后直接认为购买完成
|
|
purchase_confirmed = True
|
|
|
|
if dialog.yes():
|
|
if purchase_clicked:
|
|
purchase_confirmed = True
|
|
continue
|
|
elif purchase_confirmed:
|
|
continue
|
|
elif exit_clicked:
|
|
break
|
|
if image.find(R.InPurodyuusu.ButtonIconExchange, colored=True):
|
|
device.click()
|
|
purchase_clicked = True
|
|
continue
|
|
if purchase_confirmed and image.find(R.InPurodyuusu.ButtonEndConsult):
|
|
device.click()
|
|
exit_clicked = True
|
|
exit_cd.start()
|
|
continue
|
|
|
|
# 等待退出对话框超时，直接退出
|
|
if exit_cd.expired():
|
|
break
|
|
|
|
if not purchase_confirmed:
|
|
device.click(R.InPurodyuusu.PointConsultFirstItem)
|
|
# 处理不能购买的情况（超时）
|
|
# TODO: 应当检测画面文字/图标而不是用超时
|
|
wait_purchase_cd.start()
|
|
|
|
logger.info("相談 completed.")
|
|
|
|
@action('判断是否可以休息')
|
|
def is_rest_available():
|
|
"""
|
|
判断是否可以休息。
|
|
"""
|
|
return image.find(R.InPurodyuusu.Rest) is not None
|
|
|
|
|
|
@action('执行休息')
|
|
def rest():
|
|
"""执行休息"""
|
|
logger.info("Rest for this week.")
|
|
(SimpleDispatcher('in_produce.rest')
|
|
# 点击休息
|
|
.click(R.InPurodyuusu.Rest)
|
|
# 确定
|
|
.click(R.InPurodyuusu.RestConfirmBtn, finish=True)
|
|
).run()
|
|
|
|
@action('判断是否处于行动页面')
|
|
def at_action_scene():
|
|
return image.find_multi([
|
|
R.InPurodyuusu.TextPDiary, # 普通周
|
|
R.InPurodyuusu.ButtonFinalPracticeDance # 离考试剩余一周
|
|
]) is not None
|
|
|
|
@action('判断是否可以外出')
|
|
def outing_available():
|
|
"""
|
|
判断是否可以外出（おでかけ）。
|
|
"""
|
|
return image.find(R.InPurodyuusu.ButtonIconOuting) is not None
|
|
|
|
@action('执行外出')
|
|
def enter_outing():
|
|
"""
|
|
执行外出（おでかけ）。
|
|
|
|
前置条件：位于行动页面，且所有行动按钮清晰可见 \n
|
|
结束状态：位于行动页面
|
|
"""
|
|
logger.info("Executing おでかけ.")
|
|
# 点击外出
|
|
logger.info("Double clicking on おでかけ.")
|
|
device.double_click(image.expect(R.InPurodyuusu.ButtonIconOuting))
|
|
# 等待进入页面
|
|
while not image.find(R.InPurodyuusu.TitleIconOuting):
|
|
logger.debug("Waiting for おでかけ screen.")
|
|
fast_acquisitions()
|
|
# 固定选中第二个选项
|
|
# TODO: 可能需要二次处理外出事件
|
|
# [kotonebot-resource\sprites\jp\in_purodyuusu\screenshot_outing.png]
|
|
ui = CommuEventButtonUI()
|
|
buttons = ui.all()
|
|
if not buttons:
|
|
raise UnrecoverableError("Failed to find any buttons.")
|
|
target_btn = buttons[min(1, len(buttons) - 1)]
|
|
logger.debug('Clicking "%s".', target_btn.description)
|
|
if target_btn.selected:
|
|
device.click(target_btn)
|
|
else:
|
|
device.double_click(target_btn)
|
|
it = Interval()
|
|
while True:
|
|
device.screenshot()
|
|
if at_action_scene():
|
|
break
|
|
elif fast_acquisitions():
|
|
pass
|
|
# [screenshots\produce\outing_ap_confirm.png]
|
|
elif image.find(R.Common.ButtonSelect2):
|
|
logger.info("AP max out dialog found. Click to continue.")
|
|
device.click()
|
|
sleep(0.1)
|
|
it.wait()
|
|
|
|
logger.info("おでかけ completed.")
|
|
|
|
if __name__ == '__main__':
|
|
enter_consult() |