feat✨: CSGO开箱

zhenxun/plugins/open_cases/build_image.py

@@ -0,0 +1,155 @@
+from datetime import timedelta, timezone
+
+from zhenxun.configs.path_config import IMAGE_PATH
+from zhenxun.services.log import logger
+from zhenxun.utils.image_utils import BuildImage
+from zhenxun.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:
+    """构造抽取图片
+
+    参数:
+        skin (BuffSkin): BuffSkin
+        rand (str): 磨损
+
+    返回:
+        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.paste(skin_image, (10, 30))
+    await skin_bk.line((220, 10, 220, 180))
+    await skin_bk.text((10, 10), skin.name, (255, 255, 255))
+    name_icon = BuildImage(20, 20, background=ICON_PATH / "name_white.png")
+    await skin_bk.paste(name_icon, (240, 13))
+    await skin_bk.text((265, 15), f"名称:", (255, 255, 255), font_size=20)
+    await skin_bk.text(
+        (310, 15),
+        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.paste(tone_icon, (240, 45))
+    await skin_bk.text((265, 45), "品质:", (255, 255, 255), font_size=20)
+    await skin_bk.text((310, 45), COLOR2NAME[skin.color][:2], COLOR2COLOR[skin.color])
+    type_icon = BuildImage(20, 20, background=ICON_PATH / "type_white.png")
+    await skin_bk.paste(type_icon, (240, 73))
+    await skin_bk.text((265, 75), "类型:", (255, 255, 255), font_size=20)
+    await skin_bk.text((310, 75), skin.weapon_type, (255, 255, 255))
+    price_icon = BuildImage(20, 20, background=ICON_PATH / "price_white.png")
+    await skin_bk.paste(price_icon, (240, 103))
+    await skin_bk.text((265, 105), "价格:", (255, 255, 255), font_size=20)
+    await skin_bk.text((310, 105), str(skin.sell_min_price), (0, 255, 98))
+    abrasion_icon = BuildImage(20, 20, background=ICON_PATH / "abrasion_white.png")
+    await skin_bk.paste(abrasion_icon, (240, 133))
+    await skin_bk.text((265, 135), "磨损:", (255, 255, 255), font_size=20)
+    await skin_bk.text((310, 135), skin.abrasion, (255, 255, 255))
+    await skin_bk.text((228, 165), f"({rand})", (255, 255, 255))
+    return skin_bk
+
+
+async def generate_skin(skin: BuffSkin, update_count: int) -> BuildImage | None:
+    """构造皮肤图片
+
+    参数:
+        skin (BuffSkin): BuffSkin
+
+    返回:
+        BuildImage | None: 图片
+    """
+    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.paste(skin_img, (10, 10))
+        await case_bk.line((250, 10, 250, 190))
+        await case_bk.line((280, 160, 660, 160))
+        name_icon = BuildImage(30, 30, background=ICON_PATH / "box_white.png")
+        await case_bk.paste(name_icon, (260, 25))
+        await case_bk.text((295, 30), "名称:", (255, 255, 255))
+        await case_bk.text((345, 30), skin.case_name, (255, 0, 38), font_size=30)
+
+        type_icon = BuildImage(30, 30, background=ICON_PATH / "type_white.png")
+        await case_bk.paste(type_icon, (260, 70))
+        await case_bk.text((295, 75), "类型:", (255, 255, 255))
+        await case_bk.text((345, 75), "武器箱", (0, 157, 255), font_size=30)
+
+        price_icon = BuildImage(30, 30, background=ICON_PATH / "price_white.png")
+        await case_bk.paste(price_icon, (260, 114))
+        await case_bk.text((295, 120), "单价:", (255, 255, 255))
+        await case_bk.text(
+            (340, 120), 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.paste(update_count_icon, (575, 10))
+        await case_bk.text((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.paste(num_icon, (455, 70))
+        await case_bk.text((490, 75), "在售:", (255, 255, 255))
+        await case_bk.text((535, 75), 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.paste(want_buy_icon, (455, 114))
+        await case_bk.text((490, 120), "求购:", (255, 255, 255))
+        await case_bk.text((535, 120), str(skin.buy_num), (144, 0, 255), font_size=30)
+
+        await case_bk.text((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.text(
+            (350, 165),
+            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.paste(skin_image, (10, 30))
+        update_count_icon = BuildImage(
+            35, 35, background=ICON_PATH / "reload_white.png"
+        )
+        await skin_bk.line((10, 180, 220, 180))
+        await skin_bk.text((10, 10), skin.name, (255, 255, 255))
+        await skin_bk.paste(update_count_icon, (140, 10))
+        await skin_bk.text((175, 15), str(update_count), (255, 255, 255))
+        await skin_bk.text((10, 185), f"{skin.skin_name}", (255, 255, 255), "width")
+        await skin_bk.text((10, 218), "品质:", (255, 255, 255))
+        await skin_bk.text(
+            (55, 218), COLOR2NAME[skin.color][:2], COLOR2COLOR[skin.color]
+        )
+        await skin_bk.text((100, 218), "类型:", (255, 255, 255))
+        await skin_bk.text((145, 218), skin.weapon_type, (255, 255, 255))
+        return skin_bk