zhenxun_bot/zhenxun/builtin_plugins/sign_in/goods_register.py

from decimal import Decimal

import nonebot
from nonebot.drivers import Driver
from nonebot_plugin_session import EventSession

from zhenxun.configs.config import Config
from zhenxun.models.sign_user import SignUser
from zhenxun.models.user_console import UserConsole
from zhenxun.utils.decorator.shop import NotMeetUseConditionsException, shop_register

driver: Driver = nonebot.get_driver()


# @driver.on_startup
# async def _():
#     """
#     导入内置的三个商品
#     """


@shop_register(
    name=("好感度双倍加持卡Ⅰ", "好感度双倍加持卡Ⅱ", "好感度双倍加持卡Ⅲ"),
    price=(30, 150, 250),
    des=(
        "下次签到双倍好感度概率 + 10%（谁才是真命天子？）（同类商品将覆盖）",
        "下次签到双倍好感度概率 + 20%（平平庸庸）（同类商品将覆盖）",
        "下次签到双倍好感度概率 + 30%（金币才是真命天子！）（同类商品将覆盖）",
    ),
    load_status=True,
    icon=(
        "favorability_card_1.png",
        "favorability_card_2.png",
        "favorability_card_3.png",
    ),
    **{"好感度双倍加持卡Ⅰ_prob": 0.1, "好感度双倍加持卡Ⅱ_prob": 0.2, "好感度双倍加持卡Ⅲ_prob": 0.3},  # type: ignore
)
async def _(session: EventSession, user_id: int, group_id: int, prob: float):
    if session.id1:
        user_console = await UserConsole.get_user(session.id1, session.platform)
        user, _ = await SignUser.get_or_create(
            user_id=user_id,
            defaults={"platform": session.platform, "user_console": user_console},
        )
        user.add_probability = Decimal(prob)
        await user.save(update_fields=["add_probability"])


@shop_register(
    name="测试道具A",
    price=99,
    des="随便侧而出",
    load_status=False,
    icon="sword.png",
)
async def _(user_id: int, group_id: int):
    print(user_id, group_id, "使用测试道具")


@shop_register.before_handle(name="测试道具A", load_status=False)
async def _(user_id: int, group_id: int):
    print(user_id, group_id, "第一个使用前函数（before handle）")


@shop_register.before_handle(name="测试道具A", load_status=False)
async def _(user_id: int, group_id: int):
    print(user_id, group_id, "第二个使用前函数（before handle）222")
    raise NotMeetUseConditionsException("太笨了！")  # 抛出异常，阻断使用，并返回信息


@shop_register.after_handle(name="测试道具A", load_status=False)
async def _(user_id: int, group_id: int):
    print(user_id, group_id, "第一个使用后函数（after handle）")