From 080df928cb3fa8b23481211c8c1072453e89b774 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Mon, 4 Apr 2022 17:09:31 +0800 Subject: =?UTF-8?q?=F0=9F=9A=B8=20=E5=A2=9E=E5=8A=A0=E5=AF=B9=E9=83=A8?= =?UTF-8?q?=E5=88=86=E6=8F=92=E4=BB=B6=E6=89=80=E9=9C=80key=E7=9A=84?= =?UTF-8?q?=E5=A1=AB=E5=86=99=E6=A3=80=E6=9F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/saucenao/data_source.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'ATRI') diff --git a/ATRI/plugins/saucenao/data_source.py b/ATRI/plugins/saucenao/data_source.py index 738c647..9e2bd75 100644 --- a/ATRI/plugins/saucenao/data_source.py +++ b/ATRI/plugins/saucenao/data_source.py @@ -1,9 +1,11 @@ from random import choice -from ATRI.service import Service +from ATRI.service import Service, ServiceTools from ATRI.rule import is_in_service from ATRI.exceptions import RequestError from ATRI.utils import request +from ATRI.config import SauceNAO as sa +from ATRI.log import logger as log URL = "https://saucenao.com/search.php" @@ -21,6 +23,12 @@ class SaouceNao(Service): ): Service.__init__(self, "以图搜图", "以图搜图，仅限二刺螈", rule=is_in_service("以图搜图")) + if not sa.key: + data = ServiceTools.load_service("以图搜图") + data["enabled"] = False + ServiceTools.save_service(data, "以图搜图") + log.warning("key 未配置，本插件将被禁用") + params = dict() params["api_key"] = api_key params["output_type"] = output_type -- cgit v1.2.3