mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-02-23 16:29:16 +08:00
66 lines
2.5 KiB
Python
66 lines
2.5 KiB
Python
# encoding:utf-8
|
|
|
|
import json
|
|
import os
|
|
from bridge.context import ContextType
|
|
from bridge.reply import Reply, ReplyType
|
|
import plugins
|
|
from plugins import *
|
|
from common.log import logger
|
|
from .WordsSearch import WordsSearch
|
|
|
|
|
|
@plugins.register(name="Banwords", desc="判断消息中是否有敏感词、决定是否回复。", version="1.0", author="lanvent", desire_priority= 100)
|
|
class Banwords(Plugin):
|
|
def __init__(self):
|
|
super().__init__()
|
|
try:
|
|
curdir=os.path.dirname(__file__)
|
|
config_path=os.path.join(curdir,"config.json")
|
|
conf=None
|
|
if not os.path.exists(config_path):
|
|
conf={"action":"ignore"}
|
|
with open(config_path,"w") as f:
|
|
json.dump(conf,f,indent=4)
|
|
else:
|
|
with open(config_path,"r") as f:
|
|
conf=json.load(f)
|
|
self.searchr = WordsSearch()
|
|
self.action = conf["action"]
|
|
banwords_path = os.path.join(curdir,"banwords.txt")
|
|
with open(banwords_path, 'r', encoding='utf-8') as f:
|
|
words=[]
|
|
for line in f:
|
|
word = line.strip()
|
|
if word:
|
|
words.append(word)
|
|
self.searchr.SetKeywords(words)
|
|
self.handlers[Event.ON_HANDLE_CONTEXT] = self.on_handle_context
|
|
logger.info("[Banwords] inited")
|
|
except Exception as e:
|
|
logger.warn("Banwords init failed: %s, ignore or see https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins/banwords ." % e)
|
|
|
|
|
|
|
|
def on_handle_context(self, e_context: EventContext):
|
|
|
|
if e_context['context'].type not in [ContextType.TEXT,ContextType.IMAGE_CREATE]:
|
|
return
|
|
|
|
content = e_context['context'].content
|
|
logger.debug("[Banwords] on_handle_context. content: %s" % content)
|
|
if self.action == "ignore":
|
|
f = self.searchr.FindFirst(content)
|
|
if f:
|
|
logger.info("Banwords: %s" % f["Keyword"])
|
|
e_context.action = EventAction.BREAK_PASS
|
|
return
|
|
elif self.action == "replace":
|
|
if self.searchr.ContainsAny(content):
|
|
reply = Reply(ReplyType.INFO, "发言中包含敏感词，请重试: \n"+self.searchr.Replace(content))
|
|
e_context['reply'] = reply
|
|
e_context.action = EventAction.BREAK_PASS
|
|
return
|
|
|
|
def get_help_text(self, **kwargs):
|
|
return Banwords.desc |