diff options
author | Kyomotoi <[email protected]> | 2023-04-08 21:52:51 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2023-04-08 21:52:51 +0800 |
commit | 53a714a84aea6316cf5f887b469805b934372fb0 (patch) | |
tree | c3a1faeb0cc9066fe67640d725159ccbbf7096e0 | |
parent | c905eb776d0e6b8c8930faf857ab07f070bdfb49 (diff) | |
download | ATRI-53a714a84aea6316cf5f887b469805b934372fb0.tar.gz ATRI-53a714a84aea6316cf5f887b469805b934372fb0.tar.bz2 ATRI-53a714a84aea6316cf5f887b469805b934372fb0.zip |
🚑️ 修复tag搜图无结果时的报错
-rw-r--r-- | ATRI/plugins/setu/__init__.py | 3 | ||||
-rw-r--r-- | ATRI/plugins/setu/data_source.py | 7 |
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 |