diff options
| author | 2023-04-19 01:23:44 +0800 | |
|---|---|---|
| committer | 2023-04-19 01:23:44 +0800 | |
| commit | 23b32b900423ae8fe36a50224f0308cb3d97d4d0 (patch) | |
| tree | b5a59158f31d7aed0e3d65b0cb1563e7e5ff7edc /ChienDice/plugins/iamai_plugin_reply | |
| parent | a69a07edfedf5e15d8013fb639ed6574e0078988 (diff) | |
| download | HydroRoll-23b32b900423ae8fe36a50224f0308cb3d97d4d0.tar.gz HydroRoll-23b32b900423ae8fe36a50224f0308cb3d97d4d0.zip | |
🏷bump
Diffstat (limited to 'ChienDice/plugins/iamai_plugin_reply')
| -rw-r--r-- | ChienDice/plugins/iamai_plugin_reply/__init__.py | 43 | ||||
| -rw-r--r-- | ChienDice/plugins/iamai_plugin_reply/config.py | 11 |
2 files changed, 54 insertions, 0 deletions
diff --git a/ChienDice/plugins/iamai_plugin_reply/__init__.py b/ChienDice/plugins/iamai_plugin_reply/__init__.py new file mode 100644 index 0000000..3be8abd --- /dev/null +++ b/ChienDice/plugins/iamai_plugin_reply/__init__.py @@ -0,0 +1,43 @@ +import re +import json + +from plugins.iamai_plugin_base import BasePlugin + +from .config import Config + + +class Reply(BasePlugin[None, Config]): + priority: int = 1 + Config = Config + + def __post_init__(self): + with open(self.config.data_file, "r") as fp: + if self.config.data_type == "json": + json_data = json.load(fp) + else: + raise ValueError(f"data_type must be json, not {self.config.data_type}") + self.rule_to_message = { + item["rule"]: item["message"] + for item in json_data + if isinstance(item, dict) + and "rule" in item.keys() + and "message" in item.keys() + } + + async def handle(self) -> None: + msg = self.rule_to_message[self.msg_match.re.pattern] + if isinstance(msg, str): + await self.event.reply(self.format_str(msg, self.msg_match.string)) + else: + await self.event.reply(msg) + + def str_match(self, msg_str: str) -> bool: + msg_str = msg_str.strip() + for rule in self.rule_to_message.keys(): + msg_match = re.fullmatch( + rule, msg_str, flags=re.I if self.config.ignore_case else 0 + ) + if msg_match: + self.msg_match = msg_match + return bool(self.msg_match) + return False diff --git a/ChienDice/plugins/iamai_plugin_reply/config.py b/ChienDice/plugins/iamai_plugin_reply/config.py new file mode 100644 index 0000000..e024fb6 --- /dev/null +++ b/ChienDice/plugins/iamai_plugin_reply/config.py @@ -0,0 +1,11 @@ +from plugins.iamai_plugin_base import CommandPluginConfig + + +class Config(CommandPluginConfig): + __config_name__ = "plugin_reply" + data_type: str = "json" + """数据类型,目前只支持 json。""" + data_file: str = "data/reply_data.json" + """数据文件位置。""" + ignore_case: bool = True + """是否忽略大小写,默认为 True。""" |
