442 lines
16 KiB
Python
442 lines
16 KiB
Python
import logging
|
||
from itertools import cycle
|
||
from typing import Optional, Literal
|
||
from typing_extensions import assert_never
|
||
|
||
from kotonebot.kaa.config.schema import produce_solution
|
||
from kotonebot.ui import user
|
||
from kotonebot.kaa.tasks import R
|
||
from kotonebot.kaa.config import conf
|
||
from kotonebot.kaa.game_ui import dialog
|
||
from ..actions.scenes import at_home, goto_home
|
||
from kotonebot.backend.loop import Loop, StatedLoop
|
||
from kotonebot.util import Countdown, Interval, Throttler
|
||
from kotonebot.kaa.game_ui.primary_button import find_button
|
||
from kotonebot.kaa.game_ui.idols_overview import locate_idol, match_idol
|
||
from ..produce.in_purodyuusu import hajime_pro, hajime_regular, hajime_master, resume_pro_produce, resume_regular_produce, \
|
||
resume_master_produce
|
||
from kotonebot import device, image, ocr, task, action, sleep, contains, regex
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
def format_time(seconds):
|
||
minutes = int(seconds // 60)
|
||
seconds = int(seconds % 60)
|
||
return f"{minutes}m {seconds}s"
|
||
|
||
def unify(arr: list[int]):
|
||
# 先对数组进行排序
|
||
arr.sort()
|
||
result = []
|
||
i = 0
|
||
while i < len(arr):
|
||
# 将当前元素加入结果
|
||
result.append(arr[i])
|
||
# 跳过所有与当前元素相似的元素
|
||
j = i + 1
|
||
while j < len(arr) and abs(arr[j] - arr[i]) <= 10:
|
||
j += 1
|
||
i = j
|
||
return result
|
||
|
||
@action('选择P偶像', screenshot_mode='manual-inherit')
|
||
def select_idol(skin_id: str):
|
||
"""
|
||
选择目标P偶像
|
||
|
||
前置条件：偶像选择页面 1.アイドル選択\n
|
||
结束状态：偶像选择页面 1.アイドル選択\n
|
||
"""
|
||
logger.info("Find and select idol: %s", skin_id)
|
||
# 进入总览
|
||
device.screenshot()
|
||
it = Interval()
|
||
while not image.find(R.Common.ButtonConfirmNoIcon):
|
||
if image.find(R.Produce.ButtonPIdolOverview):
|
||
device.click()
|
||
device.screenshot()
|
||
it.wait()
|
||
# 选择偶像
|
||
pos = locate_idol(skin_id)
|
||
if pos is None:
|
||
raise ValueError(f"Idol {skin_id} not found.")
|
||
# 确认
|
||
it.reset()
|
||
while btn_confirm := image.find(R.Common.ButtonConfirmNoIcon):
|
||
device.click(pos)
|
||
sleep(0.3)
|
||
device.click(btn_confirm)
|
||
it.wait()
|
||
|
||
@action('培育开始.编成翻页', screenshot_mode='manual-inherit')
|
||
def select_set(index: int):
|
||
"""
|
||
选择指定编号的支援卡/回忆编成。
|
||
|
||
前置条件：STEP 2/3 页面
|
||
结束状态：STEP 2/3 页面
|
||
|
||
:param index: 支援卡/回忆编成的编号，从 1 开始。
|
||
"""
|
||
def _current():
|
||
numbers = []
|
||
while not numbers:
|
||
device.screenshot()
|
||
numbers = ocr.ocr(rect=R.Produce.BoxSetCountIndicator).squash().numbers()
|
||
if not numbers:
|
||
logger.warning('Failed to get current set number. Retrying...')
|
||
sleep(0.2)
|
||
return numbers[0]
|
||
|
||
max_retries = 3
|
||
retry_count = 0
|
||
|
||
while retry_count < max_retries:
|
||
current = _current()
|
||
logger.info(f'Navigate to set #{index}. Now at set #{current}.')
|
||
|
||
# 计算需要点击的次数
|
||
click_count = abs(index - current)
|
||
if click_count == 0:
|
||
logger.info(f'Already at set #{current}.')
|
||
return
|
||
click_target = R.Produce.PointProduceNextSet if current < index else R.Produce.PointProducePrevSet
|
||
|
||
# 点击
|
||
for _ in range(click_count):
|
||
device.click(click_target)
|
||
sleep(0.1)
|
||
|
||
# 确认
|
||
final_current = _current()
|
||
if final_current == index:
|
||
logger.info(f'Arrived at set #{final_current}.')
|
||
return
|
||
else:
|
||
retry_count += 1
|
||
logger.warning(f'Failed to navigate to set #{index}. Current set is #{final_current}. Retrying... ({retry_count}/{max_retries})')
|
||
|
||
logger.error(f'Failed to navigate to set #{index} after {max_retries} retries.')
|
||
|
||
@action('继续当前培育', screenshot_mode='manual-inherit')
|
||
def resume_produce():
|
||
"""
|
||
继续当前培育
|
||
|
||
前置条件：游戏首页，且当前有进行中培育\n
|
||
结束状态：游戏首页
|
||
"""
|
||
device.screenshot()
|
||
# 点击 プロデュース中
|
||
# [res/sprites/jp/daily/home_1.png]
|
||
logger.info('Click ongoing produce button.')
|
||
device.click(R.Produce.BoxProduceOngoing)
|
||
btn_resume = image.expect_wait(R.Produce.ButtonResume)
|
||
# 判断信息
|
||
mode_result = image.find_multi([
|
||
R.Produce.ResumeDialogRegular,
|
||
R.Produce.ResumeDialogPro,
|
||
R.Produce.ResumeDialogMaster
|
||
])
|
||
if not mode_result:
|
||
raise ValueError('Failed to detect produce mode.')
|
||
if mode_result.index == 0:
|
||
mode = 'regular'
|
||
elif mode_result.index == 1:
|
||
mode = 'pro'
|
||
else:
|
||
mode = 'master'
|
||
logger.info(f'Produce mode: {mode}')
|
||
retry_count = 0
|
||
max_retries = 5
|
||
current_week = None
|
||
while retry_count < max_retries:
|
||
week_text = ocr.ocr(R.Produce.BoxResumeDialogWeeks, lang='en').squash().regex(r'\d+/\d+')
|
||
if week_text:
|
||
weeks = week_text[0].split('/')
|
||
logger.info(f'Current week: {weeks[0]}/{weeks[1]}')
|
||
if len(weeks) >= 2:
|
||
current_week = int(weeks[0])
|
||
break
|
||
retry_count += 1
|
||
logger.warning(f'Failed to detect weeks. week_text="{week_text}". Retrying... ({retry_count}/{max_retries})')
|
||
sleep(0.5)
|
||
device.screenshot()
|
||
|
||
if retry_count >= max_retries:
|
||
raise ValueError('Failed to detect weeks after multiple retries.')
|
||
if current_week is None:
|
||
raise ValueError('Failed to detect current_week.')
|
||
# 点击 再開する
|
||
# [kotonebot-resource/sprites/jp/produce/produce_resume.png]
|
||
logger.info('Click resume button.')
|
||
device.click(btn_resume)
|
||
match mode:
|
||
case 'regular':
|
||
resume_regular_produce(current_week)
|
||
case 'pro':
|
||
resume_pro_produce(current_week)
|
||
case 'master':
|
||
resume_master_produce(current_week)
|
||
case _:
|
||
assert_never(mode)
|
||
|
||
@action('执行培育', screenshot_mode='manual-inherit')
|
||
def do_produce(
|
||
idol_skin_id: str,
|
||
mode: Literal['regular', 'pro', 'master'],
|
||
memory_set_index: Optional[int] = None
|
||
) -> bool:
|
||
"""
|
||
进行培育流程
|
||
|
||
前置条件：可导航至首页的任意页面\n
|
||
结束状态：游戏首页\n
|
||
|
||
:param memory_set_index: 回忆编成编号。
|
||
:param idol_skin_id: 要培育的偶像。如果为 None，则使用配置文件中的偶像。
|
||
:param mode: 培育模式。
|
||
:return: 是否因为 AP 不足而跳过本次培育。
|
||
:raises ValueError: 如果 `memory_set_index` 不在 [1, 10] 的范围内。
|
||
"""
|
||
if memory_set_index is not None and not 1 <= memory_set_index <= 10:
|
||
raise ValueError('`memory_set_index` must be in range [1, 10].')
|
||
|
||
if not at_home():
|
||
goto_home()
|
||
|
||
device.screenshot()
|
||
# 点击培育按钮，然后判断是新开还是再开培育
|
||
for _ in Loop(interval=0.6):
|
||
if image.find(R.Produce.TitleIconProudce):
|
||
# 新开
|
||
break
|
||
elif image.find(R.Produce.ButtonResume):
|
||
# 再开
|
||
resume_produce()
|
||
return True
|
||
else:
|
||
device.click(R.Produce.BoxProduceOngoing)
|
||
sleep(2)
|
||
|
||
# 0. 进入培育页面
|
||
logger.info(f'Enter produce page. Mode: {mode}')
|
||
match mode:
|
||
case 'regular':
|
||
target_buttons = [R.Produce.ButtonHajime0Regular, R.Produce.ButtonHajime1Regular]
|
||
case 'pro':
|
||
target_buttons = [R.Produce.ButtonHajime0Pro, R.Produce.ButtonHajime1Pro]
|
||
case 'master':
|
||
target_buttons = [R.Produce.ButtonHajime1Master]
|
||
case _:
|
||
assert_never(mode)
|
||
result = None
|
||
for _ in Loop():
|
||
if image.find(R.Produce.ButtonProduce):
|
||
device.click()
|
||
elif image.find_multi(target_buttons):
|
||
device.click()
|
||
elif image.find(R.Produce.ButtonPIdolOverview):
|
||
result = True
|
||
break
|
||
elif image.find(R.Produce.TextAPInsufficient):
|
||
result = False
|
||
break
|
||
if not result:
|
||
if produce_solution().data.use_ap_drink:
|
||
# [kotonebot-resource\sprites\jp\produce\screenshot_no_enough_ap_1.png]
|
||
# [kotonebot-resource\sprites\jp\produce\screenshot_no_enough_ap_2.png]
|
||
# [kotonebot-resource\sprites\jp\produce\screenshot_no_enough_ap_3.png]
|
||
logger.info('AP insufficient. Try to use AP drink.')
|
||
it = Interval()
|
||
while True:
|
||
if image.find(R.Produce.ButtonUse):
|
||
device.click()
|
||
elif image.find(R.Produce.ButtonRefillAP):
|
||
device.click()
|
||
elif image.find_multi(target_buttons):
|
||
device.click()
|
||
elif image.find(R.Produce.ButtonPIdolOverview):
|
||
break
|
||
device.screenshot()
|
||
it.wait()
|
||
else:
|
||
logger.info('AP insufficient. Exiting produce.')
|
||
device.click(image.expect_wait(R.InPurodyuusu.ButtonCancel))
|
||
return False
|
||
|
||
idol_located = False
|
||
memory_set_selected = False
|
||
support_auto_set_done = False
|
||
next_throttler = Throttler(interval=4)
|
||
for lp in StatedLoop[Literal[0, 1, 2, 3]]():
|
||
if image.find(R.Produce.TextStepIndicator1):
|
||
lp.state = 1
|
||
|
||
if lp.state == 0:
|
||
pass
|
||
# 1. 选择 PIdol [screenshots/produce/screenshot_produce_start_1_p_idol.png]
|
||
if lp.state == 1:
|
||
if image.find(R.Produce.TextStepIndicator2):
|
||
lp.state = 2
|
||
continue
|
||
if lp.when(R.Produce.TextAnotherIdolAvailableDialog):
|
||
dialog.no(msg='Closed another idol available dialog.')
|
||
# 首先判断是否已选中目标偶像
|
||
img = lp.screenshot
|
||
x, y, w, h = R.Produce.BoxSelectedIdol.xywh
|
||
if img is not None and match_idol(idol_skin_id, img[y:y+h, x:x+w]):
|
||
logger.info('Idol %s selected.', idol_skin_id)
|
||
idol_located = True
|
||
# 如果没有，才选择
|
||
if not idol_located:
|
||
select_idol(idol_skin_id)
|
||
idol_located = True
|
||
|
||
# 下一步「次へ」
|
||
if idol_located and find_button(R.Common.ButtonNextNoIcon, True) and next_throttler.request():
|
||
device.click()
|
||
# 2. 选择支援卡 自动编成 [screenshots/produce/screenshot_produce_start_2_support_card.png]
|
||
elif lp.state == 2:
|
||
if image.find(R.Produce.TextStepIndicator3):
|
||
lp.state = 3
|
||
continue
|
||
|
||
# 下一步「次へ」
|
||
if find_button(R.Common.ButtonNextNoIcon, True) and next_throttler.request():
|
||
device.click()
|
||
# 今天仍然有租用回忆次数提示（第三步的提示）
|
||
# （第二步选完之后点「次へ」大概率会卡几秒钟，这个时候脚本很可能会重复点击，
|
||
# 卡住时候的点击就会在第三步生效，出现这个提示。而此时脚本仍然处于第二步，
|
||
# 这样就会报错，或者出现误自动编成。因此需要在第二步里处理掉这个对话框。
|
||
# 理论上应该避免这种情况，但是没找到办法，只能这样 workaround 了。）
|
||
elif image.find(R.Produce.TextRentAvailable):
|
||
dialog.no(msg='Closed rent available dialog. (Step 2)')
|
||
# 确认自动编成提示
|
||
elif image.find(R.Produce.TextAutoSet):
|
||
dialog.yes(msg='Confirmed auto set.')
|
||
sleep(1) # 等对话框消失
|
||
elif not support_auto_set_done and image.find(R.Produce.ButtonAutoSet):
|
||
device.click()
|
||
support_auto_set_done = True
|
||
sleep(1)
|
||
# 3. 选择回忆 自动编成 [screenshots/produce/screenshot_produce_start_3_memory.png]
|
||
elif lp.state == 3:
|
||
if image.find(R.Produce.TextStepIndicator4):
|
||
break
|
||
|
||
# 确认自动编成提示
|
||
if image.find(R.Produce.TextAutoSet):
|
||
dialog.yes(msg='Confirmed auto set.')
|
||
continue
|
||
# 今天仍然有租用回忆次数提示
|
||
elif image.find(R.Produce.TextRentAvailable):
|
||
dialog.yes(msg='Confirmed rent available. (Step 3)')
|
||
continue
|
||
|
||
if not memory_set_selected:
|
||
# 自动编成
|
||
if memory_set_index is None:
|
||
lp.click_if(R.Produce.ButtonAutoSet)
|
||
# 指定编号
|
||
else:
|
||
# dialog.no() # TODO: 这是什么？
|
||
select_set(memory_set_index)
|
||
memory_set_selected = True
|
||
# 下一步「次へ」
|
||
if find_button(R.Common.ButtonNextNoIcon, True) and next_throttler.request():
|
||
device.click()
|
||
continue
|
||
else:
|
||
assert False, f'Invalid state of {lp.state}.'
|
||
|
||
# 4. 选择道具 [screenshots/produce/screenshot_produce_start_4_end.png]
|
||
# TODO: 如果道具不足，这里加入推送提醒
|
||
if produce_solution().data.use_note_boost:
|
||
if image.find(R.Produce.CheckboxIconNoteBoost):
|
||
device.click()
|
||
sleep(0.1)
|
||
if produce_solution().data.use_pt_boost:
|
||
if image.find(R.Produce.CheckboxIconSupportPtBoost):
|
||
device.click()
|
||
sleep(0.1)
|
||
device.click(image.expect_wait(R.Produce.ButtonProduceStart))
|
||
# 5. 相关设置弹窗 [screenshots/produce/skip_commu.png]
|
||
cd = Countdown(5).start()
|
||
while not cd.expired():
|
||
device.screenshot()
|
||
if image.find(R.Produce.RadioTextSkipCommu):
|
||
device.click()
|
||
if image.find(R.Common.ButtonConfirmNoIcon):
|
||
device.click()
|
||
match mode:
|
||
case 'regular':
|
||
hajime_regular()
|
||
case 'pro':
|
||
hajime_pro()
|
||
case 'master':
|
||
hajime_master()
|
||
case _:
|
||
assert_never(mode)
|
||
return True
|
||
|
||
@task('培育')
|
||
def produce():
|
||
"""
|
||
培育任务
|
||
"""
|
||
if not produce_solution().data.enabled:
|
||
logger.info('Produce is disabled.')
|
||
return
|
||
import time
|
||
count = produce_solution().data.produce_count
|
||
idols = produce_solution().data.idols
|
||
memory_sets = produce_solution().data.memory_sets
|
||
mode = produce_solution().data.mode
|
||
# 数据验证
|
||
if count < 0:
|
||
user.warning('配置有误', '培育次数不能小于 0。将跳过本次培育。')
|
||
return
|
||
|
||
idol_iterator = cycle(idols)
|
||
memory_set_iterator = cycle(memory_sets)
|
||
for i in range(count):
|
||
start_time = time.time()
|
||
idol = next(idol_iterator)
|
||
if produce_solution().data.auto_set_memory:
|
||
memory_set = None
|
||
else:
|
||
memory_set = next(memory_set_iterator, None)
|
||
logger.info(
|
||
f'Produce start with: '
|
||
f'idol: {idol}, mode: {mode}, memory_set: #{memory_set}'
|
||
)
|
||
if not do_produce(idol, mode, memory_set):
|
||
user.info('AP 不足', f'由于 AP 不足，跳过了 {count - i} 次培育。')
|
||
logger.info('%d produce(s) skipped because of insufficient AP.', count - i)
|
||
break
|
||
end_time = time.time()
|
||
logger.info(f"Produce time used: {format_time(end_time - start_time)}")
|
||
|
||
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)
|
||
from kotonebot.backend.context import init_context
|
||
from kotonebot.kaa.common import BaseConfig
|
||
from kotonebot.kaa.main import Kaa
|
||
|
||
produce_solution().data.enabled = True
|
||
produce_solution().data.mode = 'pro'
|
||
produce_solution().data.produce_count = 1
|
||
# produce_solution().data.idols = ['i_card-skin-hski-3-002']
|
||
produce_solution().data.memory_sets = [1]
|
||
produce_solution().data.auto_set_memory = False
|
||
# do_produce(PIdol.月村手毬_初声, 'pro', 5)
|
||
produce()
|
||
# a()
|
||
# select_idol()
|
||
# select_set(10)
|
||
# manual_context().begin()
|
||
# print(ocr.ocr(rect=R.Produce.BoxSetCountIndicator).squash().numbers()) |