python生成带有表格的图片

来源:互联网 发布:数据库安全测试方法 编辑:程序博客网 时间:2024/05/16 15:58

因为工作中需要,需要生成一个带表格的图片

例如:

直接在html中写一个table标签,然后单独把表格部分保存成图片

或者是直接将excel中的内容保存成一个图片

刚开始的思路,是直接生成一个带有table标签的html文件,然后将这个文件转成图片,经过查找资料发现需要安装webkit2png,而这个库又依赖其他的东西,遂放弃。

当初的目标是直接生成一个图片,并且是只需要安装python依赖库就行,而不需要在系统层面安装相应的依赖包

后来考虑使用Python的图片处理库Pillow,和生成表格式的库prattytable,下面的图片是最终生成的图片效果

结果图片

总体来说是分为两个步骤:

  • 使用prattytable将要展示的数据生成一个表格字符串
  • 使用pillow,将生成的表格字符串写入到图片中

下面是具体实现:

from prettytable import PrettyTablefrom PIL import Image, ImageDraw, ImageFonttab = PrettyTable()# 设置表头tab.field_names = ["Name", "Age","Country","City"]# 表格内容插入tab.add_row(['chal','23','中国','Shanghai'])tab.add_row(['charle','29','China','Xuzhou'])tab.add_row(['jack','32','United States','Washington'])tab_info = str(tab)space = 5# PIL模块中,确定写入到图片中的文本字体font = ImageFont.truetype('/home/doge/YaHeiConsolas.ttf', 15, encoding='utf-8')# Image模块创建一个图片对象im = Image.new('RGB',(10, 10),(0,0,0,0))# ImageDraw向图片中进行操作,写入文字或者插入线条都可以draw = ImageDraw.Draw(im, "RGB")# 根据插入图片中的文字内容和字体信息,来确定图片的最终大小img_size = draw.multiline_textsize(tab_info, font=font)# 图片初始化的大小为10-10,现在根据图片内容要重新设置图片的大小im_new = im.resize((img_size[0]+space*2, img_size[1]+space*2))del drawdel imdraw = ImageDraw.Draw(im_new, 'RGB')# 批量写入到图片中,这里的multiline_text会自动识别换行符draw.multiline_text((space,space), unicode(tab_info, 'utf-8'), fill=(255,255,255), font=font)im_new.save('12345.PNG', "PNG")del draw

至此就生成了上面的图片。但是还有一点问题,在使用中文时,表格会又一些错列,应该是使用字体的事,因为我没有找到合适的字体,所以这个问题暂时没有解决。

0 0