mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import nonebot
|
|
from nonebot.drivers import Driver
|
|
|
|
from zhenxun.configs.config import BotConfig
|
|
from zhenxun.utils.decorator.shop import NotMeetUseConditionsException, shop_register
|
|
|
|
from .config import base_config
|
|
from .data_source import Conversation
|
|
|
|
driver: Driver = nonebot.get_driver()
|
|
|
|
|
|
@shop_register(
|
|
name="失忆卡",
|
|
price=200,
|
|
des=f"当你养成失败或{BotConfig.self_nickname}变得奇怪时，你需要这个道具。",
|
|
icon="reload_ai_card.png",
|
|
)
|
|
async def _(user_id: str):
|
|
await Conversation.reset(user_id, None)
|
|
return f"{BotConfig.self_nickname}忘记了你之前说过的话，仿佛一切可以重新开始..."
|
|
|
|
|
|
@shop_register(
|
|
name="群组失忆卡",
|
|
price=300,
|
|
des=f"当群聊内{BotConfig.self_nickname}变得奇怪时，你需要这个道具。",
|
|
icon="reload_ai_card1.png",
|
|
)
|
|
async def _(user_id: str, group_id: str):
|
|
await Conversation.reset(user_id, group_id)
|
|
return f"前面忘了，后面忘了，{BotConfig.self_nickname}重新睁开了眼睛..."
|
|
|
|
|
|
@shop_register.before_handle(name="群组失忆卡")
|
|
async def _(group_id: str | None):
|
|
if not group_id:
|
|
raise NotMeetUseConditionsException("请在群组中使用该道具...")
|
|
if not base_config.get("ENABLE_GROUP_CHAT"):
|
|
raise NotMeetUseConditionsException(
|
|
"当前未开启群组个人记忆分离，无法使用道具。"
|
|
)
|