summaryrefslogtreecommitdiff
path: root/ATRI
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI')
-rw-r--r--ATRI/plugins/setu/__init__.py3
-rw-r--r--ATRI/plugins/setu/data_source.py7
2 files changed, 9 insertions, 1 deletions
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