[Update]

1 files changed, 53 insertions, 0 deletions

diff --git a/ATRI/plugins/plugin_anime/body.py b/ATRI/plugins/plugin_anime/body.py
new file mode 100644
index 0000000..02c57bb
--- /dev/null
+++ b/ATRI/plugins/plugin_anime/body.py
@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import json
+
+from utils.utils_error import errorRepo
+from utils.utils_request import request_get
+
+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
+        return request_get(self.api, self.params)
+
+def resultRepo(user: str, key: str, img_url: str):
+    try:
+        task = SauceNAO(key)
+        data = task.search(img_url)
+    except Exception:
+        return errorRepo('请求数据失败')
+
+    data = json.loads(data)['results'][0]
+    msg0 = ''
+    print(data)
+
+    msg0 += f'[CQ:at,qq={user}]\n'
+    msg0 += f"SauceNAO INFO:\n"
+    msg0 += f"[CQ:image,file={data['header'].get('thumbnail', None)}]\n"
+    msg0 += f"Like：{data['header'].get('similarity', 0)}%\n"
+    msg0 += f"Title：{data['data'].get('title', None)}\n"
+    msg0 += f"Pixiv ID：{data['data'].get('pixiv_id', None)}\n"
+    msg0 += f"Author：{data['data'].get('member_name', None)}\n"
+    msg0 += f"Autoor ID：{data['data'].get('member_id', None)}\n"
+    msg0 += f"Pixiv URL: https://www.pixiv.net/artworks/{data['data'].get('pixiv_id', None)}\n"
+    msg0 += f"Pic URL: https://pixiv.cat/{data['data'].get('pixiv_id', None)}.jpg"
+
+    if float(data['header'].get('similarity', 0)) < 65:
+        msg0 += '注：相似率小于65%不一定正确'
+    
+    return msg0
\ No newline at end of file