【Python实用技能】建议收藏：自动化实现网页内容转PDF并保存的方法探索（含代码，亲测可用）

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from time import sleep
import os
 
def save_pdf(url, output_path):
    # 初始化Chrome驱动器
    chrome_options = Options()
    chrome_options.add_argument('--headless')  # 无界面模式
    driver = webdriver.Chrome(options=chrome_options)
    
    # 打开网页
    driver.get(url)
    
    # 等待网页加载完成
    WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, 'myElementId')))
    
    # 执行JavaScript脚本，下载PDF
    driver.execute_script("""
        function downloadPDF() {
            var pdfLink = document.querySelector('a[href$=".pdf"]');
            if (pdfLink) {
                window.location.href = pdfLink.getAttribute('href');
            }
        }
        downloadPDF();
    """)
    
    # 等待PDF文件开始下载
    sleep(5)  # 假设下载需要5秒钟
    
    # 获取下载的PDF文件
    pdf_file = [f for f in os.listdir(output_path) if f.endswith('.pdf')][0]
    
    # 移动PDF文件到指定目录
    os.rename(os.path.join(output_path, pdf_file), output_path + 'article.pdf')
    
    # 关闭Chrome驱动器
    driver.quit()
 
# 使用示例
save_pdf('http://example.com', '/path/to/output/directory/')

这段代码使用了Selenium库来自动化浏览器操作。它首先初始化了一个无头Chrome驱动器，然后导航到指定的网页。接着它等待特定的元素加载完成，然后通过注入JavaScript代码来寻找并下载网页上的PDF文件。最后，它将下载的文件重命名并移动到指定的输出目录。这个过程是在后台执行的，不需要用户直接参与。