145 lines
3.6 KiB
Python
145 lines
3.6 KiB
Python
from pathlib import Path
|
|
from datetime import datetime,date
|
|
import aiofiles
|
|
import base64
|
|
import os
|
|
|
|
from astrbot import logger
|
|
|
|
async def load_template(template_name: str) -> str:
|
|
"""
|
|
异步加载模板内容（非阻塞）
|
|
"""
|
|
plugin_dir = Path(__file__).parent.parent
|
|
template_path = plugin_dir / "templates" / template_name
|
|
|
|
if not template_path.exists():
|
|
raise FileNotFoundError(f"模板文件不存在: {template_path}")
|
|
|
|
async with aiofiles.open(template_path, "r", encoding="utf-8") as f:
|
|
return await f.read()
|
|
|
|
|
|
def gold_to_string(gold_amount):
|
|
"""
|
|
将金钱数值转换为字符串表示形式
|
|
|
|
Args:
|
|
gold_amount (int): 金钱数值，单位为铜币
|
|
|
|
Returns:
|
|
str: 格式化后的金钱字符串，例如 "1金2银3铜"
|
|
"""
|
|
if not gold_amount:
|
|
return "无价格"
|
|
|
|
parts = []
|
|
started = False # 标记是否已经遇到第一个非零位
|
|
|
|
bricks = gold_amount // 100000000
|
|
gold = (gold_amount % 100000000) // 10000
|
|
silver = (gold_amount % 10000) // 100
|
|
copper = gold_amount % 100
|
|
|
|
for value, unit in [(bricks, "砖"), (gold, "金"), (silver, "银"), (copper, "铜")]:
|
|
if value != 0:
|
|
started = True
|
|
if started:
|
|
parts.append(f"{value}{unit}")
|
|
|
|
return "".join(parts)
|
|
|
|
|
|
def gold_to_parts(gold_amount):
|
|
"""将铜钱数拆成模板可渲染的货币片段"""
|
|
try:
|
|
amount = int(gold_amount or 0)
|
|
except (TypeError, ValueError):
|
|
amount = 0
|
|
|
|
if amount <= 0:
|
|
return []
|
|
|
|
bricks = amount // 100000000
|
|
gold = (amount % 100000000) // 10000
|
|
silver = (amount % 10000) // 100
|
|
copper = amount % 100
|
|
|
|
parts = []
|
|
started = False
|
|
for key, name, value in [
|
|
("zhuang", "砖", bricks),
|
|
("jin", "金", gold),
|
|
("yin", "银", silver),
|
|
("tong", "铜", copper),
|
|
]:
|
|
if value != 0:
|
|
started = True
|
|
if started:
|
|
parts.append({"key": key, "name": name, "value": value})
|
|
|
|
return parts
|
|
|
|
|
|
def week_to_num(week :str):
|
|
week_map = {
|
|
"一": 0,
|
|
"二": 1,
|
|
"三": 2,
|
|
"四": 3,
|
|
"五": 4,
|
|
"六": 5,
|
|
"日": 6,
|
|
}
|
|
return week_map.get(week,None)
|
|
|
|
|
|
def compare_date_str(date_str: str) -> str:
|
|
"""
|
|
date_str 格式：YYYY-MM-DD
|
|
"""
|
|
d = datetime.strptime(date_str, "%Y-%m-%d").date()
|
|
today = date.today()
|
|
|
|
if d < today:
|
|
return "过去"
|
|
elif d == today:
|
|
return "今天"
|
|
else:
|
|
return "将来"
|
|
|
|
|
|
def load_as_base64(icons_dir: str) -> dict[str, str]:
|
|
"""
|
|
将指定目录下的图标文件转换为 base64 字符串。
|
|
返回 {"文件名（不含扩展名）": "data:image/xxx;base64,xxx"} 的字典。
|
|
"""
|
|
supported_ext = {".png", ".jpg", ".jpeg", ".gif", ".webp", ".svg", ".ico"}
|
|
icons = {}
|
|
|
|
for filename in os.listdir(icons_dir):
|
|
name, ext = os.path.splitext(filename)
|
|
if ext.lower() not in supported_ext:
|
|
continue
|
|
|
|
filepath = os.path.join(icons_dir, filename)
|
|
with open(filepath, "rb") as f:
|
|
data = base64.b64encode(f.read()).decode()
|
|
|
|
display_name = name
|
|
try:
|
|
display_name = name.encode("latin-1").decode("gbk")
|
|
except (UnicodeEncodeError, UnicodeDecodeError):
|
|
pass
|
|
|
|
if ext.lower() == ".svg":
|
|
mime = "image/svg+xml"
|
|
elif ext.lower() == ".jpg":
|
|
mime = "image/jpeg"
|
|
else:
|
|
mime = f"image/{ext.lower().lstrip('.')}"
|
|
|
|
icons[display_name] = f"data:{mime};base64,{data}"
|
|
|
|
return icons
|