summaryrefslogtreecommitdiff
path: root/ATRI/plugins/setu
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins/setu')
-rw-r--r--ATRI/plugins/setu/__init__.py6
-rw-r--r--ATRI/plugins/setu/data_source.py6
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