summaryrefslogtreecommitdiff
path: root/ATRI/plugins/setu/data_source.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins/setu/data_source.py')
-rw-r--r--ATRI/plugins/setu/data_source.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/ATRI/plugins/setu/data_source.py b/ATRI/plugins/setu/data_source.py
index 2c665ae..39f3815 100644
--- a/ATRI/plugins/setu/data_source.py
+++ b/ATRI/plugins/setu/data_source.py
@@ -7,6 +7,7 @@ from nonebot.adapters.cqhttp import MessageSegment
from ATRI.service import Service
from ATRI.rule import is_in_service
from ATRI.utils import request
+from .tf_dealer import detect_image
LOLICON_URL = "https://api.lolicon.app/setu/v2"
@@ -61,6 +62,14 @@ class Setu(Service):
return repo, setu
@staticmethod
+ async def detecter(url: str) -> list:
+ """
+ 涩值检测.
+ """
+ data = await detect_image(url)
+ return data
+
+ @staticmethod
async def scheduler() -> str:
"""
每隔指定时间随机抽取一个群发送涩图.