基于Python实现一个网页转PDF工具

pip install pdfkit

import pdfkit
import argparse
import os

def parse_args():
    """解析命令行参数"""
    parser = argparse.ArgumentParser(description='Python网页转PDF工具')
    # 单个URL参数
    parser.add_argument('-u', '--url', type=str, help='单个网页URL')
    # 批量URL文件参数
    parser.add_argument('-f', '--file', type=str, help='包含多个URL的文本文件路径（每行一个URL）')
    # 保存路径参数
    parser.add_argument('-o', '--output', type=str, default='./output', help='PDF保存目录（默认：./output）')
    # 页面大小参数
    parser.add_argument('-s', '--size', type=str, default='A4', help='PDF页面大小（默认：A4，可选：Letter、Legal等）')
    # 页面方向参数
    parser.add_argument('-O', '--orientation', type=str, default='Portrait', help='页面方向（默认：Portrait纵向，可选：Landscape横向）')
    args = parser.parse_args()
    return args

def validate_args(args):
    """验证命令行参数合法性"""
    if not args.url and not args.file:
        raise ValueError('必须指定单个URL（-u/--url）或URL文件（-f/--file）')
    # 检查保存目录是否存在，不存在则创建
    if not os.path.exists(args.output):
        os.makedirs(args.output)
    # 检查页面大小和方向是否合法
    valid_sizes = ['A4', 'Letter', 'Legal', 'A3', 'A5']
    valid_orientations = ['Portrait', 'Landscape']
    if args.size not in valid_sizes:
        raise ValueError(f'无效的页面大小，可选值：{valid_sizes}')
    if args.orientation not in valid_orientations:
        raise ValueError(f'无效的页面方向，可选值：{valid_orientations}')

def url_to_pdf(url, output_path, page_size, orientation):
    """将单个URL转换为PDF"""
    # 生成PDF文件名（从URL中提取关键信息）
    pdf_name = url.replace('http://', '').replace('https://', '').replace('/', '_').replace(':', '_') + '.pdf'
    pdf_path = os.path.join(output_path, pdf_name)
    
    # 配置pdfkit选项
    options = {
        'page-size': page_size,
        'orientation': orientation,
        'margin-top': '0.75in',
        'margin-right': '0.75in',
        'margin-bottom': '0.75in',
        'margin-left': '0.75in',
        'encoding': "UTF-8",
        'no-outline': None
    }
    
    try:
        print(f'正在转换：{url}')
        pdfkit.from_url(url, pdf_path, options=options)
        print(f'转换成功，保存路径：{pdf_path}')
    except Exception as e:
        raise RuntimeError(f'转换URL {url} 失败：{str(e)}')

def batch_url_to_pdf(file_path, output_path, page_size, orientation):
    """从文本文件批量转换URL为PDF"""
    if not os.path.exists(file_path):
        raise FileNotFoundError(f'URL文件不存在：{file_path}')
    
    with open(file_path, 'r', encoding='utf-8') as f:
        urls = [line.strip() for line in f if line.strip()]
    
    if not urls:
        raise ValueError('URL文件中无有效URL')
    
    print(f'共读取到 {len(urls)} 个URL，开始批量转换...')
    for idx, url in enumerate(urls, 1):
        print(f'[{idx}/{len(urls)}]', end=' ')
        try:
            url_to_pdf(url, output_path, page_size, orientation)
        except Exception as e:
            print(f'转换失败：{str(e)}')

def main():
    try:
        # 解析并验证参数
        args = parse_args()
        validate_args(args)
        
        # 执行转换逻辑
        if args.url:
            url_to_pdf(args.url, args.output, args.size, args.orientation)
        if args.file:
            batch_url_to_pdf(args.file, args.output, args.size, args.orientation)
    except Exception as e:
        print(f'工具执行失败：{str(e)}')

if __name__ == '__main__':
    main()

python web_to_pdf.py [参数]

python web_to_pdf.py -u https://www.baidu.com

python web_to_pdf.py -u https://www.csdn.net -s Letter -O Landscape -o D:/pdf

https://www.baidu.com
https://www.csdn.net
https://www.github.com

python web_to_pdf.py -f urls.txt -o ./batch_output

python web_to_pdf.py -h

usage: web_to_pdf.py [-h] [-u URL] [-f FILE] [-o OUTPUT] [-s SIZE] [-O ORIENTATION]

Python网页转PDF工具

optional arguments:
  -h, --help            show this help message and exit
  -u URL, --url URL     单个网页URL
  -f FILE, --file FILE  包含多个URL的文本文件路径（每行一个URL）
  -o OUTPUT, --output OUTPUT
                        PDF保存目录（默认：./output）
  -s SIZE, --size SIZE  PDF页面大小（默认：A4，可选：Letter、Legal等）
  -O ORIENTATION, --orientation ORIENTATION
                        页面方向（默认：Portrait纵向，可选：Landscape横向）