|
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
-
-
- import json
- import os
-
- import plugins
- from bridge.context import ContextType
- from bridge.reply import Reply, ReplyType
- from common.log import logger
- from plugins import *
-
-
- @plugins.register(
- name="Keyword",
- desire_priority=900,
- hidden=True,
- desc="关键词匹配过滤",
- version="0.1",
- author="fengyege.top",
- )
- class Keyword(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):
- logger.debug(f"[keyword]不存在配置文件{config_path}")
- conf = {"keyword": {}}
- with open(config_path, "w", encoding="utf-8") as f:
- json.dump(conf, f, indent=4)
- else:
- logger.debug(f"[keyword]加载配置文件{config_path}")
- with open(config_path, "r", encoding="utf-8") as f:
- conf = json.load(f)
-
- self.keyword = conf["keyword"]
-
- logger.info("[keyword] {}".format(self.keyword))
- self.handlers[Event.ON_HANDLE_CONTEXT] = self.on_handle_context
- logger.info("[keyword] inited.")
- except Exception as e:
- logger.warn(
- "[keyword] init failed, ignore or see https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins/keyword ."
- )
- raise e
-
- def on_handle_context(self, e_context: EventContext):
- if e_context["context"].type != ContextType.TEXT:
- return
-
- content = e_context["context"].content.strip()
- logger.debug("[keyword] on_handle_context. content: %s" % content)
- if content in self.keyword:
- logger.debug(f"[keyword] 匹配到关键字【{content}】")
- reply_text = self.keyword[content]
-
- reply = Reply()
- reply.type = ReplyType.TEXT
- reply.content = reply_text
- e_context["reply"] = reply
- e_context.action = EventAction.BREAK_PASS
-
- def get_help_text(self, **kwargs):
- help_text = "关键词过滤"
- return help_text
|