xtex/akari-bot
Archived
1
0
Fork 0
Code Issues Pull requests Activity

Fix potential function loop

Browse source
This commit is contained in:
yzhh 2023-03-02 01:50:56 +08:00
parent 9bac902066
commit 388e6c3a74
2 changed files with 14 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
core/utils/image_table.py
View file

@ -73,9 +73,12 @@ async def image_table_render(table: Union[ImageTable, List[ImageTable]], save_so
'Content-Type': 'application/json',
}, post_data=json.dumps(html), request_private_ip=True)
except aiohttp.ClientConnectorError:
pic = await download_to_cache(web_render, method='POST', headers={
'Content-Type': 'application/json',
}, post_data=json.dumps(html), request_private_ip=True)
if use_local:
pic = await download_to_cache(web_render, method='POST', headers={
'Content-Type': 'application/json',
}, post_data=json.dumps(html), request_private_ip=True)
else:
return False
return pic
except Exception:
Logger.error(traceback.format_exc())

10
modules/wiki/utils/screenshot_image.py
View file

@ -52,7 +52,10 @@ async def generate_screenshot_v2(page_link, section=None, allow_special_page=Fal
request_private_ip=True
)
except aiohttp.ClientConnectorError:
return await generate_screenshot_v2(page_link, section, allow_special_page, content_mode, use_local=False)
if use_local:
return await generate_screenshot_v2(page_link, section, allow_special_page, content_mode, use_local=False)
else:
return False
except ValueError:
Logger.info('[Webrender] Generation Failed.')
return False
@ -71,7 +74,10 @@ async def generate_screenshot_v2(page_link, section=None, allow_special_page=Fal
request_private_ip=True
)
except aiohttp.ClientConnectorError:
return await generate_screenshot_v2(page_link, section, allow_special_page, content_mode, use_local=False)
if use_local:
return await generate_screenshot_v2(page_link, section, allow_special_page, content_mode, use_local=False)
else:
return False
except ValueError:
Logger.info('[Webrender] Generation Failed.')
return False