笔者在今天的工作中，遇到了一个需求，那就是如何将python字符串生成pdf。比如，需要把python字符串‘这是测试文件'生成为pdf, 该pdf中含有文字‘这是测试文件'。
经过一番检索，笔者决定采用wkhtmltopdf这个软件，它可以将html转化为pdf。wkhtmltopdf的访问网址为：https://wkhtmltopdf.org/downloads.html ，读者可根据自己的系统下载对应的文件并安装。安装好wkhtmltopdf，我们再安装这个软件的python第三方模块——pdfkit，安装方式如下：
pip install pdfkit
我们再讨论如下问题：
•如何将python字符串生成pdf；
•如何生成pdf中的表格；
•解决pdf生成速度慢的问题。
如何将python字符串生成pdf
该问题的解决思路还是利用将python字符串嵌入到html代码中解决，注意换行需要用<br>标签，示例代码如下：
|
1
2
3
4
5
6
7
|
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# 转换为pdfpdfkit.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如下：
如何生成pdf中的表格
接下来我们考虑如何将csv文件转换为pdf中的表格，思路还是利用html代码。示例的iris.csv文件（部分）如下：
将csv文件转换为pdf中的表格的python代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
import pdfkit# 读取csv文件with open('iris.csv', 'r') as f: lines = [_.strip() for _ in f.readlines()]# 转化为html中的表格样式td_width = 100content = '<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 += trcontent += '</table>'html = '<html><head><meta charset="utf-8"></head>' \ '<body><div align="center">%s</div></body></html>' % content# 转换为pdfpdfkit.from_string(html, './iris.pdf') |
生成的pdf文件为iris.pdf，部分内容如下：
解决pdf生成速度慢的问题
用pdfkit生成pdf文件虽然方便，但有一个比较大的缺点，那就是生成pdf的速度比较慢，这里我们可以做个简单的测试，比如生成100份pdf文件，里面的文字为“这是第*份测试文件！”。python代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
|
import pdfkitimport timestart_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代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import pdfkitimport timefrom concurrent.futures import threadpoolexecutor, wait, all_completedstart_time = time.time()# 函数: 生成pdfdef 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))# 利用多线程生成pdfexecutor = 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的相关知识，希望对大家有所帮助，如果大家有任何疑问欢迎给我留言，小编会及时回复大家的！
原文链接：http://www.cnblogs.com/jclian91/p/10880822.html
