diff options
Diffstat (limited to 'ATRI/plugins/setu/data_source.py')
-rw-r--r-- | ATRI/plugins/setu/data_source.py | 9 |
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: """ 每隔指定时间随机抽取一个群发送涩图. |