This commit is contained in:
hibiki
2021-05-26 20:08:13 +08:00
parent 7359c6fa98
commit c27a1c4fb8
162 changed files with 922 additions and 932 deletions
+32 -37
View File
@@ -1,14 +1,14 @@
import os
from util.init_result import image
import nonebot
import random
from .update_game_info import update_info
from .util import generate_img, init_star_rst, BaseData, set_list
from .config import GENSHIN_FIVE_P, GENSHIN_FOUR_P, GENSHIN_G_FIVE_P, GENSHIN_THREE_P, I72_ADD
from .util import generate_img, init_star_rst, BaseData, set_list, get_star
from .config import GENSHIN_FIVE_P, GENSHIN_FOUR_P, GENSHIN_G_FIVE_P, GENSHIN_G_FOUR_P, GENSHIN_THREE_P, I72_ADD, \
DRAW_PATH, GENSHIN_FLAG
from dataclasses import dataclass
from .init_card_pool import init_game_pool
from configs.path_config import DRAW_PATH
from util.init_result import image
try:
import ujson as json
except ModuleNotFoundError:
@@ -21,7 +21,7 @@ genshin_count = {}
genshin_pl_count = {}
ALL_CHAR = []
ALL_ARM = []
ALL_ARMS = []
@dataclass
@@ -32,28 +32,28 @@ class GenshinChar(BaseData):
async def genshin_draw(user_id: int, count: int):
# 0 1 2
cnlist = ['★★★★★', '★★★★', '★★★']
genshin_list, five_list, five_olist, five_dict, star_list = _format_card_information(count, user_id)
rst = init_star_rst(star_list, cnlist, five_list, five_olist)
print(five_list)
char_list, five_list, five_index_list, char_dict, star_list = _format_card_information(count, user_id)
rst = init_star_rst(star_list, cnlist, five_list, five_index_list)
temp = ''
if count > 90:
genshin_list = set_list(genshin_list)
return image(b64=await generate_img(genshin_list, 'genshin', star_list)) + '\n' + rst[:-1] + \
char_list = set_list(char_list)
return image(b64=await generate_img(char_list, 'genshin', star_list)) + '\n' + rst[:-1] + \
temp[:-1] + f'\n距离保底发还剩 {90 - genshin_count[user_id] if genshin_count.get(user_id) else "^"} 抽' \
+ "\n【五星:0.6%,四星:5.1%\n第72抽开始五星概率每抽加0.585%】"
async def update_genshin_info():
global ALL_CHAR, ALL_ARM
global ALL_CHAR, ALL_ARMS
url = 'https://wiki.biligame.com/ys/角色筛选'
data, code = await update_info(url, 'genshin')
if code == 200:
ALL_CHAR = init_game_pool('genshin', data, GenshinChar)
url = 'https://wiki.biligame.com/ys/武器图鉴'
data, code = await update_info(url, 'genshin_arm', ['头像', '名称', '类型', '稀有度.alt', '初始基础属性1',
'初始基础属性2', '攻击力(MAX)', '副属性(MAX)', '技能'])
data, code = await update_info(url, 'genshin_arms', ['头像', '名称', '类型', '稀有度.alt',
'获取途径', '初始基础属性1', '初始基础属性2',
'攻击力(MAX)', '副属性(MAX)', '技能'])
if code == 200:
ALL_ARM = init_game_pool('genshin', data, GenshinChar)
ALL_ARMS = init_game_pool('genshin', data, GenshinChar)
# asyncio.get_event_loop().run_until_complete(update_genshin_info())
@@ -61,39 +61,34 @@ async def update_genshin_info():
@driver.on_startup
async def init_data():
global ALL_CHAR, ALL_ARM
if not os.path.exists(DRAW_PATH + '/draw_card_config/genshin.json') or \
not os.path.exists(DRAW_PATH + '/draw_card_config/genshin_arm.json'):
await update_genshin_info()
else:
with open(DRAW_PATH + '/draw_card_config/genshin.json', 'r', encoding='utf8') as f:
genshin_dict = json.load(f)
with open(DRAW_PATH + '/draw_card_config/genshin_arm.json', 'r', encoding='utf8') as f:
genshin_arm_dict = json.load(f)
ALL_CHAR = init_game_pool('genshin', genshin_dict, GenshinChar)
ALL_ARM = init_game_pool('genshin', genshin_arm_dict, GenshinChar)
global ALL_CHAR, ALL_ARMS
if GENSHIN_FLAG:
if not os.path.exists(DRAW_PATH + 'genshin.json') or not os.path.exists(DRAW_PATH + 'genshin_arms.json'):
await update_genshin_info()
else:
with open(DRAW_PATH + 'genshin.json', 'r', encoding='utf8') as f:
genshin_dict = json.load(f)
with open(DRAW_PATH + 'genshin_arms.json', 'r', encoding='utf8') as f:
genshin_ARMS_dict = json.load(f)
ALL_CHAR = init_game_pool('genshin', genshin_dict, GenshinChar)
ALL_ARMS = init_game_pool('genshin', genshin_ARMS_dict, GenshinChar)
# 抽取卡池
def _get_genshin_card(mode: int = 1, add: float = 0.0):
global ALL_ARM, ALL_CHAR
global ALL_ARMS, ALL_CHAR
if mode == 1:
star = random.sample([5, 4, 3],
counts=[int(GENSHIN_FIVE_P * 1000) + int(add * 1000), int(GENSHIN_FOUR_P * 1000),
int(GENSHIN_THREE_P * 1000)],
k=1)[0]
star = get_star([5, 4, 3], [GENSHIN_FIVE_P + add, GENSHIN_FOUR_P, GENSHIN_THREE_P])
elif mode == 2:
star = random.sample([5, 4],
counts=[int(GENSHIN_G_FIVE_P * 1000) + int(add * 1000), int(GENSHIN_FOUR_P * 1000)],
k=1)[0]
star = get_star([5, 4], [GENSHIN_G_FIVE_P + add, GENSHIN_G_FOUR_P])
else:
star = 5
chars = [x for x in (ALL_ARM if random.random() < 0.5 or star == 3 else ALL_CHAR) if x.star == star]
chars = [x for x in (ALL_ARMS if random.random() < 0.5 or star == 3 else ALL_CHAR) if x.star == star]
return random.choice(chars), abs(star - 5)
def _format_card_information(_count: int, user_id):
genshin_list = []
char_list = []
star_list = [0, 0, 0]
five_index_list = []
five_list = []
@@ -137,11 +132,11 @@ def _format_card_information(_count: int, user_id):
five_dict[char.name] += 1
except KeyError:
five_dict[char.name] = 1
genshin_list.append(char)
char_list.append(char)
if _count <= 90:
genshin_count[user_id] = f_count
genshin_pl_count[user_id] = count
return genshin_list, five_list, five_index_list, five_dict, star_list
return char_list, five_list, five_index_list, five_dict, star_list
def reset_count(user_id: int):