summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2023-04-08 21:52:51 +0800
committerKyomotoi <[email protected]>2023-04-08 21:52:51 +0800
commit53a714a84aea6316cf5f887b469805b934372fb0 (patch)
treec3a1faeb0cc9066fe67640d725159ccbbf7096e0
parentc905eb776d0e6b8c8930faf857ab07f070bdfb49 (diff)
downloadATRI-53a714a84aea6316cf5f887b469805b934372fb0.tar.gz
ATRI-53a714a84aea6316cf5f887b469805b934372fb0.tar.bz2
ATRI-53a714a84aea6316cf5f887b469805b934372fb0.zip
🚑️ 修复tag搜图无结果时的报错
-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