From 0722b04734418cab58a6827d0b1fa5b4b62d4573 Mon Sep 17 00:00:00 2001
From: Yiwen-Chan <yiwen_chan@foxmail.com>
Date: Sat, 29 Aug 2020 10:08:51 +0800
Subject: Add files via upload

---
 ATRI/plugins/SauceNAO.py | 112 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100644 ATRI/plugins/SauceNAO.py

(limited to 'ATRI/plugins')

diff --git a/ATRI/plugins/SauceNAO.py b/ATRI/plugins/SauceNAO.py
new file mode 100644
index 0000000..a80e23b
--- /dev/null
+++ b/ATRI/plugins/SauceNAO.py
@@ -0,0 +1,112 @@
+import requests
+import json
+import re
+from datetime import datetime
+from random import choice
+from nonebot import on_command, CommandSession
+from aiohttp import ClientSession
+
+from ATRI.modules.funcControl import checkSwitch, checkNoob # type: ignore
+
+__plugin_name__ = "saucenao_search"
+
+async def get_bytes(url, headers = None):
+    async with ClientSession() as asyncSession:
+        async with asyncSession.get(url, headers = headers) as response:
+            a = await response.read()
+    return a
+
+def now_time():
+    now_ = datetime.now()
+    hour = now_.hour
+    minute = now_.minute
+    now = hour + minute / 60
+    return now
+
+class SauceNAO:
+
+    def __init__(self, api_key, output_type=2, testmode=0, dbmask=None, dbmaski=32768, db=5, numres=1):
+        api = 'https://saucenao.com/search.php'
+        self.api = api
+        params = dict()
+        params['api_key'] = api_key
+        params['output_type'] = output_type
+        params['testmode'] = testmode
+        params['dbmask'] = dbmask
+        params['dbmaski'] = dbmaski
+        params['db'] = db
+        params['numres'] = numres
+        self.params = params
+
+    def search(self, url):
+        self.params['url'] = url
+        res = requests.get(url=self.api,params=self.params)
+        return res.content
+
+@on_command('SauceNAO', aliases = ['以图识图'], only_to_me = False)
+async def SaucenaoSearch(session: CommandSession):
+    user = session.event.user_id
+    group = session.event.group_id
+    msg = session.current_arg.strip()
+
+    if checkNoob(user, group):
+        if 0 <= now_time() < 5.5:
+            await session.send(
+                choice(
+                    [
+                        'zzzz......',
+                        'zzzzzzzz......',
+                        'zzz...好涩哦..zzz....',
+                        '别...不要..zzz..那..zzz..',
+                        '嘻嘻..zzz..呐~..zzzz..'
+                    ]
+                )
+            )
+        else:
+            if checkSwitch(__plugin_name__):
+                if not msg:
+                    msg = session.get('message', prompt="请发送一张图片")
+
+                await session.send("开始以图识图\n(如搜索时间过长或无反应则为图片格式有问题)")
+
+                p = '\\[CQ\\:image\\,file\\=.*?\\,url\\=(.*?)\\]'
+
+                img = re.findall(p, msg)
+
+                task = SauceNAO(api_key='2cc2772ca550dbacb4c35731a79d341d1a143cb5')
+                data = task.search(url=img)
+                msg0 = ''
+                try:
+                    data = json.loads(data)['results'][0]
+                    url = data['data']['ext_urls'][0]
+                    title = data['data']['title']
+                    pixiv_id = data['data']['pixiv_id']
+                    member_name = data['data']['member_name']
+                    member_id = data['data']['member_id']
+                    similarity = data['header']['similarity']
+                    mini_url = data['header']['thumbnail']
+                    msg0 = f'SauceNAO结果：'
+                    msg0 += f'[CQ:image,file={mini_url}]\n'
+                    msg0 += f'相似度：{similarity}%\n'
+                    msg0 += f'标题：{title}\n'
+                    msg0 += f'插画ID：{pixiv_id}\n'
+                    msg0 += f'画师：{member_name}\n'
+                    msg0 += f'画师ID：{member_id}\n'
+                    msg0 += f'直链：https://pixiv.cat/{pixiv_id}.jpg'
+                except:
+                    print('网络请求失败...')
+                if msg0:
+                    if float(similarity) > 70:
+                        await session.send(msg0)
+                    else:
+                        await session.send("找不到相似的图呢...")
+                else:
+                        await session.send("搜索似乎失败了呢...")
+            else:
+                session.finish('该功能已关闭...')
+
+
+@SaucenaoSearch.args_parser
+async def _(session: CommandSession):
+    if not session.is_first_run and session.current_arg.startswith('算了'):
+        session.switch(session.current_arg[len('算了'):])
-- 
cgit v1.2.3