Update buy.py

This commit is contained in:
HibiKier 2022-02-10 11:55:36 +08:00 committed by GitHub
parent c1bc4eff2b
commit 1e63ca2549
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -41,11 +41,15 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
goods = None goods = None
if get_message_text(event.json()) in ["神秘药水"]: if get_message_text(event.json()) in ["神秘药水"]:
await buy.finish("你们看看就好啦,这是不可能卖给你们的~", at_sender=True) await buy.finish("你们看看就好啦,这是不可能卖给你们的~", at_sender=True)
goods_name_lst = [ goods_list = [
x.goods_name x
for x in await GoodsInfo.get_all_goods() for x in await GoodsInfo.get_all_goods()
if x.goods_limit_time > time.time() or x.goods_limit_time == 0 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 = get_message_text(event.json()).split() msg = get_message_text(event.json()).split()
num = 1 num = 1
if len(msg) > 1: if len(msg) > 1:
@ -55,14 +59,14 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
await buy.finish("购买的数量要是数字且大于0", at_sender=True) await buy.finish("购买的数量要是数字且大于0", at_sender=True)
if is_number(msg[0]): if is_number(msg[0]):
msg = int(msg[0]) msg = int(msg[0])
if msg > len(goods_name_lst) or msg < 1: if msg > len(goods_name_list) or msg < 1:
await buy.finish("请输入正确的商品id", at_sender=True) await buy.finish("请输入正确的商品id", at_sender=True)
goods = goods_name_lst[msg - 1] goods = goods_list[msg - 1]
else: else:
if msg[0] in goods_name_lst: if msg[0] in goods_name_list:
for i in range(len(goods_name_lst)): for i in range(len(goods_name_list)):
if msg[0] == goods_name_lst[i]: if msg[0] == goods_name_list[i]:
goods = goods_name_lst[i] goods = goods_list[i]
break break
else: else:
await buy.finish("请输入正确的商品名称!") await buy.finish("请输入正确的商品名称!")