mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
127 lines
4.5 KiB
Python
Executable File
127 lines
4.5 KiB
Python
Executable File
from ..models.goods_info import GoodsInfo
|
||
from utils.image_utils import BuildImage
|
||
from utils.utils import is_number
|
||
from configs.path_config import IMAGE_PATH
|
||
from configs.config import Config
|
||
from nonebot import Driver
|
||
import nonebot
|
||
import time
|
||
import os
|
||
|
||
|
||
driver: Driver = nonebot.get_driver()
|
||
|
||
|
||
# 导入内置商品 | 重新生成商店图片
|
||
@driver.on_startup
|
||
async def init_default_shop_goods():
|
||
if os.path.exists(f"{IMAGE_PATH}/shop_help.png"):
|
||
os.remove(f"{IMAGE_PATH}/shop_help.png")
|
||
if Config.get_config("shop", "IMPORT_DEFAULT_SHOP_GOODS"):
|
||
await add_goods(["好感度双倍加持卡Ⅰ", 30, "下次签到双倍好感度概率 + 10%（谁才是真命天子？）（同类商品将覆盖）"])
|
||
await add_goods(["好感度双倍加持卡Ⅱ", 150, "下次签到双倍好感度概率 + 20%（平平庸庸）（同类商品将覆盖）"])
|
||
await add_goods(["好感度双倍加持卡Ⅲ", 250, "下次签到双倍好感度概率 + 30%（金币才是真命天子！）（同类商品将覆盖）"])
|
||
|
||
|
||
# 创建商店界面
|
||
async def create_shop_help():
|
||
goods_lst = await GoodsInfo.get_all_goods()
|
||
tmp = ""
|
||
idx = 1
|
||
for goods in goods_lst:
|
||
tmp += (
|
||
f"{idx}.{goods.goods_name}\t\t售价：{goods.goods_price}金币\n"
|
||
f"\t\t{goods.goods_description}\n"
|
||
)
|
||
idx += 1
|
||
w = 1000
|
||
h = 400 + len(goods_lst) * 40
|
||
h = 1000 if h < 1000 else h
|
||
shop_logo = BuildImage(100, 100, background=f"{IMAGE_PATH}/other/shop_text.png")
|
||
shop = BuildImage(w, h, font_size=20)
|
||
zhenxun_img = BuildImage(525, 581, background=f"{IMAGE_PATH}/zhenxun/toukan_2.png")
|
||
shop.paste(zhenxun_img, (780, 100))
|
||
shop.paste(shop_logo, (450, 30), True)
|
||
shop.text(
|
||
(int((1000 - shop.getsize("注【通过 序号 或者 商品名称 购买】")[0]) / 2), 170),
|
||
"注【通过 序号 或者 商品名称 购买】",
|
||
)
|
||
shop.text((20, 230), tmp[:-1])
|
||
shop.text((20, h - 100), "神秘药水\t\t售价：9999999金币\n\t\t鬼知道会有什么效果~")
|
||
shop.save(f"{IMAGE_PATH}/shop_help.png")
|
||
|
||
|
||
# 添加商品
|
||
async def add_goods(msg: list):
|
||
data = {
|
||
"名称": None,
|
||
"价格": None,
|
||
"描述": None,
|
||
"折扣": 1,
|
||
"限时": 0,
|
||
}
|
||
keys = list(data.keys())
|
||
idx = 0
|
||
for x in msg:
|
||
data[keys[idx]] = x
|
||
idx += 1
|
||
return await GoodsInfo.add_goods(
|
||
data["名称"], data["价格"], data["描述"], data["折扣"], data["限时"]
|
||
)
|
||
|
||
|
||
# 删除商品
|
||
async def del_goods(name: str, id_: int):
|
||
goods_lst = await GoodsInfo.get_all_goods()
|
||
if id_:
|
||
if id_ < 1 or id_ > len(goods_lst):
|
||
return "序号错误，没有该序号商品...", "", 999
|
||
goods_name = goods_lst[id_ - 1].goods_name
|
||
if await GoodsInfo.delete_goods(goods_name):
|
||
return f"删除商品 {goods_name} 成功！", goods_name, 200
|
||
else:
|
||
return f"删除商品 {goods_name} 失败！", goods_name, 999
|
||
if name:
|
||
if await GoodsInfo.delete_goods(name):
|
||
return f"删除商品 {name} 成功！", name, 200
|
||
else:
|
||
return f"删除商品 {name} 失败！", name, 999
|
||
|
||
|
||
# 更新商品信息
|
||
async def update_goods(data: dict):
|
||
goods_lst = await GoodsInfo.get_all_goods()
|
||
if is_number(data["name"]):
|
||
if data["name"] < 1 or data["name"] > len(goods_lst):
|
||
return "序号错误，没有该序号的商品...", "", 999
|
||
goods = goods_lst[data["name"] - 1]
|
||
else:
|
||
goods = await GoodsInfo.get_goods_info(data["name"])
|
||
if not goods:
|
||
return "名称错误，没有该名称的商品...", "", 999
|
||
name = goods.goods_name
|
||
price = goods.goods_price
|
||
des = goods.goods_description
|
||
discount = goods.goods_discount
|
||
limit_time = goods.goods_limit_time
|
||
tmp = ""
|
||
if data.get("price"):
|
||
tmp += f'价格：{price} --> {data["price"]}\n'
|
||
price = data["price"]
|
||
if data.get("des"):
|
||
tmp += f'描述：{des} --> {data["des"]}\n'
|
||
des = data["des"]
|
||
if data.get("discount"):
|
||
tmp += f'折扣：{discount} --> {data["discount"]}\n'
|
||
discount = data["discount"]
|
||
if data.get("time"):
|
||
old_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(limit_time))
|
||
new_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(data["time"]))
|
||
tmp += f"折扣：{old_time} --> {new_time}\n"
|
||
limit_time = data["time"]
|
||
return (
|
||
await GoodsInfo.update_goods(name, price, des, discount, limit_time),
|
||
name,
|
||
tmp[:-1],
|
||
)
|