mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
import time
|
|
|
|
from nonebot_plugin_htmlrender import template_to_pic
|
|
from pydantic import BaseModel
|
|
from tortoise.expressions import Q
|
|
|
|
from zhenxun.configs.config import BotConfig
|
|
from zhenxun.configs.path_config import TEMPLATE_PATH
|
|
from zhenxun.models.goods_info import GoodsInfo
|
|
from zhenxun.utils._build_image import BuildImage
|
|
|
|
from .config import ICON_PATH
|
|
|
|
|
|
class GoodsItem(BaseModel):
|
|
goods_list: list[dict]
|
|
"""商品列表"""
|
|
partition: str
|
|
"""分区名称"""
|
|
|
|
|
|
async def html_image() -> bytes:
|
|
"""构建图片"""
|
|
goods_list = (
|
|
await GoodsInfo.filter(
|
|
Q(goods_limit_time__gte=time.time()) | Q(goods_limit_time=0),
|
|
goods_limit_time=0,
|
|
)
|
|
.annotate()
|
|
.order_by("id")
|
|
.all()
|
|
)
|
|
partition_dict: dict[str, list[dict]] = {}
|
|
for idx, goods in enumerate(goods_list):
|
|
if not goods.partition:
|
|
goods.partition = "默认分区"
|
|
if goods.partition not in partition_dict:
|
|
partition_dict[goods.partition] = []
|
|
icon = None
|
|
if goods.icon:
|
|
path = ICON_PATH / goods.icon
|
|
if path.exists():
|
|
icon = (
|
|
"data:image/png;base64,"
|
|
f"{BuildImage.open(ICON_PATH / goods.icon).pic2bs4()[9:]}"
|
|
)
|
|
partition_dict[goods.partition].append(
|
|
{
|
|
"id": idx + 1,
|
|
"price": goods.goods_price,
|
|
"daily_limit": goods.daily_limit or "∞",
|
|
"name": goods.goods_name,
|
|
"icon": icon,
|
|
"description": goods.goods_description,
|
|
}
|
|
)
|
|
data_list = [
|
|
GoodsItem(goods_list=value, partition=partition)
|
|
for partition, value in partition_dict.items()
|
|
]
|
|
return await template_to_pic(
|
|
template_path=str((TEMPLATE_PATH / "shop").absolute()),
|
|
template_name="main.html",
|
|
templates={"name": BotConfig.self_nickname, "data_list": data_list},
|
|
pages={
|
|
"viewport": {"width": 850, "height": 1024},
|
|
"base_url": f"file://{TEMPLATE_PATH}",
|
|
},
|
|
wait=2,
|
|
)
|