summaryrefslogtreecommitdiff
path: root/ATRI/plugins/wife/data_source.py
diff options
context:
space:
mode:
authorLint Action <[email protected]>2021-05-04 06:29:07 +0000
committerLint Action <[email protected]>2021-05-04 06:29:07 +0000
commit8804f1e10d507ba293fafc77a7e93d3a84a5814b (patch)
treee107da61f76dfe64dfb96feb5e961bbc4bbd8d52 /ATRI/plugins/wife/data_source.py
parentea7f48011c34fdaec7e91af7eb373c8174e439e6 (diff)
downloadATRI-8804f1e10d507ba293fafc77a7e93d3a84a5814b.tar.gz
ATRI-8804f1e10d507ba293fafc77a7e93d3a84a5814b.tar.bz2
ATRI-8804f1e10d507ba293fafc77a7e93d3a84a5814b.zip
:rotating_light: 自动进行代码格式化
Diffstat (limited to 'ATRI/plugins/wife/data_source.py')
-rw-r--r--ATRI/plugins/wife/data_source.py51
1 files changed, 24 insertions, 27 deletions
diff --git a/ATRI/plugins/wife/data_source.py b/ATRI/plugins/wife/data_source.py
index f01985a..9665351 100644
--- a/ATRI/plugins/wife/data_source.py
+++ b/ATRI/plugins/wife/data_source.py
@@ -3,8 +3,8 @@ import json
from pathlib import Path
-WIFE_DIR = Path('.') / 'ATRI' / 'data' / 'database' / 'wife'
-MERRY_LIST_PATH = WIFE_DIR / 'merry_list.json'
+WIFE_DIR = Path(".") / "ATRI" / "data" / "database" / "wife"
+MERRY_LIST_PATH = WIFE_DIR / "merry_list.json"
os.makedirs(WIFE_DIR, exist_ok=True)
@@ -14,27 +14,24 @@ class Tsuma:
try:
return json.loads(MERRY_LIST_PATH.read_bytes())
except FileNotFoundError:
- with open(MERRY_LIST_PATH, 'w') as r:
+ with open(MERRY_LIST_PATH, "w") as r:
r.write(json.dumps({}, indent=4))
return dict()
-
+
@staticmethod
def _store_tsuma(data: dict) -> None:
- with open(MERRY_LIST_PATH, 'w') as r:
+ with open(MERRY_LIST_PATH, "w") as r:
r.write(json.dumps(data, indent=4))
-
+
@classmethod
def check_tsuma(cls, user: str):
data = cls._load_tsuma()
if user in data:
- msg = (
- "阿,你已经有老婆惹!"
- f"ta是:{data[user]['lassie']['nickname']}"
- )
+ msg = "阿,你已经有老婆惹!" f"ta是:{data[user]['lassie']['nickname']}"
return msg, True
else:
return "悲——你还没老婆...", False
-
+
@classmethod
def inquire_tsuma(cls, user: str) -> str:
data = cls._load_tsuma()
@@ -42,7 +39,7 @@ class Tsuma:
return f"你的老婆是:{data[user]['lassie']['nickname']} 哦~❤"
else:
return "悲——你还没老婆..."
-
+
@classmethod
def got_tsuma(cls, user: str, d: dict) -> str:
check_repo, if_h = cls.check_tsuma(user) # 防止出现多人同时操作导致 NTR 事件
@@ -51,27 +48,27 @@ class Tsuma:
else:
data = cls._load_tsuma()
data[user] = {
- "nickname": d['nickname'],
- "gender": d['gender'],
+ "nickname": d["nickname"],
+ "gender": d["gender"],
"lassie": {
- "nickname": d['lassie']['nickname'],
- "qq": d['lassie']['qq'],
- "gender": d['lassie']['gender']
- }
+ "nickname": d["lassie"]["nickname"],
+ "qq": d["lassie"]["qq"],
+ "gender": d["lassie"]["gender"],
+ },
}
cls._store_tsuma(data)
-
- data[d['lassie']['qq']] = {
- "nickname": d['lassie']['nickname'],
- "gender": d['lassie']['gender'],
+
+ data[d["lassie"]["qq"]] = {
+ "nickname": d["lassie"]["nickname"],
+ "gender": d["lassie"]["gender"],
"lassie": {
- "nickname": d['nickname'],
+ "nickname": d["nickname"],
"qq": user,
- "gender": d['gender']
- }
+ "gender": d["gender"],
+ },
}
cls._store_tsuma(data)
-
+
msg = (
f"> {d['lassie']['nickname']}({d['lassie']['qq']})\n"
f"恭喜成为 {d['nickname']} 的老婆~⭐"
@@ -83,7 +80,7 @@ class Tsuma:
data = cls._load_tsuma()
if not user in data:
return "悲——你还没老婆。。"
-
+
msg = f"悲——,({data[user]['nickname']})抛弃了({data[user]['lassie']['nickname']})"
del data[user]
cls._store_tsuma(data)