diff options
Diffstat (limited to 'ATRI')
-rw-r--r-- | ATRI/plugins/saucenao/data_source.py | 10 |
1 files changed, 9 insertions, 1 deletions
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 |