run_plan.py

from Arknights.helper import logger
from Arknights.shell_next import _create_helper
import config
import json
import os


if __name__ == '__main__':
    if not os.path.exists('config/plan.json'):
        logger.error('未能检测到刷图计划文件.')
        exit(-1)
    with open('config/plan.json', 'r') as f:
        plan = json.load(f)
    helper, _ = _create_helper()
    if plan['stages']:
        update_flag = False
        has_remain_sanity = True
        for task in plan['stages']:
            remain = task.get('remain', task['count'])
            if remain == 0:
                continue
            logger.info('准备执行关卡 [%s], 计划次数: %s, 剩余次数: %s' % (task['stage'], task['count'], remain))
            update_flag = True
            try:
                c_id, remain = helper.module_battle(task['stage'], remain)
            except Exception as e:
                if isinstance(e, ValueError):
                    if config.get('plan/use_start_sp_stage', False):
                        logger.info('尝试进入活动关卡')
                        from addons.start_sp_stage import StartSpStageAddon
                        c_id, remain = StartSpStageAddon(helper).run(task['stage'], remain)
                    else:
                        logger.error(f"不支持普通跳转的关卡 [{task['stage']}], 启用特殊关卡支持需要安装 cnocr "
                                     f"并将配置中的 plan/use_start_sp_stage 设置为 true")
                else:
                    raise e

            task['remain'] = remain
            if remain > 0:
                logger.info('理智不足, 退出计划执行')
                has_remain_sanity = False
                break
        if update_flag:
            with open('config/plan.json', 'w') as f:
                json.dump(plan, f, indent=4, sort_keys=True)
            print('刷图计划已更新至: config/plan.json')
        if has_remain_sanity and config.get('plan/idle_stage', None) is not None:
            # todo recheck
            idle_stage = config.get('plan/idle_stage')
            logger.info('刷图计划已执行完毕, 理智还有剩余, 执行 idle stage [%s]' % idle_stage)
            helper.refill_with_item = config.get('plan/refill_ap_with_item', False)
            helper.refill_with_originium = config.get('plan/refill_ap_with_originium', False)
            helper.module_battle(idle_stage, 1000)
    else:
        logger.error('未能检测到刷图计划')