diff --git a/pkg/qqbot/filter.py b/pkg/qqbot/filter.py index 7166d8b4..f0efeda9 100644 --- a/pkg/qqbot/filter.py +++ b/pkg/qqbot/filter.py @@ -8,6 +8,7 @@ import logging class ReplyFilter: sensitive_words = [] mask = "*" + mask_word = "" # 默认值( 兼容性考虑 ) baidu_check = False @@ -15,9 +16,10 @@ class ReplyFilter: baidu_secret_key = "" inappropriate_message_tips = "[百度云]请珍惜机器人,当前返回内容不合规" - def __init__(self, sensitive_words: list, mask: str = "*"): + def __init__(self, sensitive_words: list, mask: str = "*", mask_word: str = ""): self.sensitive_words = sensitive_words self.mask = mask + self.mask_word = mask_word import config if hasattr(config, 'baidu_check') and hasattr(config, 'baidu_api_key') and hasattr(config, 'baidu_secret_key'): self.baidu_check = config.baidu_check @@ -38,7 +40,10 @@ class ReplyFilter: match = re.findall(word, message) if len(match) > 0: for i in range(len(match)): - message = message.replace(match[i], self.mask * len(match[i])) + if self.mask_word == "": + message = message.replace(match[i], self.mask * len(match[i])) + else: + message = message.replace(match[i], self.mask_word) # 百度云审核 if self.baidu_check: diff --git a/pkg/qqbot/manager.py b/pkg/qqbot/manager.py index 6535cac3..b9733991 100644 --- a/pkg/qqbot/manager.py +++ b/pkg/qqbot/manager.py @@ -85,7 +85,8 @@ class QQBotManager: sensitive_json = json.load(f) self.reply_filter = pkg.qqbot.filter.ReplyFilter( sensitive_words=sensitive_json['words'], - mask=sensitive_json['mask'] if 'mask' in sensitive_json else '*' + mask=sensitive_json['mask'] if 'mask' in sensitive_json else '*', + mask_word=sensitive_json['mask_word'] if 'mask_word' in sensitive_json else '' ) else: self.reply_filter = pkg.qqbot.filter.ReplyFilter([]) diff --git a/sensitive-template.json b/sensitive-template.json index 11e85353..61d15ff9 100644 --- a/sensitive-template.json +++ b/sensitive-template.json @@ -1,5 +1,7 @@ { + "说明": "mask将替换敏感词中的每一个字,若mask_word值不为空,则将敏感词整个替换为mask_word的值", "mask": "*", + "mask_word": "", "words": [ "习近平", "胡锦涛",