Fix potential function loop
This commit is contained in:
parent
9bac902066
commit
388e6c3a74
2 changed files with 14 additions and 5 deletions
|
@ -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:
|
||||
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())
|
||||
|
|
|
@ -52,7 +52,10 @@ async def generate_screenshot_v2(page_link, section=None, allow_special_page=Fal
|
|||
request_private_ip=True
|
||||
)
|
||||
except aiohttp.ClientConnectorError:
|
||||
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:
|
||||
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
|
||||
|
|
Reference in a new issue