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文件。最后,它将下载的文件重命名并移动到指定的输出目录。这个过程是在后台执行的,不需要用户直接参与。