mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 19:57:04 +08:00
20 lines
485 B
Python
20 lines
485 B
Python
# 敏感词过滤模块
|
|
import re
|
|
|
|
|
|
class ReplyFilter:
|
|
|
|
sensitive_words = []
|
|
|
|
def __init__(self, sensitive_words: list):
|
|
self.sensitive_words = sensitive_words
|
|
|
|
def process(self, message: str) -> str:
|
|
for word in self.sensitive_words:
|
|
match = re.findall(word, message)
|
|
if len(match) > 0:
|
|
for i in range(len(match)):
|
|
message = message.replace(match[i], "*" * len(match[i]))
|
|
|
|
return message
|