diff options
Diffstat (limited to 'ATRI/plugins')
| -rw-r--r-- | ATRI/plugins/SauceNAO.py | 114 | ||||
| -rw-r--r-- | ATRI/plugins/switch.py | 10 | ||||
| -rw-r--r-- | ATRI/plugins/switch/switch.json | 2 |
3 files changed, 123 insertions, 3 deletions
diff --git a/ATRI/plugins/SauceNAO.py b/ATRI/plugins/SauceNAO.py new file mode 100644 index 0000000..cb32345 --- /dev/null +++ b/ATRI/plugins/SauceNAO.py @@ -0,0 +1,114 @@ +import requests
+import json
+import re
+from datetime import datetime
+from random import choice
+from nonebot import on_command, CommandSession
+from aiohttp import ClientSession
+
+import config # type: ignore
+from ATRI.modules.funcControl import checkSwitch, checkNoob # type: ignore
+
+API_KEY = config.SAUCENAO_KEY()
+__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=API_KEY)
+ 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('算了'):])
diff --git a/ATRI/plugins/switch.py b/ATRI/plugins/switch.py index 8abb408..71b7f62 100644 --- a/ATRI/plugins/switch.py +++ b/ATRI/plugins/switch.py @@ -49,7 +49,10 @@ async def _(session: CommandSession): elif com == 'Kyaru': data["chouYou"] = "on" - + + elif com == 'SauceNAO': + data["saucenao_search"] = "on" + else: pass @@ -84,6 +87,9 @@ async def _(session: CommandSession): elif com == 'Kyaru': data["chouYou"] = "off" + elif com == 'SauceNAO': + data["saucenao_search"] = "off" + else: pass @@ -94,4 +100,4 @@ async def _(session: CommandSession): await session.send('Success！') else: - await session.send('恁哪位？')
\ No newline at end of file + await session.send('恁哪位？') diff --git a/ATRI/plugins/switch/switch.json b/ATRI/plugins/switch/switch.json index 13cdfa0..df65fb7 100644 --- a/ATRI/plugins/switch/switch.json +++ b/ATRI/plugins/switch/switch.json @@ -1 +1 @@ -{"pixiv_seach_img": "on", "pixiv_seach_author": "off", "pixiv_daily_rank": "on", "approve_friend_add": "on", "approve_invite_join_group": "on", "setu": "on", "setu_img": "on", "anime_search": "on", "change_face": "on", "chouYou": "off"}
\ No newline at end of file +{"pixiv_seach_img": "on", "pixiv_seach_author": "off", "pixiv_daily_rank": "on", "approve_friend_add": "on", "approve_invite_join_group": "on", "setu": "on", "setu_img": "on", "anime_search": "on", "change_face": "on", "chouYou": "off", "saucenao_search": "on"} |