diff options
Diffstat (limited to 'ATRI/plugins')
| -rw-r--r-- | ATRI/plugins/kimo/__init__.py (renamed from ATRI/plugins/chat/__init__.py) | 33 | ||||
| -rw-r--r-- | ATRI/plugins/kimo/data_source.py (renamed from ATRI/plugins/chat/data_source.py) | 8 | 
2 files changed, 13 insertions, 28 deletions
| diff --git a/ATRI/plugins/chat/__init__.py b/ATRI/plugins/kimo/__init__.py index 3cb5dbb..b1c5698 100644 --- a/ATRI/plugins/chat/__init__.py +++ b/ATRI/plugins/kimo/__init__.py @@ -6,27 +6,27 @@ from nonebot.adapters.onebot.v11 import MessageEvent, Message  from nonebot.adapters.onebot.v11.helpers import Cooldown  from ATRI.utils.apscheduler import scheduler -from .data_source import Chat +from .data_source import Kimo  _chat_flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇会~~", "我开始为你以后的伴侣担心了..."]) -chat = Chat().on_message("文爱", "闲聊(文爱") +kimo = Kimo().on_message("文爱", "闲聊(文爱") -@chat.handle([Cooldown(3, prompt=_chat_flmt_notice)]) +@kimo.handle([Cooldown(3, prompt=_chat_flmt_notice)])  async def _chat(event: MessageEvent):      user_id = event.get_user_id()      msg = str(event.message) -    repo = await Chat().deal(msg, user_id) +    repo = await Kimo().deal(msg, user_id)      try: -        await chat.finish(repo) +        await kimo.finish(repo)      except Exception:          return -my_name_is = Chat().on_command("叫我", "更改闲聊(文爱)时的称呼", aliases={"我是"}, priority=1) +my_name_is = Kimo().on_command("叫我", "更改kimo时的称呼", aliases={"我是"}, priority=1)  @my_name_is.handle([Cooldown(3, prompt=_chat_flmt_notice)]) @@ -47,28 +47,13 @@ async def _deal_name(event: MessageEvent, new_name: str = ArgPlainText("name")):              "很不错的称呼呢w",          ]      ) -    Chat().name_is(user_id, new_name) +    Kimo().name_is(user_id, new_name)      await my_name_is.finish(repo) -say = Chat().on_command("说", "别人让我说啥就说啥(", priority=1) - - -@say.handle([Cooldown(3, prompt=_chat_flmt_notice)]) -async def _ready_say(matcher: Matcher, args: Message = CommandArg()): -    msg = args.extract_plain_text() -    if msg: -        matcher.set_arg("say", args) - - -@say.got("say", "想要咱复读啥呢...") -async def _deal_say(msg: str = ArgPlainText("say")): -    await say.finish(msg) - - -@scheduler.scheduled_job("interval", name="闲聊词库检查更新", hours=3, misfire_grace_time=60)  # type: ignore +@scheduler.scheduled_job("interval", name="kimo词库检查更新", hours=3, misfire_grace_time=60)  # type: ignore  async def _check_kimo():      try: -        await Chat().update_data() +        await Kimo().update_data()      except BaseException:          pass diff --git a/ATRI/plugins/chat/data_source.py b/ATRI/plugins/kimo/data_source.py index 70c6345..6dd448a 100644 --- a/ATRI/plugins/chat/data_source.py +++ b/ATRI/plugins/kimo/data_source.py @@ -11,15 +11,15 @@ from ATRI.utils import request  from ATRI.exceptions import ReadFileError, WriteError -CHAT_PATH = Path(".") / "data" / "database" / "chat" +CHAT_PATH = Path(".") / "data" / "database" / "kimo"  os.makedirs(CHAT_PATH, exist_ok=True)  KIMO_URL = "https://cdn.jsdelivr.net/gh/Kyomotoi/AnimeThesaurus/data.json" -class Chat(Service): +class Kimo(Service):      def __init__(self):          Service.__init__( -            self, "闲聊", "好像有点涩?", rule=to_bot() & is_in_service("闲聊"), priority=5 +            self, "kimo", "好像有点涩?", rule=to_bot() & is_in_service("kimo"), priority=5          )      @staticmethod @@ -69,7 +69,7 @@ class Chat(Service):          with open(path, "w", encoding="utf-8") as w:              w.write(json.dumps(data, indent=4)) -        log.info("闲聊词库更新完成") +        log.info("kimo词库更新完成")      @staticmethod      def name_is(user_id: str, new_name: str): | 
