zhenxun_bot/plugins/open_cases/build_image.py

from datetime import timedelta, timezone
from typing import Optional

from configs.path_config import IMAGE_PATH
from services.log import logger
from utils.image_utils import BuildImage
from utils.utils import cn2py

from .config import COLOR2COLOR, COLOR2NAME
from .models.buff_skin import BuffSkin

BASE_PATH = IMAGE_PATH / "csgo_cases"

ICON_PATH = IMAGE_PATH / "_icon"


async def draw_card(skin: BuffSkin, rand: str) -> BuildImage:
    """构造抽取图片

    Args:
        skin (BuffSkin): BuffSkin
        rand (str): 磨损

    Returns:
        BuildImage: BuildImage
    """
    name = skin.name + "-" + skin.skin_name + "-" + skin.abrasion
    file_path = BASE_PATH / cn2py(skin.case_name.split(",")[0]) / f"{cn2py(name)}.jpg"
    if not file_path.exists():
        logger.warning(f"皮肤图片: {name} 不存在", "开箱")
    skin_bk = BuildImage(
        460, 200, color=(25, 25, 25, 100), font_size=25, font="CJGaoDeGuo.otf"
    )
    if file_path.exists():
        skin_image = BuildImage(205, 153, background=file_path)
        await skin_bk.apaste(skin_image, (10, 30), alpha=True)
    await skin_bk.aline((220, 10, 220, 180))
    await skin_bk.atext((10, 10), skin.name, (255, 255, 255))
    name_icon = BuildImage(20, 20, background=ICON_PATH / "name_white.png")
    await skin_bk.apaste(name_icon, (240, 13), True)
    await skin_bk.atext((265, 15), f"名称:", (255, 255, 255), font_size=20)
    await skin_bk.atext(
        (300, 9),
        f"{skin.skin_name + ('(St)' if skin.is_stattrak else '')}",
        (255, 255, 255),
    )
    tone_icon = BuildImage(20, 20, background=ICON_PATH / "tone_white.png")
    await skin_bk.apaste(tone_icon, (240, 45), True)
    await skin_bk.atext((265, 45), "品质:", (255, 255, 255), font_size=20)
    await skin_bk.atext((300, 40), COLOR2NAME[skin.color][:2], COLOR2COLOR[skin.color])
    type_icon = BuildImage(20, 20, background=ICON_PATH / "type_white.png")
    await skin_bk.apaste(type_icon, (240, 73), True)
    await skin_bk.atext((265, 75), "类型:", (255, 255, 255), font_size=20)
    await skin_bk.atext((300, 70), skin.weapon_type, (255, 255, 255))
    price_icon = BuildImage(20, 20, background=ICON_PATH / "price_white.png")
    await skin_bk.apaste(price_icon, (240, 103), True)
    await skin_bk.atext((265, 105), "价格:", (255, 255, 255), font_size=20)
    await skin_bk.atext((300, 102), str(skin.sell_min_price), (0, 255, 98))
    abrasion_icon = BuildImage(20, 20, background=ICON_PATH / "abrasion_white.png")
    await skin_bk.apaste(abrasion_icon, (240, 133), True)
    await skin_bk.atext((265, 135), "磨损:", (255, 255, 255), font_size=20)
    await skin_bk.atext((300, 130), skin.abrasion, (255, 255, 255))
    await skin_bk.atext((228, 165), f"({rand})", (255, 255, 255))
    return skin_bk


