summaryrefslogtreecommitdiff
path: root/ATRI/plugins/key-repo/data_source.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins/key-repo/data_source.py')
1 files changed, 0 insertions, 116 deletions
diff --git a/ATRI/plugins/key-repo/data_source.py b/ATRI/plugins/key-repo/data_source.py
deleted file mode 100644
index 3dd331d..0000000
--- a/ATRI/plugins/key-repo/data_source.py
+++ /dev/null
@@ -1,116 +0,0 @@
-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