python读取EXCEL表格中第一列中的链接打开并截图
时间: 2023-06-15 10:07:17 浏览: 115
你可以使用 `openpyxl` 库读取excel表格中的链接,使用 `selenium` 库打开链接并截图。
以下是一个示例代码:
```python
from selenium import webdriver
from openpyxl import load_workbook
# 加载excel文件
workbook = load_workbook('example.xlsx')
# 选择第一个工作表
worksheet = workbook.worksheets[0]
# 遍历第一列中的链接
for row in worksheet.iter_rows(min_row=1, max_col=1, max_row=worksheet.max_row):
link = row[0].value
if link.strip().startswith("http"):
# 打开链接
driver = webdriver.Chrome()
driver.get(link)
# 截图并保存
screenshot_path = f"{link.strip('/').split('/')[-1]}.png"
driver.save_screenshot(screenshot_path)
# 关闭浏览器
driver.quit()
```
在上面的代码中,我们首先使用 `load_workbook` 函数加载excel文件,然后选择第一个工作表。接着,我们使用 `iter_rows` 函数遍历第一列中的链接,并检查链接是否以 "http" 开头。如果是,我们就使用 `selenium` 库打开链接,并使用 `save_screenshot` 函数截图并保存。最后,我们关闭浏览器。
阅读全文