### 使用Python将HTML文件转换为PDF 在Web开发和报告任务中，将HTML内容转换为PDF是一种常见需求。Python作为一个庞大的生态系统，提供了多种库来完成此操作。其中一种常用的方式是使用`pdfkit`库，它是`wkhtmltopdf`命令行工具的Python封装。 #### 安装依赖项 为了能够顺利运行代码，需要安装两个主要组件： - `wkhtmltopdf`: 这是一个开源的命令行工具，用于将网页渲染成PDF或图片。 - `pdfkit`: Python包，作为`wkhtmltopdf`的接口[^1]。 可以通过以下命令安装这两个软件包： ```bash pip install pdfkit ``` 对于`wkhtmltopdf`，则需根据操作系统下载并安装相应的版本。 #### 示例实现 下面展示了一个简单的例子，说明如何利用上述提到的技术栈把HTML字符串或者URL地址保存为PDF文档。 ```python import pdfkit # 将在线页面转存为PDF url = 'http://example.com' output_pdf_path = './example.pdf' options = { 'page-size': 'A4', 'margin-top': '0mm', 'margin-right': '0mm', 'margin-bottom': '0mm', 'margin-left': '0mm', } pdfkit.from_url(url, output_pdf_path, options=options) print(f'Successfully converted {url} to PDF and saved at {output_pdf_path}') ``` 这段脚本会访问指定网址并将整个页面的内容按照设定好的纸张大小(A4)及边距参数导出至本地磁盘上的目标路径下。 如果要处理的是本地存储的HTML文件而不是网络资源，则可采用如下方式读取文件内容再进行转换: ```python with open('input.html', encoding='utf8') as f: html_content = f.read() css_file = "style.css" config = pdfkit.configuration(wkhtmltopdf='/usr/local/bin/wkhtmltopdf') pdfkit.from_string(html_content, 'out.pdf', css=css_file, configuration=config) ``` 这里还展示了如何应用外部CSS样式表给最终生成的PDF添加美观效果。 #### 单元测试编写 针对以上功能模块编写的简单单元测试可以帮助验证其正确性和稳定性。以下是基于unittest框架的一个基本示例： ```python import unittest from pathlib import Path class TestHtmlToPdfConversion(unittest.TestCase): def test_conversion_from_html(self): input_html = '<h1>Hello World</h1>' output_pdf = str(Path(__file__).parent / 'test_output.pdf') try: pdfkit.from_string(input_html, output_pdf) self.assertTrue(Path(output_pdf).exists(), msg="The generated PDF file does not exist.") finally: if Path(output_pdf).exists(): Path(output_pdf).unlink() # 清理临时创建的文件 if __name__ == '__main__': unittest.main() ``` 通过这种方式可以在项目集成过程中持续监控HTML到PDF转换逻辑的质量状况。