From d03636606de4c2f8c5c163f0e797325c7d182346 Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Tue, 13 Apr 2021 23:38:31 +0800
Subject: =?UTF-8?q?=E2=9C=A8=F0=9F=90=9B=20=E5=B0=8F=E6=94=B9=E5=8A=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

新增：添加索取取消
新增：挖坑以图搜图

修复：发言屏蔽报错
修复：群消息记录写入类型错误
---
 ATRI/plugins/saucenao/__init__.py    | 12 ++++++++++++
 ATRI/plugins/saucenao/data_source.py | 29 +++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+)
 create mode 100644 ATRI/plugins/saucenao/__init__.py
 create mode 100644 ATRI/plugins/saucenao/data_source.py

(limited to 'ATRI/plugins/saucenao')

diff --git a/ATRI/plugins/saucenao/__init__.py b/ATRI/plugins/saucenao/__init__.py
new file mode 100644
index 0000000..c36fde1
--- /dev/null
+++ b/ATRI/plugins/saucenao/__init__.py
@@ -0,0 +1,12 @@
+from nonebot.adapters.cqhttp import Bot, MessageEvent
+
+from ATRI.service import Service as sv
+from ATRI.rule import is_in_service
+
+
+__doc__ = """
+以图搜图
+权限组：所有人
+用法：
+  以图搜图 (pic)
+"""
diff --git a/ATRI/plugins/saucenao/data_source.py b/ATRI/plugins/saucenao/data_source.py
new file mode 100644
index 0000000..636c540
--- /dev/null
+++ b/ATRI/plugins/saucenao/data_source.py
@@ -0,0 +1,29 @@
+from ATRI.utils.request import post_bytes
+
+
+URL = "https://saucenao.com/search.php"
+
+
+class SauceNao:
+    def __init__(self,
+                 api_key: str,
+                 output_type=2,
+                 testmode=0,
+                 dbmask=None,
+                 dbmaski=32768,
+                 db=5,
+                 numres=1) -> None:
+        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
+    
+    async def search(self, url: str):
+        self.params['url'] = url
+        res = await post_bytes(url=URL, params=self.params)
+        return res
-- 
cgit v1.2.3