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

use tempfile, test

Browse Source
This commit is contained in:
felinae98 2021-02-26 20:16:16 +08:00
parent b14604fd3a
commit c47b0c4425
No known key found for this signature in database
GPG Key ID: 00C8B010587FF610
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/plugins/nonebot_hk_reporter/utils.py
View File

@ -7,6 +7,7 @@ import base64
from pyppeteer import launch
from html import escape
from hashlib import sha256
from tempfile import NamedTemporaryFile
from .plugin_config import plugin_config
@ -44,14 +45,14 @@ class Render(metaclass=Singleton):
return str(data)
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))
html_text = '<div style="width:17em;padding:1em">{}</div>'.format(''.join(parsed_lines))
with open('/tmp/text-{}.html'.format(hash_text), 'w') as f:
f.write(html_text)
data = await self.render('file:///tmp/text-{}.html'.format(hash_text), target='div')
os.remove('/tmp/text-{}.html'.format(hash_text))
with NamedTemporaryFile('wt', suffix='.html', delete=False) as tmp:
tmp_path = tmp.name
tmp.write(html_text)
data = await self.render('file://{}'.format(tmp_path), target='div')
os.remove(tmp_path)
return data
async def text_to_pic_cqcode(self, text:str) -> str:
@ -68,3 +69,8 @@ async def parse_text(text: str):
else:
return text
async def test():
ren = Render()
res = await ren.text_to_pic('12333333')
logger.debug(res)
nonebot.get_driver().on_startup(test)