diff options
Diffstat (limited to 'ATRI/plugins/setu')
-rw-r--r-- | ATRI/plugins/setu/__init__.py | 6 | ||||
-rw-r--r-- | ATRI/plugins/setu/data_source.py | 6 |
2 files changed, 6 insertions, 6 deletions
diff --git a/ATRI/plugins/setu/__init__.py b/ATRI/plugins/setu/__init__.py index 7962af8..1e8af97 100644 --- a/ATRI/plugins/setu/__init__.py +++ b/ATRI/plugins/setu/__init__.py @@ -8,7 +8,7 @@ from nonebot.params import CommandArg, ArgPlainText from nonebot.adapters.onebot.v11 import Bot, MessageEvent, Message, MessageSegment from nonebot.adapters.onebot.v11.helpers import extract_image_urls, Cooldown -from ATRI.config import BotSelfConfig +from ATRI import conf from .data_source import Setu @@ -122,7 +122,7 @@ async def _setu_catcher(bot: Bot, event: MessageEvent): s_repo = f"多张涩图来咧!\n{ss}\n最涩的达到:{'{:.2%}'.format(hso[0])}" await bot.send(event, u_repo) - for superuser in BotSelfConfig.superusers: + for superuser in conf.BotConfig.superusers: await bot.send_private_msg(user_id=superuser, message=s_repo) @@ -143,7 +143,7 @@ async def _deal_check(bot: Bot, event: MessageEvent): if hso >= 0.75: resu += "hso!不行我要发给别人看" repo = f"涩图来咧!\n{MessageSegment.image(args[0])}\n涩值:{'{:.2%}'.format(hso)}" - for superuser in BotSelfConfig.superusers: + for superuser in conf.BotConfig.superusers: await bot.send_private_msg(user_id=superuser, message=repo) elif 0.75 > hso >= 0.5: diff --git a/ATRI/plugins/setu/data_source.py b/ATRI/plugins/setu/data_source.py index 71649fe..faf8668 100644 --- a/ATRI/plugins/setu/data_source.py +++ b/ATRI/plugins/setu/data_source.py @@ -1,10 +1,10 @@ import asyncio from nonebot.adapters.onebot.v11 import Bot, MessageSegment +from ATRI import conf from ATRI.service import Service from ATRI.rule import is_in_service from ATRI.utils import request -from ATRI.config import Setu as ST from .nsfw_checker import detect_image, init_model @@ -20,8 +20,8 @@ class Setu(Service): @staticmethod def _use_proxy(url: str) -> str: - if ST.reverse_proxy: - return url.replace("i.pixiv.cat", ST.reverse_proxy_domain) + if conf.Setu.reverse_proxy: + return url.replace("i.pixiv.cat", conf.Setu.reverse_proxy_domain) else: return url |