mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
|
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=bool(Config.get_config("shop", "IMPORT_DEFAULT_SHOP_GOODS")),
|
|||
|
|
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):
|
|||
|
|
user_console, _ = await UserConsole.get_or_create(
|
|||
|
|
user_id=session.id1,
|
|||
|
|
defaults={
|
|||
|
|
"uid": await UserConsole.get_new_uid(),
|
|||
|
|
"platform": 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）")
|