mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
|
from decimal import Decimal
|
||
|
|
import random
|
||
|
|
|
||
|
|
import nonebot
|
||
|
|
from nonebot.drivers import Driver
|
||
|
|
|
||
|
|
from zhenxun.configs.config import BotConfig
|
||
|
|
from zhenxun.models.sign_user import SignUser
|
||
|
|
from zhenxun.models.user_console import UserConsole
|
||
|
|
|
||
|
|
from .gift_register import gift_register
|
||
|
|
|
||
|
|
driver: Driver = nonebot.get_driver()
|
||
|
|
|
||
|
|
|
||
|
|
@gift_register(
|
||
|
|
name="可爱的钱包",
|
||
|
|
icon="wallet.png",
|
||
|
|
description=f"这是{BotConfig.self_nickname}的小钱包，里面装了一些金币。",
|
||
|
|
)
|
||
|
|
async def _(user_id: str):
|
||
|
|
rand = random.randint(100, 500)
|
||
|
|
await UserConsole.add_gold(user_id, rand, "BYM_AI")
|
||
|
|
return f"钱包里装了{BotConfig.self_nickname}送给你的枚{rand}金币哦~"
|
||
|
|
|
||
|
|
|
||
|
|
@gift_register(
|
||
|
|
name="小发夹",
|
||
|
|
icon="hairpin.png",
|
||
|
|
description=f"这是{BotConfig.self_nickname}的发夹，里面是真寻对你的期望。",
|
||
|
|
)
|
||
|
|
async def _(user_id: str):
|
||
|
|
rand = random.uniform(0.01, 0.5)
|
||
|
|
user = await SignUser.get_user(user_id)
|
||
|
|
user.impression += Decimal(rand)
|
||
|
|
await user.save(update_fields=["impression"])
|
||
|
|
return f"你使用了小发夹，{BotConfig.self_nickname}对你提升了{rand:.2f}好感度~"
|
||
|
|
|
||
|
|
|
||
|
|
@driver.on_startup
|
||
|
|
async def _():
|
||
|
|
await gift_register.load_register()
|