mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-08 21:30:01 +08:00
fit nonebot.beta2
This commit is contained in:
@@ -1,13 +1,12 @@
|
||||
|
||||
from .update_game_info import update_info
|
||||
from nonebot.adapters.onebot.v11 import MessageSegment
|
||||
from .announcement import PrettyAnnouncement
|
||||
from .update_game_info import update_info
|
||||
from .util import init_star_rst, generate_img, max_card, BaseData, \
|
||||
set_list, get_star, format_card_information, init_up_char
|
||||
import random
|
||||
from .config import PRETTY_THREE_P, PRETTY_TWO_P, DRAW_PATH, PRETTY_ONE_P, PRETTY_FLAG
|
||||
from .config import DRAW_DATA_PATH, draw_config
|
||||
from dataclasses import dataclass
|
||||
from .init_card_pool import init_game_pool
|
||||
from nonebot.adapters.cqhttp import MessageSegment, Message
|
||||
|
||||
try:
|
||||
import ujson as json
|
||||
@@ -74,7 +73,8 @@ async def pretty_draw(count: int, pool_name):
|
||||
rst = init_star_rst(star_list, cnlist, three_list, three_olist, up_list)
|
||||
if count > 90:
|
||||
obj_list = set_list(obj_list)
|
||||
return pool_info + MessageSegment.image("base64://" + await generate_img(obj_list, 'pretty', star_list)) \
|
||||
return pool_info + MessageSegment.image(
|
||||
"base64://" + await generate_img(obj_list, 'pretty', star_list)) \
|
||||
+ '\n' + rst[:-1] + '\n' + max_card(obj_dict)
|
||||
|
||||
|
||||
@@ -93,10 +93,10 @@ async def update_pretty_info():
|
||||
|
||||
async def init_pretty_data():
|
||||
global ALL_CHAR, ALL_CARD
|
||||
if PRETTY_FLAG:
|
||||
with open(DRAW_PATH + 'pretty.json', 'r', encoding='utf8') as f:
|
||||
if draw_config.PRETTY_FLAG:
|
||||
with (DRAW_DATA_PATH / 'pretty.json').open('r', encoding='utf8') as f:
|
||||
pretty_char_dict = json.load(f)
|
||||
with open(DRAW_PATH + 'pretty_card.json', 'r', encoding='utf8') as f:
|
||||
with (DRAW_DATA_PATH / 'pretty_card.json').open('r', encoding='utf8') as f:
|
||||
pretty_card_dict = json.load(f)
|
||||
ALL_CHAR = init_game_pool('pretty', pretty_char_dict, PrettyChar)
|
||||
ALL_CARD = init_game_pool('pretty_card', pretty_card_dict, PrettyChar)
|
||||
@@ -106,10 +106,11 @@ async def init_pretty_data():
|
||||
# 抽取卡池
|
||||
def _get_pretty_card(pool_name: str, mode: int = 1):
|
||||
global ALL_CHAR, ALL_CARD, _CURRENT_CHAR_POOL_TITLE, _CURRENT_CARD_POOL_TITLE
|
||||
pretty_config = draw_config.pretty
|
||||
if mode == 1:
|
||||
star = get_star([3, 2, 1], [PRETTY_THREE_P, PRETTY_TWO_P, PRETTY_ONE_P])
|
||||
star = get_star([3, 2, 1], [pretty_config.PRETTY_THREE_P, pretty_config.PRETTY_TWO_P, pretty_config.PRETTY_ONE_P])
|
||||
else:
|
||||
star = get_star([3, 2], [PRETTY_THREE_P, PRETTY_TWO_P])
|
||||
star = get_star([3, 2], [pretty_config.PRETTY_THREE_P, pretty_config.PRETTY_TWO_P])
|
||||
if pool_name == 'card':
|
||||
title = _CURRENT_CARD_POOL_TITLE
|
||||
up_data = UP_CARD
|
||||
@@ -127,7 +128,6 @@ def _get_pretty_card(pool_name: str, mode: int = 1):
|
||||
acquire_operator = random.choice(all_up_star)
|
||||
if pool_name == 'char':
|
||||
acquire_operator = acquire_operator.split(']')[1]
|
||||
print(acquire_operator)
|
||||
acquire_operator = [x for x in data if x.name == acquire_operator][0]
|
||||
else:
|
||||
acquire_operator = random.choice([x for x in data if x.star == star and not x.limited])
|
||||
@@ -144,6 +144,5 @@ async def _pretty_init_up_char():
|
||||
|
||||
async def reload_pretty_pool():
|
||||
await _pretty_init_up_char()
|
||||
return Message(f'当前UP池子:{_CURRENT_CHAR_POOL_TITLE} & {_CURRENT_CARD_POOL_TITLE} {POOL_IMG}')
|
||||
|
||||
return f'当前UP池子:{_CURRENT_CHAR_POOL_TITLE} & {_CURRENT_CARD_POOL_TITLE} {POOL_IMG}'
|
||||
|
||||
|
||||
Reference in New Issue
Block a user