mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 03:55:55 +00:00
209f16af76
* style: remove necessary imports * style: fix F841 * style: fix F401 * style: fix F811 * style: fix E402 * style: fix E721 * style: fix E722 * style: fix E722 * style: fix F541 * style: ruff format * style: all passed * style: add ruff in deps * style: more ignores in ruff.toml * style: add pre-commit
31 lines
1018 B
Python
31 lines
1018 B
Python
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('prefix')
|
|
class PrefixRule(rule_model.GroupRespondRule):
|
|
async def match(
|
|
self,
|
|
message_text: str,
|
|
message_chain: platform_message.MessageChain,
|
|
rule_dict: dict,
|
|
query: core_entities.Query,
|
|
) -> entities.RuleJudgeResult:
|
|
prefixes = rule_dict['prefix']
|
|
|
|
for prefix in prefixes:
|
|
if message_text.startswith(prefix):
|
|
# 查找第一个plain元素
|
|
for me in message_chain:
|
|
if isinstance(me, platform_message.Plain):
|
|
me.text = me.text[len(prefix) :]
|
|
|
|
return entities.RuleJudgeResult(
|
|
matching=True,
|
|
replacement=message_chain,
|
|
)
|
|
|
|
return entities.RuleJudgeResult(matching=False, replacement=message_chain)
|