zhenxun_bot/plugins/draw_card/fgo_handle.py

from nonebot.adapters.onebot.v11 import MessageSegment
import random
from .update_game_requests_info import update_requests_info
from .util import generate_img, init_star_rst, BaseData, set_list, get_star, max_card
from .config import DRAW_DATA_PATH, draw_config
from dataclasses import dataclass
from .init_card_pool import init_game_pool

try:
    import ujson as json
except ModuleNotFoundError:
    import json

ALL_CHAR = []
ALL_CARD = []


@dataclass
class FgoChar(BaseData):
    pass


async def fgo_draw(count: int):
    #            0      1      2
    cnlist = ['★★★★★', '★★★★', '★★★']
    obj_list, obj_dict, max_star_list, star_list, max_star_index_list = _format_card_information(count)
    rst = init_star_rst(star_list, cnlist, max_star_list, max_star_index_list)
    if count > 90:
        obj_list = set_list(obj_list)
    return MessageSegment.image("base64://" + await generate_img(obj_list, 'fgo', star_list)) \
           + '\n' + rst[:-1] + '\n' + max_card(obj_dict)


async def update_fgo_info():
    global ALL_CHAR, ALL_CARD
    data, code = await update_requests_info('fgo')
    if code == 200:
        ALL_CHAR = init_game_pool('fgo', data, FgoChar)
    data, code = await update_requests_info('fgo_card')
    if code == 200:
        ALL_CARD = init_game_pool('fgo_card', data, FgoChar)


async def init_fgo_data():
    global ALL_CHAR, ALL_CARD
    if draw_config.FGO_FLAG:
        with (DRAW_DATA_PATH / 'fgo.json').open('r', encoding='utf8') as f:
            fgo_dict = json.load(f)
        ALL_CHAR = init_game_pool('fgo', fgo_dict, FgoChar)
        with (DRAW_DATA_PATH / 'fgo_card.json').open('r', encoding='utf8') as f:
            fgo_dict = json.load(f)
        ALL_CARD = init_game_pool('fgo', fgo_dict, FgoChar)


# 抽取卡池
def _get_fgo_card(mode: int = 1):
    global ALL_CHAR, ALL_CARD
    fgo_config = draw_config.fgo
    if mode == 1:
        star = get_star([8, 7, 6, 5, 4, 3], [fgo_config.FGO_SERVANT_FIVE_P, fgo_config.FGO_SERVANT_FOUR_P, fgo_config.FGO_SERVANT_THREE_P,
                                             fgo_config.FGO_CARD_FIVE_P, fgo_config.FGO_CARD_FOUR_P, fgo_config.FGO_CARD_THREE_P])
    elif mode == 2:
        star = get_star([5, 4], [fgo_config.FGO_CARD_FIVE_P, fgo_config.FGO_CARD_FOUR_P])
    else:
        star = get_star([8, 7, 6], [fgo_config.FGO_SERVANT_FIVE_P, fgo_config.FGO_SERVANT_FOUR_P, fgo_config.FGO_SERVANT_THREE_P])
    if star > 5:
        type_ = 'servant'
        star -= 3
        chars = [x for x in ALL_CHAR if x.star == star if not x.limited]
    else:
        type_ = 'card'
        chars = [x for x in ALL_CARD if x.star == star if not x.limited]
    return random.choice(chars), 5 - star, type_


# 整理数据
def _format_card_information(count: int):
    max_star_lst = []  # 获取的最高星级角色列表
    max_index_lst = []  # 获取最高星级角色的次数
    star_list = [0, 0, 0]
    obj_list = []  # 获取所有角色
    obj_dict = {}  # 获取角色次数字典
    servant_count = 0  # 保底计算
    card_count = 0  # 保底计算
    for i in range(count):
        servant_count += 1
        card_count += 1
        # 四星卡片保底
        if card_count == 9:
            obj, code, type_ = _get_fgo_card(2)
        # 三星从者保底
        elif servant_count == 10:
            obj, code, type_ = _get_fgo_card(3)
            _count = 0
        # 普通抽
        else:
            obj, code, type_ = _get_fgo_card()
        star_list[code] += 1
        if type_ == 'card' and code < 2:
            card_count = 0
        if type_ == 'servant':
            servant_count = 0
        if code == 0:
            max_star_lst.append(obj.name)
            max_index_lst.append(i)
        try:
            obj_dict[obj.name] += 1
        except KeyError:
            obj_dict[obj.name] = 1
        obj_list.append(obj)
    return obj_list, obj_dict, max_star_lst, star_list, max_index_lst