mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 19:57:04 +08:00
35 lines
889 B
Python
35 lines
889 B
Python
import re
|
|
|
|
|
|
from .. import rule as rule_model
|
|
from .. import entities
|
|
from ....core import entities as core_entities
|
|
from ....platform.types import message as platform_message
|
|
|
|
|
|
@rule_model.rule_class("regexp")
|
|
class RegExpRule(rule_model.GroupRespondRule):
|
|
|
|
async def match(
|
|
self,
|
|
message_text: str,
|
|
message_chain: platform_message.MessageChain,
|
|
rule_dict: dict,
|
|
query: core_entities.Query
|
|
) -> entities.RuleJudgeResult:
|
|
regexps = rule_dict['regexp']
|
|
|
|
for regexp in regexps:
|
|
match = re.match(regexp, message_text)
|
|
|
|
if match:
|
|
return entities.RuleJudgeResult(
|
|
matching=True,
|
|
replacement=message_chain,
|
|
)
|
|
|
|
return entities.RuleJudgeResult(
|
|
matching=False,
|
|
replacement=message_chain
|
|
)
|