欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

将Python字符串生成PDF的实例代码详解

程序员文章站 2023-11-30 21:26:22
笔者在今天的工作中,遇到了一个需求,那就是如何将python字符串生成pdf。比如,需要把python字符串‘这是测试文件'生成为pdf, 该pdf中含有文字‘这是测试文件...

笔者在今天的工作中,遇到了一个需求,那就是如何将python字符串生成pdf。比如,需要把python字符串‘这是测试文件'生成为pdf, 该pdf中含有文字‘这是测试文件'。

  经过一番检索,笔者决定采用wkhtmltopdf这个软件,它可以将html转化为pdf。wkhtmltopdf的访问网址为: ,读者可根据自己的系统下载对应的文件并安装。安装好wkhtmltopdf,我们再安装这个软件的python第三方模块——pdfkit,安装方式如下:

pip install pdfkit

  我们再讨论如下问题:

•如何将python字符串生成pdf;
•如何生成pdf中的表格;
•解决pdf生成速度慢的问题。

如何将python字符串生成pdf

  该问题的解决思路还是利用将python字符串嵌入到html代码中解决,注意换行需要用<br>标签,示例代码如下:

import pdfkit
# pdf中包含的文字
content = '这是一个测试文件。' + '<br>' + 'hello from python!'
html = '<html><head><meta charset="utf-8"></head>' \
 '<body><div align="center"><p>%s</p></div></body></html>'%content
# 转换为pdf
pdfkit.from_string(html, './test.pdf')

输出的结果如下:

loading pages (1/6)
 counting pages (2/6)
 resolving links (4/6)
 loading headers and footers (5/6)
 printing pages (6/6)
 done

生成的test.pdf如下:

将Python字符串生成PDF的实例代码详解

如何生成pdf中的表格

  接下来我们考虑如何将csv文件转换为pdf中的表格,思路还是利用html代码。示例的iris.csv文件(部分)如下:

将Python字符串生成PDF的实例代码详解

 将csv文件转换为pdf中的表格的python代码如下:

import pdfkit
# 读取csv文件
with open('iris.csv', 'r') as f:
 lines = [_.strip() for _ in f.readlines()]
# 转化为html中的表格样式
td_width = 100
content = '<table width="%s" border="1" cellspacing="0px" style="border-collapse:collapse">' % (td_width*len(lines[0].split(',')))
for i in range(len(lines)):
 tr = '<tr>'+''.join(['<td width="%d">%s</td>'%(td_width, _) for _ in lines[i].split(',')])+'</tr>'
 content += tr
content += '</table>'
html = '<html><head><meta charset="utf-8"></head>' \
 '<body><div align="center">%s</div></body></html>' % content
# 转换为pdf
pdfkit.from_string(html, './iris.pdf')

  生成的pdf文件为iris.pdf,部分内容如下:

将Python字符串生成PDF的实例代码详解

解决pdf生成速度慢的问题

  用pdfkit生成pdf文件虽然方便,但有一个比较大的缺点,那就是生成pdf的速度比较慢,这里我们可以做个简单的测试,比如生成100份pdf文件,里面的文字为“这是第*份测试文件!”。python代码如下:

import pdfkit
import time
start_time = time.time()
for i in range(100):
 content = '这是第%d份测试文件!'%(i+1)
 html = '<html><head><meta charset="utf-8"></head>' \
  '<body><div align="center">%s</div></body></html>' % content
 # 转换为pdf
 pdfkit.from_string(html, './test/%s.pdf'%(i+1))
end_time = time.time()
print('一共耗时:%s 秒.' %(end_time-start_time))

在这个程序中,生成100份pdf文件一共耗时约192秒。输出结果如下:

......
loading pages (1/6)
counting pages (2/6)                                              
resolving links (4/6)                                                      
loading headers and footers (5/6)                                          
printing pages (6/6)
done                                                                     
一共耗时:191.9226369857788 秒.

  如果想要加快生成的速度,我们可以使用多线程来实现,主要使用concurrent.futures模块,完整的python代码如下:

import pdfkit
import time
from concurrent.futures import threadpoolexecutor, wait, all_completed
start_time = time.time()
# 函数: 生成pdf
def convert_2_pdf(i):
 content = '这是第%d份测试文件!'%(i+1)
 html = '<html><head><meta charset="utf-8"></head>' \
  '<body><div align="center">%s</div></body></html>' % content
 # 转换为pdf
 pdfkit.from_string(html, './test/%s.pdf'%(i+1))
# 利用多线程生成pdf
executor = threadpoolexecutor(max_workers=10) # 可以自己调整max_workers,即线程的个数
# submit()的参数: 第一个为函数, 之后为该函数的传入参数,允许有多个
future_tasks = [executor.submit(convert_2_pdf, i) for i in range(100)]
# 等待所有的线程完成,才进入后续的执行
wait(future_tasks, return_when=all_completed)
end_time = time.time()
print('一共耗时:%s 秒.' %(end_time-start_time))

在这个程序中,生成100份pdf文件一共耗时约41秒,明显快了很多~

总结

以上所述是小编给大家介绍的将python字符串生成pdf的相关知识,希望对大家有所帮助