mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
79 lines
4.4 KiB
Python
79 lines
4.4 KiB
Python
import os
|
|
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
|
|
from utils.alias_handler import get_id_by_name
|
|
from utils.message_util import MessageBuild
|
|
from .abyss_rate_data import get_rate, get_formation_rate
|
|
|
|
res_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'res')
|
|
|
|
|
|
def get_font(size, font_type='msyh.ttc'):
|
|
return ImageFont.truetype(os.path.join(res_path, font_type), size)
|
|
|
|
|
|
async def draw_rate_rank(type: str = 'role', mode: str = 'used'):
|
|
data = await get_rate(type)
|
|
if not data or data['code'] != '200':
|
|
return '获取工坊数据失败，请稍后再试'
|
|
col = int(len(data['result']['rateList']) / 5) if len(data['result']['rateList']) % 5 == 0 else int(len(data['result']['rateList']) / 5) + 1
|
|
top_img = Image.open(os.path.join(res_path, 'player_card', '卡片顶部无字.png')).resize((1080, 316))
|
|
body_img = Image.open(os.path.join(res_path, 'player_card', '卡片身体.png')).resize((1080, 226))
|
|
bottom_img = Image.open(os.path.join(res_path, 'player_card', '卡片底部.png')).resize((1080, 72))
|
|
bg_img = Image.new('RGBA', (1080, top_img.height + col * 220 + bottom_img.height + 50))
|
|
bg_img.paste(top_img, (0, 0))
|
|
bg_draw = ImageDraw.Draw(bg_img)
|
|
bg_draw.text((70, 55), '2.6深境螺旋登场率排行榜', fill='black', font=get_font(40, 'msyhbd.ttc'))
|
|
bg_draw.text((70, 110), f'当前共{data["result"]["userCount"]}份样本数据', fill='black', font=get_font(35))
|
|
for i in range(0, col + 1):
|
|
bg_img.paste(body_img, (0, top_img.height + i * body_img.height))
|
|
bg_img.paste(bottom_img, (0, top_img.height + col * 220 + 50))
|
|
bg_draw.text((130, bg_img.height - 86), 'Created by LittlePaimon | Data from 原神创意工坊', fill='black', font=get_font(35))
|
|
n = 0
|
|
for role in data['result']['rateList']:
|
|
role_img = Image.open(os.path.join(res_path, 'role_card', f'{get_id_by_name(role["name"])}.png')).resize((160, 200))
|
|
role_draw = ImageDraw.Draw(role_img)
|
|
role_draw.text((28 if len(role['rate']) == 6 else 38, 158), role['rate'], font=get_font(30), fill='black')
|
|
bg_img.alpha_composite(role_img, (50 + 204 * (n % 5), 180 + 240 * int(n / 5)))
|
|
n += 1
|
|
|
|
return MessageBuild.Image(bg_img, quality=75)
|
|
|
|
|
|
async def draw_teams_rate(floor='上半半'):
|
|
data = await get_formation_rate(1 if floor == '上半' else 2)
|
|
if not data or data['code'] != '200':
|
|
return '获取工坊数据失败，请稍后再试'
|
|
rateList = data['result']['rateList'][0:10]
|
|
top_img = Image.open(os.path.join(res_path, 'player_card', '卡片顶部无字.png')).resize((1080, 316))
|
|
body_img = Image.open(os.path.join(res_path, 'player_card', '卡片身体.png')).resize((1080, 240))
|
|
bottom_img = Image.open(os.path.join(res_path, 'player_card', '卡片底部.png')).resize((1080, 72))
|
|
bg_img = Image.new('RGBA', (1080, top_img.height + 10 * body_img.height + bottom_img.height - 130))
|
|
bg_img.paste(top_img, (0, 0))
|
|
bg_draw = ImageDraw.Draw(bg_img)
|
|
bg_draw.text((70, 55), f'2.6深境螺旋阵容出场率 | {floor}', fill='black', font=get_font(40, 'msyhbd.ttc'))
|
|
bg_draw.text((70, 110), f'当前共{data["result"]["userCount"]}份样本数据', fill='black', font=get_font(35))
|
|
for i in range(0, 10):
|
|
bg_img.paste(body_img, (0, top_img.height + i * body_img.height))
|
|
bg_img.paste(bottom_img, (0, top_img.height + 10 * 240 - 130))
|
|
bg_draw.text((130, bg_img.height - 86), 'Created by LittlePaimon | Data from 原神创意工坊', fill='black', font=get_font(35))
|
|
n = 0
|
|
for team in rateList:
|
|
bg_draw.text((48, 240 + 240 * n), '排名', fill='grey', font=get_font(30))
|
|
bg_draw.text((65 if n != 9 else 55, 275 + 240 * n), str(n + 1), fill='black', font=get_font(35))
|
|
bg_draw.text((934, 240 + 240 * n), '出场率', fill='grey', font=get_font(30))
|
|
bg_draw.text((930, 275 + 240 * n), team['rate'], fill='black', font=get_font(35))
|
|
r = 0
|
|
for role in team['formation']:
|
|
role_img = Image.open(
|
|
os.path.join(res_path, 'role_card', f'{get_id_by_alias(role["name"])[0]}.png')).resize((160, 200))
|
|
role_draw = ImageDraw.Draw(role_img)
|
|
role_draw.text((80 - 15 * len(role['name']), 158), role['name'], font=get_font(30), fill='black')
|
|
bg_img.alpha_composite(role_img, (130 + 204 * r, 180 + 240 * n))
|
|
r += 1
|
|
n += 1
|
|
|
|
return MessageBuild.Image(bg_img, quality=75)
|
|
|