mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
110 lines
4.0 KiB
Python
110 lines
4.0 KiB
Python
import time
|
||
|
||
from nonebot import on_command
|
||
from nonebot.adapters.onebot.v11 import GroupMessageEvent, Message
|
||
from nonebot.adapters.onebot.v11.permission import GROUP
|
||
from nonebot.params import CommandArg
|
||
|
||
from models.bag_user import BagUser
|
||
from models.goods_info import GoodsInfo
|
||
from models.user_shop_gold_log import UserShopGoldLog
|
||
from services.log import logger
|
||
from utils.utils import is_number
|
||
|
||
__zx_plugin_name__ = "商店 - 购买道具"
|
||
__plugin_usage__ = """
|
||
usage：
|
||
购买道具
|
||
指令：
|
||
购买 [序号或名称] ?[数量=1]
|
||
示例：购买 好感双倍加持卡Ⅰ
|
||
示例：购买 1 4
|
||
""".strip()
|
||
__plugin_des__ = "商店 - 购买道具"
|
||
__plugin_cmd__ = ["购买 [序号或名称] ?[数量=1]"]
|
||
__plugin_type__ = ("商店",)
|
||
__plugin_version__ = 0.1
|
||
__plugin_author__ = "HibiKier"
|
||
__plugin_settings__ = {
|
||
"level": 5,
|
||
"default_status": True,
|
||
"limit_superuser": False,
|
||
"cmd": ["商店", "购买道具"],
|
||
}
|
||
__plugin_cd_limit__ = {"cd": 3}
|
||
|
||
|
||
buy = on_command("购买", aliases={"购买道具"}, priority=5, block=True, permission=GROUP)
|
||
|
||
|
||
@buy.handle()
|
||
async def _(event: GroupMessageEvent, arg: Message = CommandArg()):
|
||
goods = None
|
||
if arg.extract_plain_text().strip() in ["神秘药水"]:
|
||
await buy.finish("你们看看就好啦，这是不可能卖给你们的~", at_sender=True)
|
||
goods_list = [
|
||
x
|
||
for x in await GoodsInfo.get_all_goods()
|
||
if x.goods_limit_time > time.time() or x.goods_limit_time == 0
|
||
]
|
||
goods_name_list = [x.goods_name for x in goods_list]
|
||
msg = arg.extract_plain_text().strip().split()
|
||
num = 1
|
||
if len(msg) > 1:
|
||
if is_number(msg[1]) and int(msg[1]) > 0:
|
||
num = int(msg[1])
|
||
else:
|
||
await buy.finish("购买的数量要是数字且大于0！", at_sender=True)
|
||
if is_number(msg[0]):
|
||
msg = int(msg[0])
|
||
if msg > len(goods_name_list) or msg < 1:
|
||
await buy.finish("请输入正确的商品id！", at_sender=True)
|
||
goods = goods_list[msg - 1]
|
||
else:
|
||
if msg[0] in goods_name_list:
|
||
for i in range(len(goods_name_list)):
|
||
if msg[0] == goods_name_list[i]:
|
||
goods = goods_list[i]
|
||
break
|
||
else:
|
||
await buy.finish("请输入正确的商品名称！")
|
||
else:
|
||
await buy.finish("请输入正确的商品名称！", at_sender=True)
|
||
if (
|
||
await BagUser.get_gold(event.user_id, event.group_id)
|
||
) < goods.goods_price * num * goods.goods_discount:
|
||
await buy.finish("您的金币好像不太够哦", at_sender=True)
|
||
flag, n = await GoodsInfo.check_user_daily_purchase(
|
||
goods, event.user_id, event.group_id, num
|
||
)
|
||
if flag:
|
||
await buy.finish(f"该次购买将超过每日次数限制，目前该道具还可以购买{n}次哦", at_sender=True)
|
||
spend_gold = int(goods.goods_discount * goods.goods_price * num)
|
||
await BagUser.spend_gold(event.user_id, event.group_id, spend_gold)
|
||
await BagUser.add_property(event.user_id, event.group_id, goods.goods_name, num)
|
||
await GoodsInfo.add_user_daily_purchase(goods, event.user_id, event.group_id, num)
|
||
await buy.send(
|
||
f"花费 {goods.goods_price * num * goods.goods_discount} 金币购买 {goods.goods_name} ×{num} 成功！",
|
||
at_sender=True,
|
||
)
|
||
logger.info(
|
||
f"花费 {goods.goods_price*num} 金币购买 {goods.goods_name} ×{num} 成功！",
|
||
"购买道具",
|
||
event.user_id,
|
||
event.group_id,
|
||
)
|
||
await UserShopGoldLog.create(
|
||
user_id=str(event.user_id),
|
||
group_id=str(event.group_id),
|
||
type=0,
|
||
name=goods.goods_name,
|
||
num=num,
|
||
spend_gold=goods.goods_price * num * goods.goods_discount,
|
||
)
|
||
# else:
|
||
# await buy.send(f"{goods.goods_name} 购买失败！", at_sender=True)
|
||
# logger.info(
|
||
# f"USER {event.user_id} GROUP {event.group_id} "
|
||
# f"花费 {goods.goods_price * num * goods.goods_discount} 金币购买 {goods.goods_name} ×{num} 失败！"
|
||
# )
|