From 0ab46384acb730e371fce99f48c91f7c8f37d721 Mon Sep 17 00:00:00 2001
From: Kyomotoi 0w0 <kyomotoi@kirakira.local>
Date: Mon, 25 Oct 2021 13:31:08 +0800
Subject: =?UTF-8?q?=F0=9F=90=9B=E2=9C=A8=20=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/setu/__init__.py | 30 +++++++++++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

(limited to 'ATRI/plugins/setu/__init__.py')

diff --git a/ATRI/plugins/setu/__init__.py b/ATRI/plugins/setu/__init__.py
index fb5736a..ffaae8d 100644
--- a/ATRI/plugins/setu/__init__.py
+++ b/ATRI/plugins/setu/__init__.py
@@ -77,6 +77,9 @@ async def _tag_setu(bot: Bot, event: MessageEvent):
     await bot.delete_msg(message_id=event_id)
 
 
+_catcher_max_file_size = 128
+
+
 setu_catcher = Setu().on_message("涩图嗅探", block=False)
 
 
@@ -91,7 +94,7 @@ async def _setu_catcher(bot: Bot, event: MessageEvent):
         hso = list()
         for i in args:
             try:
-                data = await Setu().detecter(i)
+                data = await Setu().detecter(i, _catcher_max_file_size)
             except Exception:
                 return
             if data[1] > 0.7:
@@ -139,7 +142,7 @@ async def _deal_check(bot: Bot, event: MessageEvent, state: T_State):
     if not args:
         await nsfw_checker.reject("请发送图片而不是其他东西！！")
 
-    data = await Setu().detecter(args[0])
+    data = await Setu().detecter(args[0], _catcher_max_file_size)
     hso = data[1]
     if not hso:
         await nsfw_checker.finish("图太小了！不测！")
@@ -147,7 +150,7 @@ async def _deal_check(bot: Bot, event: MessageEvent, state: T_State):
     resu = f"涩值：{'{:.2%}'.format(hso)}\n"
     if hso >= 0.75:
         resu += "hso！不行我要发给别人看"
-        repo = f"涩图来咧！\n{MessageSegment.image(args[0])}\n涩值：{'{:.2%}'.format(hso[0])}"
+        repo = f"涩图来咧！\n{MessageSegment.image(args[0])}\n涩值：{'{:.2%}'.format(hso)}"
         for superuser in BotSelfConfig.superusers:
             await bot.send_private_msg(user_id=superuser, message=repo)
 
@@ -159,6 +162,27 @@ async def _deal_check(bot: Bot, event: MessageEvent, state: T_State):
     await nsfw_checker.finish(resu)
 
 
+catcher_setting = Setu().on_command("嗅探", "涩图检测图片文件大小设置")
+
+@catcher_setting.handle()
+async def _catcher_setting(bot: Bot, event: MessageEvent, state: T_State):
+    msg = str(event.message).strip()
+    if msg:
+        state["catcher_set"] = msg
+
+@catcher_setting.got("catcher_set", "数值呢？（1对应1kb，默认128）")
+async def _deal_setting(bot: Bot, event: MessageEvent, state: T_State):
+    global _catcher_max_file_size
+    msg = state["catcher_set"]
+    try:
+        _catcher_max_file_size = int(msg)
+    except Exception:
+        await catcher_setting.reject("请发送阿拉伯数字～！")
+    
+    repo = f"好诶！涩图检测文件最小值已设为：{_catcher_max_file_size}kb"
+    await catcher_setting.finish(repo)
+
+
 @scheduler.scheduled_job(
     "interval", name="涩批诱捕器", hours=1, misfire_grace_time=60, args=[Bot]
 )
-- 
cgit v1.2.3