Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-10-16 01:30:03 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor code

Browse Source
This commit is contained in:
hibiki
2021-07-30 21:21:51 +08:00
parent 2ad891aa1e
commit cc24822dca
165 changed files with 7815 additions and 8174 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/shop/buy.py
+29 -26
View File
@@ -9,31 +9,31 @@ from nonebot.adapters.cqhttp.permission import GROUP
from models.goods_info import GoodsInfo
__plugin_name__ = '商店购买'
__plugin_usage__ = '格式：购买 [名称或序号] [数量](默认1)\n\t示例：购买 好感双倍加持卡Ⅰ\n\t示例：购买 1 4'
__plugin_name__ = "商店购买"
__plugin_usage__ = "格式：购买 [名称或序号] [数量](默认1)\n\t示例：购买 好感双倍加持卡Ⅰ\n\t示例：购买 1 4"
buy = on_command("购买", aliases={'购买道具'}, priority=5, block=True, permission=GROUP)
buy = on_command("购买", aliases={"购买道具"}, priority=5, block=True, permission=GROUP)
@buy.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
if get_message_text(event.json()) in ['神秘药水']:
if get_message_text(event.json()) in ["神秘药水"]:
await buy.finish("你们看看就好啦，这是不可能卖给你们的~", at_sender=True)
goods_lst = await GoodsInfo.get_all_goods()
goods_name_lst = [x.goods_name for x in goods_lst]
msg = get_message_text(event.json()).strip().split(' ')
msg = get_message_text(event.json()).strip().split(" ")
num = 1
if len(msg) > 1:
if is_number(msg[1]):
num = int(msg[1])
else:
await buy.finish('购买的数量要是数字！', at_sender=True)
await buy.finish("购买的数量要是数字！", at_sender=True)
# print(msg, num)
if is_number(msg[0]):
msg = int(msg[0])
if msg > len(goods_lst) or msg < 1:
await buy.finish('请输入正确的商品id！', at_sender=True)
await buy.finish("请输入正确的商品id！", at_sender=True)
goods = goods_lst[msg - 1]
else:
if msg[0] in goods_name_lst:
@@ -42,27 +42,30 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
goods = goods_lst[i]
break
else:
await buy.finish('请输入正确的商品名称！')
await buy.finish("请输入正确的商品名称！")
else:
await buy.finish('请输入正确的商品名称！', at_sender=True)
await buy.finish("请输入正确的商品名称！", at_sender=True)
async with db.transaction():
if (await BagUser.get_gold(event.user_id, event.group_id)) < goods.goods_price * num:
await buy.finish('您的金币好像不太够哦', at_sender=True)
if await BagUser.spend_gold(event.user_id, event.group_id, goods.goods_price * num):
if (
await BagUser.get_gold(event.user_id, event.group_id)
) < goods.goods_price * num:
await buy.finish("您的金币好像不太够哦", at_sender=True)
if await BagUser.spend_gold(
event.user_id, event.group_id, goods.goods_price * num
):
for _ in range(num):
await BagUser.add_props(event.user_id, event.group_id, goods.goods_name)
await buy.send(f'花费 {goods.goods_price*num} 金币购买 {goods.goods_name} ×{num} 成功！', at_sender=True)
logger.info(f'USER {event.user_id} GROUP {event.group_id} '
f'花费 {goods.goods_price*num} 金币购买 {goods.goods_name} ×{num} 成功！')
await buy.send(
f"花费 {goods.goods_price*num} 金币购买 {goods.goods_name} ×{num} 成功！",
at_sender=True,
)
logger.info(
f"USER {event.user_id} GROUP {event.group_id} "
f"花费 {goods.goods_price*num} 金币购买 {goods.goods_name} ×{num} 成功！"
)
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_name} ×{num} 失败！')
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_name} ×{num} 失败！"
)