summaryrefslogtreecommitdiff
path: root/ATRI/plugins/saucenao
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins/saucenao')
-rw-r--r--ATRI/plugins/saucenao/__init__.py8
-rw-r--r--ATRI/plugins/saucenao/data_source.py6
2 files changed, 7 insertions, 7 deletions
diff --git a/ATRI/plugins/saucenao/__init__.py b/ATRI/plugins/saucenao/__init__.py
index 8850c9d..b653a13 100644
--- a/ATRI/plugins/saucenao/__init__.py
+++ b/ATRI/plugins/saucenao/__init__.py
@@ -5,14 +5,18 @@ from nonebot.adapters.onebot.v11.helpers import extract_image_urls, Cooldown
from ATRI import conf
from ATRI.log import log
-from ATRI.service import ServiceTools
+from ATRI.service import Service, ServiceTools
from .data_source import SauceNao
+plugin = Service("以图搜图").document("以图搜图,仅限二刺螈")
+sau = SauceNao()
+
+
_search_flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇会~~"])
-saucenao = SauceNao().on_command("以图搜图", "透过一张图搜索可能的来源")
+saucenao = plugin.on_command("以图搜图", "透过一张图搜索可能的来源")
@saucenao.got("saucenao_img", "图呢?", [Cooldown(5, prompt=_search_flmt_notice)])
diff --git a/ATRI/plugins/saucenao/data_source.py b/ATRI/plugins/saucenao/data_source.py
index eaeafed..004d9de 100644
--- a/ATRI/plugins/saucenao/data_source.py
+++ b/ATRI/plugins/saucenao/data_source.py
@@ -1,7 +1,5 @@
from random import choice
-from ATRI.service import Service
-from ATRI.rule import is_in_service
from ATRI.exceptions import RequestError
from ATRI.utils import request
@@ -9,7 +7,7 @@ from ATRI.utils import request
URL = "https://saucenao.com/search.php"
-class SauceNao(Service):
+class SauceNao:
def __init__(
self,
api_key: str = str(),
@@ -19,8 +17,6 @@ class SauceNao(Service):
db=5,
numres=5,
):
- Service.__init__(self, "以图搜图", "以图搜图,仅限二刺螈", rule=is_in_service("以图搜图"))
-
params = dict()
params["api_key"] = api_key
params["output_type"] = output_type