async def generate_skin(skin: BuffSkin, update_count: int) -> Optional[BuildImage]:
    """构造皮肤图片

    Args:
        skin (BuffSkin): BuffSkin

    Returns:
        Optional[BuildImage]: 图片
    """
    name = skin.name + "-" + skin.skin_name + "-" + skin.abrasion
    file_path = BASE_PATH / cn2py(skin.case_name.split(",")[0]) / f"{cn2py(name)}.jpg"
    if not file_path.exists():
        logger.warning(f"皮肤图片: {name} 不存在", "查看武器箱")
    if skin.color == "CASE":
        case_bk = BuildImage(
            700, 200, color=(25, 25, 25, 100), font_size=25, font="CJGaoDeGuo.otf"
        )
        if file_path.exists():
            skin_img = BuildImage(200, 200, background=file_path)
            await case_bk.apaste(skin_img, (10, 10), True)
        await case_bk.aline((250, 10, 250, 190))
        await case_bk.aline((280, 160, 660, 160))
        name_icon = BuildImage(30, 30, background=ICON_PATH / "box_white.png")
        await case_bk.apaste(name_icon, (260, 25), True)
        await case_bk.atext((295, 30), "名称:", (255, 255, 255))
        await case_bk.atext((345, 25), skin.case_name, (255, 0, 38), font_size=30)

        type_icon = BuildImage(30, 30, background=ICON_PATH / "type_white.png")
        await case_bk.apaste(type_icon, (260, 70), True)
        await case_bk.atext((295, 75), "类型:", (255, 255, 255))
        await case_bk.atext((345, 72), "武器箱", (0, 157, 255), font_size=30)

        price_icon = BuildImage(30, 30, background=ICON_PATH / "price_white.png")
        await case_bk.apaste(price_icon, (260, 114), True)
        await case_bk.atext((295, 120), "单价:", (255, 255, 255))
        await case_bk.atext(
            (340, 116), str(skin.sell_min_price), (0, 255, 98), font_size=30
        )

        update_count_icon = BuildImage(
            40, 40, background=ICON_PATH / "reload_white.png"
        )
        await case_bk.apaste(update_count_icon, (575, 10), True)
        await case_bk.atext((625, 12), str(update_count), (255, 255, 255), font_size=45)

        num_icon = BuildImage(30, 30, background=ICON_PATH / "num_white.png")
        await case_bk.apaste(num_icon, (455, 70), True)
        await case_bk.atext((490, 75), "在售:", (255, 255, 255))
        await case_bk.atext((535, 72), str(skin.sell_num), (144, 0, 255), font_size=30)

        want_buy_icon = BuildImage(30, 30, background=ICON_PATH / "want_buy_white.png")
        await case_bk.apaste(want_buy_icon, (455, 114), True)
        await case_bk.atext((490, 120), "求购:", (255, 255, 255))
        await case_bk.atext((535, 116), str(skin.buy_num), (144, 0, 255), font_size=30)

        await case_bk.atext((275, 165), "更新时间", (255, 255, 255), font_size=22)
        date = str(
            skin.update_time.replace(microsecond=0).astimezone(
                timezone(timedelta(hours=8))
            )
        ).split("+")[0]
        await case_bk.atext(
            (344, 170),
            date,
            (255, 255, 255),
            font_size=30,
        )
        return case_bk
    else:
        skin_bk = BuildImage(
            235, 250, color=(25, 25, 25, 100), font_size=25, font="CJGaoDeGuo.otf"
        )
        if file_path.exists():
            skin_image = BuildImage(205, 153, background=file_path)
            await skin_bk.apaste(skin_image, (10, 30), alpha=True)
        update_count_icon = BuildImage(
            35, 35, background=ICON_PATH / "reload_white.png"
        )
        await skin_bk.aline((10, 180, 220, 180))
        await skin_bk.atext((10, 10), skin.name, (255, 255, 255))
        await skin_bk.apaste(update_count_icon, (140, 10), True)
        await skin_bk.atext((175, 15), str(update_count), (255, 255, 255))
        await skin_bk.atext((10, 185), f"{skin.skin_name}", (255, 255, 255), "by_width")
        await skin_bk.atext((10, 218), "品质:", (255, 255, 255))
        await skin_bk.atext(
            (55, 218), COLOR2NAME[skin.color][:2], COLOR2COLOR[skin.color]
        )
        await skin_bk.atext((100, 218), "类型:", (255, 255, 255))
        await skin_bk.atext((145, 218), skin.weapon_type, (255, 255, 255))
        return skin_bk