From 15dfd75ce84235b07b845fdfb42e269002b92c01 Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Mon, 5 Apr 2021 16:01:22 +0800
Subject: =?UTF-8?q?=E2=9C=A8=F0=9F=90=9B=E2=9A=A1=EF=B8=8F=20=E4=B8=80?=
=?UTF-8?q?=E4=BA=9B=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
新增：转发信息伪造
新增：关键词回复/添加/删除（待更新）
新增：涩图检测（部署方式待更新）
新增：使用方法
新增：ATRI语加密/解密
新增：注入检测
新增：部分命令频率限制
移除：群垃圾检测
优化：提升了部分代码可读性
优化：对 Service 部分代码进行重构
---
ATRI/plugins/key-repo/data_source.py | 116 +++++++++++++++++++++++++++++++++++
1 file changed, 116 insertions(+)
create mode 100644 ATRI/plugins/key-repo/data_source.py
(limited to 'ATRI/plugins/key-repo/data_source.py')
diff --git a/ATRI/plugins/key-repo/data_source.py b/ATRI/plugins/key-repo/data_source.py
new file mode 100644
index 0000000..3dd331d
--- /dev/null
+++ b/ATRI/plugins/key-repo/data_source.py
@@ -0,0 +1,116 @@
+import os
+import json
+from pathlib import Path
+from typing import Optional
+
+
+KEYREPO_DIV = Path('.') / 'ATRI' / 'data' / 'database' / 'KeyRepo'
+os.makedirs(KEYREPO_DIV, exist_ok=True)
+
+
+def load_key_data() -> dict:
+ file_name = "data.json"
+ path = KEYREPO_DIV / file_name
+ try:
+ data = json.loads(path.read_bytes())
+ except:
+ with open(path, 'w') as r:
+ r.write(json.dumps({}))
+ data = {}
+ return data
+
+
+def load_key_temp_data() -> list:
+ file_name = "data.temp.json"
+ path = KEYREPO_DIV / file_name
+ try:
+ data = json.loads(path.read_bytes())
+ except:
+ with open(path, 'w') as r:
+ r.write(json.dumps([]))
+ data = []
+ return data
+
+
+def load_key_history() -> list:
+ file_name = "data.history.json"
+ path = KEYREPO_DIV / file_name
+ try:
+ data = json.loads(path.read_bytes())
+ except:
+ with open(path, 'w') as r:
+ r.write(json.dumps([]))
+ data = []
+ return data
+
+
+def save_key_data(d: dict) -> None:
+ file_name = "data.json"
+ path = KEYREPO_DIV / file_name
+ with open(path, 'w') as r:
+ r.write(json.dumps(d))
+
+
+def save_key_temp_data(d: list) -> None:
+ file_name = "data.temp.json"
+ path = KEYREPO_DIV / file_name
+ with open(path, 'w') as r:
+ r.write(json.dumps(d))
+
+
+def save_key_history_data(d: list) -> None:
+ file_name = "data.history.json"
+ path = KEYREPO_DIV / file_name
+ with open(path, 'w') as r:
+ r.write(json.dumps(d))
+
+
+def add_key(key: str, repo: str) -> str:
+ data = load_key_data()
+ data_1 = data.get(key, [])
+ if repo in data_1:
+ return "该回复已存在~！"
+ data_1.append(repo)
+ data[key] = data_1
+ save_key_data(data)
+ return "成功，又学到新知识了~！"
+
+
+def add_key_temp(d: dict) -> None:
+ data: list = load_key_temp_data()
+ data.append(d)
+ save_key_temp_data(data)
+ add_history(d, False)
+
+
+def add_history(d: dict, p: bool = True) -> None:
+ d['pass'] = p
+ data: list = load_key_history()
+ data.append(d)
+ save_key_history_data(data)
+
+
+def del_key(key: str, repo: str) -> str:
+ data = load_key_data()
+ if repo == 'isALL':
+ del data[key]
+ msg = f"成功删除关键词[{key}]下所有回复..."
+ else:
+ data_1 = data.get(key, [])
+ try:
+ data_1.remove(key)
+ except KeyError:
+ raise KeyError('Find repo error.')
+ data[key] = data_1
+ msg = f"成功删除关键词[{key}]下回复：{repo}"
+ save_key_data(data)
+ return msg
+
+
+def del_key_temp(d: dict) -> bool:
+ data = load_key_temp_data()
+ if d in data:
+ data.remove(d)
+ return True
+ else:
+ return False
--
cgit v1.2.3