github/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-03-19 21:38:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: move loading config method to base class

Browse Source
This commit is contained in:
zhayujie
2023-07-20 16:08:19 +08:00
parent aae9b64833
commit 9ef8e1be3f
6 changed files with 21 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
plugins/banwords/banwords.py
View File

@@ -24,18 +24,17 @@ class Banwords(Plugin):
def __init__(self):
super().__init__()
try:
curdir = os.path.dirname(__file__)
config_path = os.path.join(curdir, "config.json")
# loading config from global plugin config
# load config
conf = super().load_config()
curdir = os.path.dirname(__file__)
if not conf:
# 配置不存在则写入默认配置
config_path = os.path.join(curdir, "config.json")
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 = super().load_config() or json.load(f)
self.searchr = WordsSearch()
self.action = conf["action"]
banwords_path = os.path.join(curdir, "banwords.txt")