mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-06 04:10:02 +08:00
🐛 修复使用道具后未减去道具数量
This commit is contained in:
@@ -144,7 +144,7 @@ class UserConsole(Model):
|
||||
async def add_props_by_name(
|
||||
cls, user_id: str, name: str, num: int = 1, platform: str | None = None
|
||||
):
|
||||
"""添加道具
|
||||
"""根据名称添加道具
|
||||
|
||||
参数:
|
||||
user_id: 用户id
|
||||
@@ -155,3 +155,41 @@ class UserConsole(Model):
|
||||
if goods := await GoodsInfo.get_or_none(goods_name=name):
|
||||
return await cls.add_props(user_id, goods.uuid, num, platform)
|
||||
raise GoodsNotFound("未找到商品...")
|
||||
|
||||
@classmethod
|
||||
async def use_props(
|
||||
cls, user_id: str, goods_uuid: str, num: int = 1, platform: str | None = None
|
||||
):
|
||||
"""添加道具
|
||||
|
||||
参数:
|
||||
user_id: 用户id
|
||||
goods_uuid: 道具uuid
|
||||
num: 道具数量.
|
||||
platform: 平台.
|
||||
"""
|
||||
user, _ = await cls.get_or_create(
|
||||
user_id=user_id,
|
||||
defaults={"platform": platform, "uid": await cls.get_new_uid()},
|
||||
)
|
||||
|
||||
if goods_uuid not in user.props or user.props[goods_uuid] < num:
|
||||
raise GoodsNotFound("未找到商品或道具数量不足...")
|
||||
user.props[goods_uuid] -= num
|
||||
await user.save(update_fields=["props"])
|
||||
|
||||
@classmethod
|
||||
async def use_props_by_name(
|
||||
cls, user_id: str, name: str, num: int = 1, platform: str | None = None
|
||||
):
|
||||
"""根据名称添加道具
|
||||
|
||||
参数:
|
||||
user_id: 用户id
|
||||
name: 道具名称
|
||||
num: 道具数量.
|
||||
platform: 平台.
|
||||
"""
|
||||
if goods := await GoodsInfo.get_or_none(goods_name=name):
|
||||
return await cls.use_props(user_id, goods.uuid, num, platform)
|
||||
raise GoodsNotFound("未找到商品...")
|
||||
|
||||
Reference in New Issue
Block a user