blob: 3be8abd2ac1baada53f1c7d4bff93be0c798d231 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
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
|