[Update] Refactor this project

1 files changed, 49 insertions, 0 deletions

diff --git a/ATRI/plugins/setu.py b/ATRI/plugins/setu.py
new file mode 100644
index 0000000..315803b
--- /dev/null
+++ b/ATRI/plugins/setu.py
@@ -0,0 +1,49 @@
+# -*- coding:utf-8 -*-
+import time
+import json
+import nonebot
+from nonebot import on_command, CommandSession
+
+from ATRI.modules import response # type: ignore
+
+
+bot = nonebot.get_bot()
+master = bot.config.SUPERUSERS
+apikey = bot.config.LOLICONAPI # type: ignore
+
+URL = 'https://api.lolicon.app/setu/'
+
+SETU_REPLY = """Title: {title}
+Pid: {pid}
+[CQ:image,file={setu}]
+---------------
+完成时间:{time}s"""
+
+
+@on_command('setu', aliases = ['图来'], only_to_me = False)
+async def _(session: CommandSession):
+    with open(f'ATRI\\plugins\\switch\\switch.json', 'r') as f:
+        data = json.load(f)
+
+    if data["setu"] == 0:
+        start = time.perf_counter()
+        values = {
+            "apikey": apikey,
+            "r18": "0",
+            "num": "1"
+        }
+
+        dc = json.loads(response.request_api_params(URL, values))
+        end = time.perf_counter()
+
+        await session.send(
+            SETU_REPLY.format(
+                title=dc["data"][0]["title"],
+                pid=dc["data"][0]["pid"],
+                setu=dc["data"][0]["url"],
+                time = round(end - start, 3)
+            )
+        )
+
+    else:
+        await session.send('该功能已被禁用...')
\ No newline at end of file