suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-07 04:13:00 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix crash on render error

Browse Source
This commit is contained in:
felinae98 2021-11-01 13:52:39 +08:00
parent 3bd66c5f36
commit dca4c82cae
No known key found for this signature in database
GPG Key ID: 00C8B010587FF610
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/plugins/nonebot_hk_reporter/platform/arknights.py
View File

@ -41,6 +41,7 @@ class Arknights(NewMessage, NoTargetMixin):
async def parse(self, raw_post: RawPost) -> Post: async def parse(self, raw_post: RawPost) -> Post:
announce_url = raw_post['webUrl'] announce_url = raw_post['webUrl']
text = ''
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
raw_html = await client.get(announce_url) raw_html = await client.get(announce_url)
soup = bs(raw_html, 'html.parser') soup = bs(raw_html, 'html.parser')
@ -50,12 +51,14 @@ class Arknights(NewMessage, NoTargetMixin):
render = Render() render = Render()
viewport = {'width': 320, 'height': 6400, 'deviceScaleFactor': 3} viewport = {'width': 320, 'height': 6400, 'deviceScaleFactor': 3}
pic_data = await render.render(announce_url, viewport=viewport, target='div.main') pic_data = await render.render(announce_url, viewport=viewport, target='div.main')
pics.append(pic_data) if pic_data:
pics.append(pic_data)
text = '图片渲染失败'
elif (pic := soup.find('img', class_='banner-image')): elif (pic := soup.find('img', class_='banner-image')):
pics.append(pic['src']) pics.append(pic['src'])
else: else:
raise CategoryNotSupport() raise CategoryNotSupport()
return Post('arknights', text='', url='', target_name="明日方舟游戏内公告", pics=pics, compress=True, override_use_pic=False) return Post('arknights', text=text, url='', target_name="明日方舟游戏内公告", pics=pics, compress=True, override_use_pic=False)
class AkVersion(NoTargetMixin, StatusChange): class AkVersion(NoTargetMixin, StatusChange):