mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
|
||
from models.sign_group_user import SignGroupUser
|
||
from configs.config import Config
|
||
from nonebot import Driver
|
||
from utils.decorator.shop import shop_register, NotMeetUseConditionsException
|
||
import nonebot
|
||
|
||
driver: Driver = nonebot.get_driver()
|
||
|
||
|
||
@driver.on_startup
|
||
async def _():
|
||
"""
|
||
导入内置的三个商品
|
||
"""
|
||
|
||
@shop_register(
|
||
name=("好感度双倍加持卡Ⅰ", "好感度双倍加持卡Ⅱ", "好感度双倍加持卡Ⅲ"),
|
||
price=(30, 150, 250),
|
||
des=(
|
||
"下次签到双倍好感度概率 + 10%（谁才是真命天子？）（同类商品将覆盖）",
|
||
"下次签到双倍好感度概率 + 20%（平平庸庸）（同类商品将覆盖）",
|
||
"下次签到双倍好感度概率 + 30%（金币才是真命天子！）（同类商品将覆盖）",
|
||
),
|
||
load_status=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},
|
||
)
|
||
async def sign_card(user_id: int, group_id: int, prob: float):
|
||
user = await SignGroupUser.ensure(user_id, group_id)
|
||
await user.update(add_probability=prob).apply()
|
||
|
||
@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）')
|
||
|
||
|
||
@driver.on_bot_connect
|
||
async def _():
|
||
await shop_register.load_register()
|
||
|