mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 11:42:44 +08:00
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from __future__ import annotations
|
||
|
||
import mirai
|
||
|
||
from .. import rule as rule_model
|
||
from .. import entities
|
||
from ....core import entities as core_entities
|
||
|
||
|
||
@rule_model.rule_class("at-bot")
|
||
class AtBotRule(rule_model.GroupRespondRule):
|
||
|
||
async def match(
|
||
self,
|
||
message_text: str,
|
||
message_chain: mirai.MessageChain,
|
||
rule_dict: dict,
|
||
query: core_entities.Query
|
||
) -> entities.RuleJudgeResult:
|
||
|
||
if message_chain.has(mirai.At(query.adapter.bot_account_id)) and rule_dict['at']:
|
||
message_chain.remove(mirai.At(query.adapter.bot_account_id))
|
||
|
||
if message_chain.has(mirai.At(query.adapter.bot_account_id)): # 回复消息时会at两次，检查并删除重复的
|
||
message_chain.remove(mirai.At(query.adapter.bot_account_id))
|
||
|
||
return entities.RuleJudgeResult(
|
||
matching=True,
|
||
replacement=message_chain,
|
||
)
|
||
|
||
return entities.RuleJudgeResult(
|
||
matching=False,
|
||
replacement = message_chain
|
||
)
|