Merge pull request #168 from ChangingSelf/debug

resolve SengokuCola/MaiMBot#167 根据正则表达式过滤消息

src/plugins/chat/bot.py

@@ -1,3 +1,4 @@
+import re
 import time
 from random import random
 
@@ -75,6 +76,14 @@ class ChatBot:
                 logger.info(f"[过滤词识别]消息中含有{word}，filtered")
                 return
 
+        # 正则表达式过滤
+        for pattern in global_config.ban_msgs_regex:
+            if re.search(pattern, message.raw_message):
+                logger.info(
+                    f"[{message.group_name}]{message.user_nickname}:{message.raw_message}")
+                logger.info(f"[正则表达式过滤]消息匹配到{pattern}，filtered")
+                return
+
         current_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(message.time))
 
         # topic=await topic_identifier.identify_topic_llm(message.processed_plain_text)

src/plugins/chat/config.py

@@ -45,6 +45,7 @@ class BotConfig:
     EMOJI_CHECK_PROMPT: str = "符合公序良俗"  # 表情包过滤要求
 
     ban_words = set()
+    ban_msgs_regex = set()
 
     max_response_length: int = 1024  # 最大回复长度
 
@@ -278,6 +279,9 @@ class BotConfig:
                     "response_interested_rate_amplifier", config.response_interested_rate_amplifier
                 )
                 config.down_frequency_rate = msg_config.get("down_frequency_rate", config.down_frequency_rate)
+            
+            if config.INNER_VERSION in SpecifierSet(">=0.0.6"):
+                config.ban_msgs_regex = msg_config.get("ban_msgs_regex", config.ban_msgs_regex)
 
         def memory(parent: dict):
             memory_config = parent["memory"]

template/bot_config_template.toml

@@ -1,5 +1,5 @@
 [inner]
-version = "0.0.5"
+version = "0.0.6"
 
 #如果你想要修改配置文件，请在修改后将version的值进行变更
 #如果新增项目，请在BotConfig类下新增相应的变量
@@ -41,6 +41,13 @@ ban_words = [
     # "403","张三"
     ]
 
+ban_msgs_regex = [
+    # 需要过滤的消息（原始消息）匹配的正则表达式，匹配到的消息将被过滤（支持CQ码），若不了解正则表达式请勿修改
+    #"https?://[^\\s]+", # 匹配https链接
+    #"\\d{4}-\\d{2}-\\d{2}", # 匹配日期
+    # "\\[CQ:at,qq=\\d+\\]" # 匹配@
+]
+
 [emoji]
 check_interval = 120 # 检查表情包的时间间隔
 register_interval = 10 # 注册表情包的时间间隔