diff options
Diffstat (limited to 'ATRI')
| -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 |