diff options
| author | SakuraMemory <51871263+Yuki-Asuuna@users.noreply.github.com> | 2022-05-19 17:34:02 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-05-19 17:34:02 +0800 |
| commit | a592d64297155f40fce3bc68839ebaf9f0bed3c2 (patch) | |
| tree | 87744575d3313f6e920f8d73bd37f021e5691bce /ATRI/plugins/kimo | |
| parent | bb624bdd6279335861205d87dae61780775f7b10 (diff) | |
| parent | 556ce306a6de1ae8804e00a1a88ecde2f59f2d10 (diff) | |
| download | ATRI-a592d64297155f40fce3bc68839ebaf9f0bed3c2.tar.gz ATRI-a592d64297155f40fce3bc68839ebaf9f0bed3c2.tar.bz2 ATRI-a592d64297155f40fce3bc68839ebaf9f0bed3c2.zip | |
Merge pull request #7 from Kyomotoi/main
merge master
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: |