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

use native base64

Browse Source
This commit is contained in:
felinae98 2021-02-08 12:37:10 +08:00
parent ec89e42236
commit 578712546f
No known key found for this signature in database
GPG Key ID: 00C8B010587FF610
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/plugins/hk_reporter/utils.py
View File

@ -28,7 +28,7 @@ class Render(metaclass=Singleton):
browser = await launch(args=['--no-sandbox'])
self.page = await browser.newPage()
async def text_to_pic(self, text: str) -> bytes:
async def text_to_pic(self, text: str) -> str:
hash_text = sha256(text.encode()).hexdigest()[:20]
lines = text.split('\n')
parsed_lines = list(map(lambda x: '<p>{}</p>'.format(escape(x)), lines))
@ -37,12 +37,12 @@ class Render(metaclass=Singleton):
f.write(html_text)
await self.page.goto('file:///tmp/text-{}.html'.format(hash_text))
div = await self.page.querySelector('div')
return await div.screenshot(type='jpeg')
return await div.screenshot(type='jpeg', encoding='base64')
async def text_to_pic_cqcode(self, text:str) -> str:
data = await self.text_to_pic(text)
logger.debug('file size: {}'.format(len(data)))
code = '[CQ:image,file=base64://{}]'.format(base64.b64encode(data).decode())
code = '[CQ:image,file=base64://{}]'.format(data)
logger.debug(code)
return code