mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git
synced 2025-02-01 15:31:24 +08:00
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
|
import json
|
|||
|
|
|
|||
|
|
from arclet.alconna import Alconna, Args, Subcommand
|
|||
|
|
from nonebot import logger
|
|||
|
|
from nonebot.adapters import Bot, Event
|
|||
|
|
from nonebot.matcher import Matcher
|
|||
|
|
from nonebot.typing import T_State
|
|||
|
|
from nonebot_plugin_alconna import on_alconna
|
|||
|
|
from nonebot_plugin_localstore import get_plugin_data_file
|
|||
|
|
|
|||
|
|
from nonebot_plugin_marshoai.config import config
|
|||
|
|
|
|||
|
|
marsho_memory_cmd = on_alconna(
|
|||
|
|
Alconna(
|
|||
|
|
f"{config.marshoai_default_name}.memory",
|
|||
|
|
Subcommand("view", alias={"v"}),
|
|||
|
|
Subcommand("reset", alias={"r"}),
|
|||
|
|
),
|
|||
|
|
priority=10,
|
|||
|
|
block=True,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
memory_path = get_plugin_data_file("memory.json")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@marsho_memory_cmd.assign("view")
|
|||
|
|
async def view_memory(matcher: Matcher, state: T_State, event: Event):
|
|||
|
|
user_id = str(event.get_user_id())
|
|||
|
|
with open(memory_path, "r", encoding="utf-8") as f:
|
|||
|
|
memory_data = json.load(f)
|
|||
|
|
memorys = memory_data.get(user_id, [])
|
|||
|
|
if not memorys:
|
|||
|
|
await matcher.finish("好像对ta还没有任何记忆呢~")
|
|||
|
|
await matcher.finish("这些是有关ta的记忆：" + "\n".join(memorys))
|
|||
|
|
|
|||
|
|
|
|||
|
|
@marsho_memory_cmd.assign("reset")
|
|||
|
|
async def reset_memory(matcher: Matcher, state: T_State, event: Event):
|
|||
|
|
user_id = str(event.get_user_id())
|
|||
|
|
with open(memory_path, "r", encoding="utf-8") as f:
|
|||
|
|
memory_data = json.load(f)
|
|||
|
|
if user_id in memory_data:
|
|||
|
|
del memory_data[user_id]
|
|||
|
|
with open(memory_path, "w", encoding="utf-8") as f:
|
|||
|
|
json.dump(memory_data, f, ensure_ascii=False, indent=4)
|
|||
|
|
await matcher.finish("记忆已重置~")
|
|||
|
|
await matcher.finish("没有找到该用户的记忆~")
|