LittlePaimon/Paimon_Info/draw_role_card.py

from pathlib import Path

from PIL import Image, ImageDraw, ImageFont
from littlepaimon_utils import aiorequests
from littlepaimon_utils.files import load_image, load_json
from littlepaimon_utils.images import draw_right_text, draw_center_text

from .damage_cal.damage import get_role_dmg
from ..utils.enka_util import get_artifact_suit, artifact_total_value, get_expect_score, get_effective, check_effective
from ..utils.message_util import MessageBuild

res_path = Path() / 'resources' / 'LittlePaimon'
res_path2 = Path() / 'data' / 'LittlePaimon' / 'res'
weapon_url = 'https://upload-bbs.mihoyo.com/game_record/genshin/equip/{}.png'
artifact_url = 'https://upload-bbs.mihoyo.com/game_record/genshin/equip/{}.png'
talent_url = 'https://upload-bbs.mihoyo.com/game_record/genshin/constellation_icon/{}.png'
skill_url = 'https://static.cherishmoon.fun/LittlePaimon/skill/{}.png'

element_type = ['物理', '火元素', '雷元素', '水元素', '草元素', '风元素', '岩元素', '冰元素']

region = load_json(path=Path(__file__).parent.parent / 'utils' / 'json_data' / 'role_region.json')


def get_font(size, font='hywh.ttf'):
    return ImageFont.truetype(str(res_path / font), size)


