diff options
Diffstat (limited to 'ATRI/plugins/kimo')
| -rw-r--r-- | ATRI/plugins/kimo/__init__.py | 3 | ||||
| -rw-r--r-- | ATRI/plugins/kimo/data_source.py | 6 |
2 files changed, 5 insertions, 4 deletions
diff --git a/ATRI/plugins/kimo/__init__.py b/ATRI/plugins/kimo/__init__.py index b1c5698..31ecab1 100644 --- a/ATRI/plugins/kimo/__init__.py +++ b/ATRI/plugins/kimo/__init__.py @@ -21,7 +21,8 @@ async def _chat(event: MessageEvent): msg = str(event.message) repo = await Kimo().deal(msg, user_id) try: - await kimo.finish(repo) + if repo: + await kimo.send(repo) except Exception: return diff --git a/ATRI/plugins/kimo/data_source.py b/ATRI/plugins/kimo/data_source.py index 6dd448a..fb3f3f3 100644 --- a/ATRI/plugins/kimo/data_source.py +++ b/ATRI/plugins/kimo/data_source.py @@ -8,7 +8,7 @@ from ATRI.service import Service from ATRI.rule import to_bot, is_in_service from ATRI.log import logger as log from ATRI.utils import request -from ATRI.exceptions import ReadFileError, WriteError +from ATRI.exceptions import ReadFileError, WriteFileError CHAT_PATH = Path(".") / "data" / "database" / "kimo" @@ -39,8 +39,8 @@ class Kimo(Service): with open(path, "w", encoding="utf-8") as w: w.write(json.dumps(data, indent=4)) log.info("生成完成") - except WriteError: - raise WriteError("Writing kimo words failed!") + except WriteFileError: + raise WriteFileError("Writing kimo words failed!") @classmethod async def _load_data(cls) -> dict: |