summaryrefslogtreecommitdiff
path: root/ATRI/plugins/BlackList.py
diff options
context:
space:
mode:
[Fix]
Diffstat (limited to 'ATRI/plugins/BlackList.py')
1 files changed, 52 insertions, 0 deletions
diff --git a/ATRI/plugins/BlackList.py b/ATRI/plugins/BlackList.py
new file mode 100644
index 0000000..d1a50d3
--- /dev/null
+++ b/ATRI/plugins/BlackList.py
@@ -0,0 +1,52 @@
+import time
+import json
+from pathlib import Path
+from nonebot import on_command, CommandSession
+
+import config # type: ignore
+
+
+master = config.MASTER()
+
+
+@on_command('add_noobList', aliases = ['屏蔽', '移除'], only_to_me = False)
+async def _(session: CommandSession):
+ if session.event.user_id in master:
+ msg = session.event.raw_message.split(' ', 1)
+ b_type = msg[0]
+ user = msg[1]
+
+ if b_type == '屏蔽':
+ bL = {}
+ bL[f"{user}"] = f"{user}"
+ file = Path('.') / 'ATRI' / 'plugins' / 'noobList' / 'noobList.json'
+ f = open(file, 'w')
+ f.write(json.dumps(bL))
+ f.close()
+ await session.send(f'正义执行!![{user}]已被ATRI屏蔽!')
+
+ elif b_type == '移除':
+ file = Path('.') / 'ATRI' / 'plugins' / 'noobList' / 'noobList.json'
+ with open(file, 'r') as f:
+ bL = json.load(f)
+ bL.pop(f"{user}")
+ f = open(file, 'w')
+ f.write(json.dumps(bL))
+ f.close()
+ await session.send('将[{user}]移出黑名单成功~!')
+
+
+@on_command('look_noobList', aliases = ['查看黑名单'], only_to_me = False)
+async def _(session: CommandSession):
+ start = time.perf_counter()
+ with open('ATRI/plugins/noobList/noobList.json', 'r') as f:
+ data = json.load(f)
+
+ msg = f'被ATRI列入黑名单有以下账号:\n=============\n'
+ for i in data.keys():
+ msg0 = f'{i}\n'
+ msg += msg0
+ end = time.perf_counter()
+ msg0 = f"=============\nDone time: {round(end - start, 3)}s"
+ msg += msg0
+ await session.send(msg) \ No newline at end of file