145 lines
5.8 KiB
Python
145 lines
5.8 KiB
Python
"""竞赛"""
|
||
import logging
|
||
from gettext import gettext as _
|
||
|
||
from .. import R
|
||
from ..common import conf
|
||
from ..game_ui import WhiteFilter
|
||
from ..actions.scenes import at_home, goto_home
|
||
from ..actions.loading import wait_loading_end
|
||
from kotonebot import device, image, ocr, color, action, task, user, rect_expand, sleep, contains
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
@action('前往竞赛页面')
|
||
def goto_contest() -> bool:
|
||
"""
|
||
前置条件：位于首页 \n
|
||
结束状态：位于竞赛界面，且已经点击了各种奖励领取提示
|
||
|
||
:return: 是否存在未完成的挑战
|
||
"""
|
||
# TODO: 优化这一部分，加入区域检测，提高速度
|
||
device.click(image.expect(R.Common.ButtonContest))
|
||
ocr.expect_wait(contains('CONTEST'))
|
||
btn_contest = ocr.expect_wait(contains('コンテスト'))
|
||
has_ongoing_contest = image.find(R.Daily.TextContestLastOngoing) is not None
|
||
device.click(btn_contest)
|
||
if not has_ongoing_contest:
|
||
while not image.find(R.Daily.ButtonContestRanking):
|
||
# [kotonebot-resource\sprites\jp\daily\screenshot_contest_season_reward.png]
|
||
# [screenshots/contest/acquire2.png]
|
||
device.click(R.Daily.PointDissmissContestReward)
|
||
sleep(1)
|
||
# [screenshots/contest/main.png]
|
||
else:
|
||
image.expect_wait(R.Daily.ButtonContestChallengeStart)
|
||
return has_ongoing_contest
|
||
|
||
@action('选择并挑战')
|
||
def pick_and_contest(has_ongoing_contest: bool = False) -> bool:
|
||
"""
|
||
选择并挑战
|
||
|
||
前置条件：位于竞赛界面 \n
|
||
结束状态：位于竞赛界面
|
||
|
||
:param has_ongoing_contest: 是否有中断未完成的挑战
|
||
:return: 如果返回假，说明今天挑战次数已经用完了
|
||
"""
|
||
# 判断是否有中断未完成的挑战
|
||
# [screenshots/contest/ongoing.png]
|
||
if not has_ongoing_contest:
|
||
image.expect_wait(R.Daily.ButtonContestRanking)
|
||
sleep(3) # 等待动画
|
||
# 随机选一个对手 [screenshots/contest/main.png]
|
||
logger.debug('Clicking on contestant.')
|
||
contestant_list = image.find_all(R.Daily.TextContestOverallStats)
|
||
if contestant_list is None or len(contestant_list) == 0:
|
||
logger.info('No contestant found. Today\'s challenge points used up.')
|
||
return False
|
||
# 按照y坐标从上到下排序对手列表
|
||
contestant_list.sort(key=lambda x: x.position[1])
|
||
if len(contestant_list) != 3:
|
||
logger.warning('Cannot find all 3 contestants.')
|
||
# 选择配置文件中对应的对手顺序（1最强，3最弱）
|
||
target = conf().contest.select_which_contestant
|
||
if target >= 1 and target <= 3 and target <= len(contestant_list):
|
||
target -= 1 # [1, 3]映射至[0, 2]
|
||
else:
|
||
target = 0 # 出错则默认选择第一个
|
||
contestant = contestant_list[target]
|
||
logger.info('Picking up contestant #%d.', target + 1)
|
||
device.click(contestant)
|
||
# 挑战开始 [screenshots/contest/start1.png]
|
||
logger.debug('Clicking on start button.')
|
||
device.click(image.expect_wait(R.Daily.ButtonContestStart))
|
||
sleep(3) # 多延迟一点
|
||
# 进入挑战页面 [screenshots/contest/contest1.png]
|
||
# [screenshots/contest/contest2.png]
|
||
while not image.find(R.Daily.ButtonContestChallengeStart):
|
||
# 记忆未编成 [screenshots/contest/no_memo.png]
|
||
if image.find(R.Daily.TextContestNoMemory):
|
||
logger.debug('Memory not set. Using auto-compilation.')
|
||
user.warning('竞赛未编成', _('记忆未编成。将使用自动编成。'), once=True)
|
||
device.click(image.expect(R.Daily.ButtonContestChallenge))
|
||
logger.debug('Waiting for challenge start screen.')
|
||
# 勾选跳过所有
|
||
if image.find(R.Common.CheckboxUnchecked):
|
||
logger.debug('Checking skip all.')
|
||
device.click()
|
||
sleep(0.5)
|
||
# 点击 SKIP
|
||
logger.debug('Clicking on SKIP.')
|
||
device.click(image.expect_wait(R.Daily.ButtonIconSkip, timeout=10, preprocessors=[WhiteFilter()]))
|
||
while not image.wait_for(R.Common.ButtonNextNoIcon, timeout=2):
|
||
device.click_center()
|
||
logger.debug('Waiting for the result.')
|
||
# [screenshots/contest/after_contest1.png]
|
||
# 点击 次へ [screenshots/contest/after_contest2.png]
|
||
logger.debug('Challenge finished. Clicking on next.')
|
||
device.click()
|
||
# 点击 終了 [screenshots/contest/after_contest3.png]
|
||
logger.debug('Clicking on end.')
|
||
device.click(image.expect_wait(R.Common.ButtonEnd))
|
||
# 可能出现的奖励弹窗 [screenshots/contest/after_contest4.png]
|
||
sleep(1)
|
||
if image.find(R.Common.ButtonClose):
|
||
logger.debug('Clicking on close.')
|
||
device.click()
|
||
# 等待返回竞赛界面
|
||
wait_loading_end()
|
||
image.expect_wait(R.Daily.ButtonContestRanking)
|
||
logger.info('Challenge finished.')
|
||
return True
|
||
|
||
@task('竞赛')
|
||
def contest():
|
||
""""""
|
||
if not conf().contest.enabled:
|
||
logger.info('Contest is disabled.')
|
||
return
|
||
logger.info('Contest started.')
|
||
if not at_home():
|
||
goto_home()
|
||
sleep(0.3)
|
||
btn_contest = image.expect(R.Common.ButtonContest)
|
||
notification_dot = rect_expand(btn_contest.rect, top=35, right=35)
|
||
if not color.find('#ff104a', rect=notification_dot):
|
||
logger.info('No action needed.')
|
||
return
|
||
has_ongoing_contest = goto_contest()
|
||
while pick_and_contest(has_ongoing_contest):
|
||
sleep(1.3)
|
||
has_ongoing_contest = False
|
||
goto_home()
|
||
logger.info('Contest all finished.')
|
||
|
||
|
||
if __name__ == '__main__':
|
||
import logging
|
||
logging.basicConfig(level=logging.INFO, format='[%(asctime)s] [%(levelname)s] [%(name)s] [%(funcName)s] [%(lineno)d] %(message)s')
|
||
logging.getLogger('kotonebot').setLevel(logging.DEBUG)
|
||
logger.setLevel(logging.DEBUG)
|
||
|
||
contest() |