From 7d129eb9e9cca5d0a06f7e699b425306fb8ce2f6 Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Sun, 21 Jun 2020 00:29:51 +0800
Subject: (FIX)

---
 AyaBot/plugins/pixiv.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 AyaBot/plugins/pixiv.py

(limited to 'AyaBot/plugins/pixiv.py')

diff --git a/AyaBot/plugins/pixiv.py b/AyaBot/plugins/pixiv.py
new file mode 100644
index 0000000..930c152
--- /dev/null
+++ b/AyaBot/plugins/pixiv.py
@@ -0,0 +1,37 @@
+import json
+import nonebot
+import requests
+from nonebot import on_command, CommandSession
+
+
+@on_command('pixivsearch', aliases=['p站搜索'], only_to_me=False)
+async def pixivsearch(session: CommandSession):
+    gets = session.get('gets', pormpt='请输入需要查询的pid码')
+    try:
+        URL = 'https://api.imjad.cn/pixiv/v1/?type=illust&id=' + gets
+        print(URL)
+        response = requests.request("GET", URL)
+        html = response.text
+        sr = json.loads(html)
+        i = sr["response"][0]["id"]
+        title = sr["response"][0]["title"]
+        width = sr["response"][0]["width"]
+        height = sr["response"][0]["height"]
+        userid = sr["response"][0]["user"]["id"]
+        account = sr["response"][0]["user"]["account"]
+        name = sr["response"][0]["user"]["name"]
+        await session.send(f'搜索结果如下:\nPid:{i}\nTitle:{title}\n宽高:{width},{height}\nTags:{tags}\n作者id:{userid}\n账号名称:{account}\n名称:{name}')
+    except:
+        await session.send('连接似乎失败了呢...请稍后尝试')
+   
+@pixivsearch.args_parser
+async def _(session: CommandSession):
+    stripped_arg = session.current_arg_text.strip()
+    if session.is_first_run:
+        if stripped_arg:
+            session.state['gets'] = stripped_arg
+        return
+    
+    if not stripped_arg:
+        session.pause('请输入需要查询的pid码')
+    session.state[session.current_key] = stripped_arg
\ No newline at end of file
-- 
cgit v1.2.3