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

Update image_table.py

Browse source
This commit is contained in:
wyapx 2023-06-03 01:50:38 +08:00 committed by GitHub
parent 0c6fd1c00f
commit fcd254a4e5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
core/utils/image_table.py
View file

@ -28,6 +28,8 @@ async def image_table_render(table: Union[ImageTable, List[ImageTable]], save_so
Logger.warn('[Webrender] Webrender is not configured.') Logger.warn('[Webrender] Webrender is not configured.')
return False return False
use_local = False use_local = False
pic = False
try: try:
tblst = [] tblst = []
if isinstance(table, ImageTable): if isinstance(table, ImageTable):
@ -64,23 +66,31 @@ async def image_table_render(table: Union[ImageTable, List[ImageTable]], save_so
with open(fname, 'w', encoding='utf-8') as fi: with open(fname, 'w', encoding='utf-8') as fi:
fi.write(tblst + css) fi.write(tblst + css)
pic = False
try: try:
pic = await download_to_cache(web_render_local if use_local else web_render, method='POST', headers={ pic = await download_to_cache(
'Content-Type': 'application/json', web_render_local if use_local else web_render,
}, post_data=json.dumps(html), request_private_ip=True) method='POST',
post_data=json.dumps(html),
request_private_ip=True,
headers={
'Content-Type': 'application/json',
}
)
except aiohttp.ClientConnectorError: except aiohttp.ClientConnectorError:
if use_local: if use_local:
pic = await download_to_cache(web_render, method='POST', headers={ pic = await download_to_cache(
'Content-Type': 'application/json', web_render,
}, post_data=json.dumps(html), request_private_ip=True) method='POST',
else: post_data=json.dumps(html),
return False request_private_ip=True,
return pic headers={
'Content-Type': 'application/json',
}
)
except Exception: except Exception:
Logger.error(traceback.format_exc()) Logger.exception("error at image_table_render")
return False
return pic
__all__ = ['ImageTable', 'image_table_render'] __all__ = ['ImageTable', 'image_table_render']