async def draw_role_card(uid, data):
    bg_card = load_image(res_path / 'player_card2' / f'背景_{data["元素"]}.png', mode='RGBA')
    try:
        dmg_img = get_role_dmg(data)
    except Exception:
        dmg_img = None
    if dmg_img:
        bg = Image.new('RGBA', (1080, 1920 + dmg_img.size[1] + 20), (0, 0, 0, 0))
        bg_card_center = bg_card.crop((0, 730, 1080, 1377)).resize((1080, dmg_img.size[1] + 667))
        bg.alpha_composite(bg_card.crop((0, 0, 1080, 730)), (0, 0))
        bg.alpha_composite(bg_card_center, (0, 730))
        bg.alpha_composite(bg_card.crop((0, 1377, 1080, 1920)), (0, dmg_img.size[1] + 1397))
        bg.alpha_composite(dmg_img, (71, 1846))
    else:
        bg = Image.new('RGBA', (1080, 1920), (0, 0, 0, 0))
        bg.alpha_composite(bg_card, (0, 0))
    base_mask = load_image(res_path / 'player_card2' / '底遮罩.png')
    bg.alpha_composite(base_mask, (0, 0))
    if data['名称'] not in ['荧', '空', '埃洛伊']:
        region_icon = load_image(path=res_path / 'player_card2' / f'{region[data["名称"]]}.png', size=(130, 130))
        bg.alpha_composite(region_icon, (0, 4))
    bg_draw = ImageDraw.Draw(bg)
    bg_draw.text((131, 100), f"UID{uid}", fill='white', font=get_font(48, 'number.ttf'))
    bg_draw.text((134, 150), data['名称'], fill='white', font=get_font(72, '优设标题黑.ttf'))

    level_mask = load_image(res_path / 'player_card2' / '等级遮罩.png')
    bg.alpha_composite(level_mask, (298 + 60 * (len(data['名称']) - 2), 172))
    draw_center_text(bg_draw, f'LV{data["等级"]}', 298 + 60 * (len(data['名称']) - 2),
                     298 + 60 * (len(data['名称']) - 2) + 171, 174, 'black', get_font(48, 'number.ttf'))
    # 属性值
    prop = data['属性']
    bg_draw.text((89, 262), '生命值', fill='white', font=get_font(34, 'hywh.ttf'))
    text_length = bg_draw.textlength(f"+{prop['额外生命']}", font=get_font(34, 'number.ttf'))
    draw_right_text(bg_draw, f"{prop['基础生命']}", 480 - text_length - 5, 264, 'white', get_font(34, 'number.ttf'))
    draw_right_text(bg_draw, f"+{prop['额外生命']}", 480, 264, '#59c538', get_font(34, 'number.ttf'))

    bg_draw.text((89, 319), '攻击力', fill='white', font=get_font(34, 'hywh.ttf'))
    text_length = bg_draw.textlength(f"+{prop['额外攻击']}", font=get_font(34, 'number.ttf'))
    draw_right_text(bg_draw, f"{prop['基础攻击']}", 480 - text_length - 5, 321, 'white', get_font(34, 'number.ttf'))
    draw_right_text(bg_draw, f"+{prop['额外攻击']}", 480, 321, '#59c538', get_font(34, 'number.ttf'))

    bg_draw.text((89, 377), '防御力', fill='white', font=get_font(34, 'hywh.ttf'))
    text_length = bg_draw.textlength(f"+{prop['额外防御']}", font=get_font(34, 'number.ttf'))
    draw_right_text(bg_draw, f"{prop['基础防御']}", 480 - text_length - 5, 379, 'white', get_font(34, 'number.ttf'))
    draw_right_text(bg_draw, f"+{prop['额外防御']}", 480, 379, '#59c538', get_font(34, 'number.ttf'))

    text = round(prop['暴击率'] * 100, 1)
    bg_draw.text((89, 436), '暴击率', fill='white', font=get_font(34, 'hywh.ttf'))
    draw_right_text(bg_draw, f"{text}%", 480, 438, 'white', get_font(34, 'number.ttf'))

    text = round(prop['暴击伤害'] * 100, 1)
    bg_draw.text((89, 493), '暴击伤害', fill='white', font=get_font(34, 'hywh.ttf'))
    draw_right_text(bg_draw, f"{text}%", 480, 495, 'white', get_font(34, 'number.ttf'))

    bg_draw.text((89, 551), '元素精通', fill='white', font=get_font(34, 'hywh.ttf'))
    draw_right_text(bg_draw, str(prop['元素精通']), 480, 553, 'white', get_font(34, 'number.ttf'))

    text = round(prop['元素充能效率'] * 100, 1)
    bg_draw.text((89, 610), '元素充能效率', fill='white', font=get_font(34, 'hywh.ttf'))
    draw_right_text(bg_draw, f"{text}%", 480, 612, 'white', get_font(34, 'number.ttf'))

    max_element = max(prop['伤害加成'])
    text = round(max_element * 100, 1)

    bg_draw.text((89, 669), f'{element_type[prop["伤害加成"].index(max_element)]}伤害加成', fill='white',
                 font=get_font(34, 'hywh.ttf'))
    draw_right_text(bg_draw, f"{text}%", 480, 671, 'white', get_font(34, 'number.ttf'))

    # 天赋
    base_icon = load_image(res_path / 'player_card2' / f'图标_{data["元素"]}.png', mode='RGBA')
    base_icon_grey = load_image(res_path / 'player_card2' / '图标_灰.png', mode='RGBA')
    if data['名称'] in ['神里绫华', '莫娜']:
        data['天赋'].pop(2)
    for i in range(3):
        bg.alpha_composite(base_icon.resize((132, 142)), (564, 253 + 147 * i))
        draw_center_text(bg_draw, str(data['天赋'][i]['等级']), 510, 552, 310 + 147 * i, 'black',
                         get_font(34, 'number.ttf'))
        skill_icon = res_path2 / 'skill' / f'{data["天赋"][i]["图标"]}.png'
        skill_icon = await aiorequests.get_img(url=skill_url.format(data["天赋"][i]["图标"]), size=(57, 57),
                                               save_path=skill_icon, mode='RGBA')
        bg.alpha_composite(skill_icon, (603, 298 + 147 * i))

    # 命座
    lock = load_image(res_path / 'player_card2' / '锁.png', mode='RGBA', size=(45, 45))
    t = 0
    for talent in data['命座']:
        bg.alpha_composite(base_icon.resize((83, 90)), (510 + t * 84, 790))
        talent_icon = res_path2 / 'skill' / f'{talent["图标"]}.png'
        talent_icon = await aiorequests.get_img(url=talent_url.format(talent["图标"]), size=(45, 45),
                                                save_path=talent_icon, mode='RGBA')
        bg.alpha_composite(talent_icon, (529 + t * 84, 813))
        t += 1
    for t2 in range(t, 6):
        bg.alpha_composite(base_icon_grey.resize((83, 90)), (510 + t2 * 84, 790))
        bg.alpha_composite(lock, (530 + t2 * 84, 813))

    # 武器
    weapon_bg = load_image(res_path / 'other' / f'star{data["武器"]["星级"]}.png', size=(150, 150))
    bg.alpha_composite(weapon_bg, (91, 760))
    weapon_icon = res_path2 / 'weapon' / f'{data["武器"]["图标"]}.png'
    weapon_icon = await aiorequests.get_img(url=weapon_url.format(data["武器"]["图标"]), size=(150, 150),
                                            save_path=weapon_icon, mode='RGBA')
    bg.alpha_composite(weapon_icon, (91, 760))
    bg_draw.text((268, 758), data['武器']['名称'], fill='white', font=get_font(34, 'hywh.ttf'))
    star = load_image(res_path / 'player_card2' / 'star.png')
    for i in range(data['武器']['星级']):
        bg.alpha_composite(star, (267 + i * 30, 799))
    draw_center_text(bg_draw, f'LV{data["武器"]["等级"]}', 268, 268 + 98, 835, 'black', get_font(27, 'number.ttf'))
    bg_draw.text((266, 869), f'精炼{data["武器"]["精炼等级"]}阶', fill='white', font=get_font(34, 'hywh.ttf'))

    # 圣遗物
    effective = get_effective(data['名称'], data['武器']['名称'], data['圣遗物'], data['元素'])
    average = get_expect_score(effective)
    total_score = 0
    # 第一排
    for i in range(2):
        try:
            artifact = data['圣遗物'][i]
        except IndexError:
            break
        artifact_bg = load_image(res_path / 'other' / f'star{artifact["星级"]}.png', size=(100, 100))
        bg.alpha_composite(artifact_bg, (587 + 317 * i, 1002))
        reli_path = res_path2 / 'reli' / f'{artifact["图标"]}.png'
        reli_path = await aiorequests.get_img(url=artifact_url.format(artifact["图标"]), size=(100, 100),
                                              save_path=reli_path, mode='RGBA')
        bg.alpha_composite(reli_path, (587 + 317 * i, 1002))
        bg_draw.text((411 + 317 * i, 951), artifact['名称'], fill='white', font=get_font(40))
        value, score = artifact_total_value(data['属性'], artifact, effective)
        total_score += value
        rank = 'SSS' if score >= 140 else 'SS' if 120 <= score < 140 else 'S' if 100 <= score < 120 else 'A' if 75 <= score < 100 else 'B' if 50 <= score < 75 else 'C'
        bg_draw.text((412 + 317 * i, 998), f'{rank}-{value}', fill='#ffde6b', font=get_font(28, 'number.ttf'))
        bg.alpha_composite(level_mask.resize((98, 30)), (412 + 317 * i, 1032))
        draw_center_text(bg_draw, f"LV{artifact['等级']}", 412 + 317 * i, 412 + 317 * i + 98, 1033, 'black',
                         get_font(27, 'number.ttf'))
        bg_draw.text((411 + 317 * i, 1069), artifact['主属性']['属性名'], fill='white', font=get_font(25))
        if artifact['主属性']['属性名'] not in ['生命值', '攻击力', '元素精通']:
            bg_draw.text((408 + 317 * i, 1100), f"+{artifact['主属性']['属性值']}%", fill='white',
                         font=get_font(25, 'number.ttf'))
        else:
            bg_draw.text((408 + 317 * i, 1100), f"+{artifact['主属性']['属性值']}", fill='white',
                         font=get_font(48, 'number.ttf'))
        for j in range(len(artifact['词条'])):
            if '百分比' in artifact['词条'][j]['属性名']:
                text = artifact['词条'][j]['属性名'].replace('百分比', '')
            else:
                text = artifact['词条'][j]['属性名']
            bg_draw.text((411 + 317 * i, 1163 + 50 * j), text,
                         fill='white' if check_effective(artifact['词条'][j]['属性名'], effective) else '#afafaf',
                         font=get_font(25))
            if artifact['词条'][j]['属性名'] not in ['攻击力', '防御力', '生命值', '元素精通']:
                num = '+' + str(artifact['词条'][j]['属性值']) + '%'
            else:
                num = '+' + str(artifact['词条'][j]['属性值'])
            draw_right_text(bg_draw, num, 679 + 317 * i, 1163 + 50 * j,
                            fill='white' if check_effective(artifact['词条'][j]['属性名'], effective) else '#afafaf',
                            font=get_font(25, 'number.ttf'))
    # 第二排
    for i in range(3):
        try:
            artifact = data['圣遗物'][i + 2]
        except IndexError:
            break
        artifact_bg = load_image(res_path / 'other' / f'star{artifact["星级"]}.png', size=(100, 100))
        bg.alpha_composite(artifact_bg, (270 + 317 * i, 1439))
        reli_path = res_path2 / 'reli' / f'{artifact["图标"]}.png'
        reli_path = await aiorequests.get_img(url=artifact_url.format(artifact["图标"]), size=(100, 100),
                                              save_path=reli_path, mode='RGBA')
        bg.alpha_composite(reli_path, (270 + 317 * i, 1439))
        bg_draw.text((94 + 317 * i, 1388), artifact['名称'], fill='white', font=get_font(40))
        value, score = artifact_total_value(data['属性'], artifact, effective)
        total_score += value
        rank = 'SSS' if score >= 140 else 'SS' if 120 <= score < 140 else 'S' if 100 <= score < 120 else 'A' if 75 <= score < 100 else 'B' if 50 <= score < 75 else 'C'
        bg_draw.text((95 + 317 * i, 1435), f'{rank}-{value}', fill='#ffde6b', font=get_font(28, 'number.ttf'))
        bg.alpha_composite(level_mask.resize((98, 30)), (95 + 317 * i, 1469))
        draw_center_text(bg_draw, f"LV{artifact['等级']}", 95 + 317 * i, 95 + 317 * i + 98, 1470, 'black',
                         get_font(27, 'number.ttf'))
        bg_draw.text((94 + 317 * i, 1506), artifact['主属性']['属性名'], fill='white', font=get_font(25))
        if artifact['主属性']['属性名'] not in ['生命值', '攻击力', '元素精通']:
            bg_draw.text((91 + 317 * i, 1537), f"+{artifact['主属性']['属性值']}%", fill='white',
                         font=get_font(48, 'number.ttf'))
        else:
            bg_draw.text((91 + 317 * i, 1537), f"+{artifact['主属性']['属性值']}", fill='white',
                         font=get_font(48, 'number.ttf'))
        for j in range(len(artifact['词条'])):
            if '百分比' in artifact['词条'][j]['属性名']:
                text = artifact['词条'][j]['属性名'].replace('百分比', '')
            else:
                text = artifact['词条'][j]['属性名']
            bg_draw.text((94 + 317 * i, 1600 + 50 * j), text,
                         fill='white' if check_effective(artifact['词条'][j]['属性名'], effective) else '#afafaf',
                         font=get_font(25))
            if artifact['词条'][j]['属性名'] not in ['攻击力', '防御力', '生命值', '元素精通']:
                num = '+' + str(artifact['词条'][j]['属性值']) + '%'
            else:
                num = '+' + str(artifact['词条'][j]['属性值'])
            draw_right_text(bg_draw, num, 362 + 317 * i, 1600 + 50 * j,
                            fill='white' if check_effective(artifact['词条'][j]['属性名'], effective) else '#afafaf',
                            font=get_font(25, 'number.ttf'))

    # 圣遗物评分
    bg_draw.text((119, 1057), '总有效词条数', fill='#afafaf', font=get_font(36))
    score_pro = total_score / (average * 5) * 100
    total_rank = 'SSS' if score_pro >= 140 else 'SS' if 120 <= score_pro < 140 else 'S' if 100 <= score_pro < 120 else 'A' if 75 <= score_pro < 100 else 'B' if 50 <= score_pro < 75 else 'C'
    rank_icon = load_image(res_path / 'player_card2' / f'评分{total_rank[0]}.png', mode='RGBA')
    if len(total_rank) == 3:
        bg.alpha_composite(rank_icon, (95, 964))
        bg.alpha_composite(rank_icon, (145, 964))
        bg.alpha_composite(rank_icon, (195, 964))
        bg_draw.text((250, 974), str(round(total_score, 1)), fill='white', font=get_font(60, 'number.ttf'))
    elif len(total_rank) == 2:
        bg.alpha_composite(rank_icon, (125, 964))
        bg.alpha_composite(rank_icon, (175, 964))
        bg_draw.text((235, 974), str(round(total_score, 1)), fill='white', font=get_font(60, 'number.ttf'))
    else:
        bg.alpha_composite(rank_icon, (143, 964))
        bg_draw.text((217, 974), str(round(total_score, 1)), fill='white', font=get_font(60, 'number.ttf'))

    # 圣遗物套装
    suit = get_artifact_suit(data['圣遗物'])
    if not suit:
        bg_draw.text((184, 1168), '未激活套装', fill='white', font=get_font(36))
        bg_draw.text((184, 1292), '未激活套装', fill='white', font=get_font(36))
    elif len(suit) == 1:
        artifact_path = res_path2 / 'reli' / f'{suit[0][1]}.png'
        artifact_path = await aiorequests.get_img(url=artifact_url.format(suit[0][1]), size=(110, 110),
                                                  save_path=artifact_path, mode='RGBA')
        bg.alpha_composite(artifact_path, (76, 1130))
        bg_draw.text((184, 1168), f'{suit[0][0][:2]}二件套', fill='white', font=get_font(36))
        bg_draw.text((184, 1292), '未激活套装', fill='white', font=get_font(36))
    else:
        if suit[0][0] == suit[1][0]:
            artifact_path1 = res_path2 / 'reli' / f'{suit[0][1]}.png'
            artifact_path1 = artifact_path2 = await aiorequests.get_img(url=artifact_url.format(suit[0][1]),
                                                                        size=(110, 110),
                                                                        save_path=artifact_path1,
                                                                        mode='RGBA')
            bg_draw.text((184, 1168), f'{suit[0][0][:2]}四件套', fill='white', font=get_font(36))
            bg_draw.text((184, 1292), f'{suit[0][0][:2]}四件套', fill='white', font=get_font(36))
        else:
            artifact_path1 = res_path2 / 'reli' / f'{suit[0][1]}.png'
            artifact_path1 = await aiorequests.get_img(url=artifact_url.format(suit[0][1]), size=(110, 110),
                                                       save_path=artifact_path1,
                                                       mode='RGBA')
            artifact_path2 = res_path2 / 'reli' / f'{suit[1][1]}.png'
            artifact_path2 = await aiorequests.get_img(url=artifact_url.format(suit[1][1]), size=(110, 110),
                                                       save_path=artifact_path2,
                                                       mode='RGBA')
            bg_draw.text((184, 1168), f'{suit[0][0][:2]}两件套', fill='white', font=get_font(36))
            bg_draw.text((184, 1292), f'{suit[1][0][:2]}两件套', fill='white', font=get_font(36))
        bg.alpha_composite(artifact_path1, (76, 1130))
        bg.alpha_composite(artifact_path2, (76, 1255))

    # 立绘
    paint_path = res_path / 'player_card2' / '立绘' / f'{data["名称"]}.png'
    bg.alpha_composite(load_image(path=paint_path), (695, 234))

    draw_center_text(bg_draw, f'更新于{data["更新时间"].replace("2022-", "")[:-3]}', 0, 1080, bg.size[1] - 95, '#afafaf', get_font(33, '优设标题黑.ttf'))
    bg_draw.text((24, bg.size[1] - 50), 'Created by LittlePaimon | Powered by Enka.Network', fill='white',
                 font=get_font(36, '优设标题黑.ttf'))

    return MessageBuild.Image(bg, quality=75, mode='RGB')