打印屏幕循环

Question

需要一些帮助。我得到了这个python代码来加载url并进行屏幕打印。

我需要实现这一目标：

1. 从数组中读取URL，而不是数组
2. 对已加载的每个URL进行屏幕打印并保存。屏幕上的代码被覆盖了

from selenium import webdriver
from time import sleep

driver = webdriver.Firefox()
url = ["http://google.com", "http://cisco.com"]
for element in url:
    driver.get(element)
    driver.get_screenshot_as_file("screenshot.png")
sleep(2)
driver.quit()
print("end...")

Answer 1

将URL存储在文本文件中，然后逐行读取。然后使用文件名和URL的主机名截屏。

我已经修改了您的代码，可以将每个url的屏幕快照存储在单独的文件中。我使用过Python 3.6.9。

目录结构：

.
├── links.txt
├── requirements.txt
└── screenshots_of_links.py

links.txt：

http://google.com
http://cisco.com

requirements.txt：

selenium==3.141.0
urllib3==1.25.10

screenshots_of_links.py：

from selenium import webdriver
from urllib.parse import urlparse
from time import sleep


driver = webdriver.Firefox()

with open("links.txt") as url_file:
    for line in url_file.readlines():
        url = line.strip()
        if url != "":        
            driver.get(url)
            host = urlparse(url).hostname
            driver.get_screenshot_as_file("{}.png".format(host))            
            sleep(2)

driver.quit()
print("end...")

输出：

修改详细信息：

• 从links.txt文本文件中读取URL。
• 修剪文件的每一行。
• 解析每个URL，并将主机名用作屏幕快照的文件名。 urlparse(url).hostname返回有效URL的主机名。

参考：

• URL parsing Python documentation