From 53a714a84aea6316cf5f887b469805b934372fb0 Mon Sep 17 00:00:00 2001
From: Kyomotoi <0w0@imki.moe>
Date: Sat, 8 Apr 2023 21:52:51 +0800
Subject: =?UTF-8?q?=F0=9F=9A=91=EF=B8=8F=20=E4=BF=AE=E5=A4=8Dtag=E6=90=9C?=
 =?UTF-8?q?=E5=9B=BE=E6=97=A0=E7=BB=93=E6=9E=9C=E6=97=B6=E7=9A=84=E6=8A=A5?=
 =?UTF-8?q?=E9=94=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/setu/__init__.py    | 3 +++
 ATRI/plugins/setu/data_source.py | 7 ++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/ATRI/plugins/setu/__init__.py b/ATRI/plugins/setu/__init__.py
index 3cc2327..9b0d7a0 100644
--- a/ATRI/plugins/setu/__init__.py
+++ b/ATRI/plugins/setu/__init__.py
@@ -61,6 +61,9 @@ async def _(bot: Bot, event: MessageEvent):
     pattern = r"来[张点丶份](.*?)的?[涩色🐍]图"
     tag = re.findall(pattern, msg)[0]
     setu, setu_data = await Setu.new(tag)
+    if not setu_data.url:
+        await tag_setu.finish("没有合适的涩图呢...")
+
     setu_info = f"Title: {setu_data.title}\nPid: {setu_data.pid}"
     await bot.send(event, setu_info)
 
diff --git a/ATRI/plugins/setu/data_source.py b/ATRI/plugins/setu/data_source.py
index 03f19a4..5f7d36e 100644
--- a/ATRI/plugins/setu/data_source.py
+++ b/ATRI/plugins/setu/data_source.py
@@ -36,7 +36,12 @@ class Setu:
         except Exception:
             raise RequestError("setu: 请求失败")
 
-        data = LoliconResponse.parse_obj(req.json()).data[0]
+        raw_data = LoliconResponse.parse_obj(req.json()).data
+        if not raw_data:
+            return MessageSegment.text(str()), SetuInfo(
+                title=str(), pid=int(), url=str()
+            )
+        data = raw_data[0]
         title = data.title
         pid = data.pid
         url = data.urls.original
-- 
cgit v1.2.3