python - 如何对新加载的网页进行截图

Question

我正在使用 selenium 登录网页并在登录后截取屏幕截图。这是我的代码:

import os
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
driver = webdriver.Firefox()
driver.get("some website")
user_name = driver.find_element_by_name("USERNAME")
password = driver.find_element_by_name("PIN")

user_name.clear()
user_name.send_keys("username")

password.clear()
password.send_keys("password)

###Sends enter
password.send_keys(Keys.RETURN)

# Wait for some time for the next page to load
wait = WebDriverWait(driver, 10)

driver.save_screenshot('screenshot.png')
driver.quit()

当我这样做时，我得到的是登录页面的屏幕截图，而不是下一页。

理想情况下，预期输出应该是登录后页面的屏幕截图。我已经通过监控 selenium 打开网页、登录来手动验证，10 秒内。

所以我做了别的事情，我插入了 sleep (10) 而不是 driver.wait，如下所示:

wait = WebDriverWait(driver, 10)                   

替换为

os.sleep(10)

这样做时，我得到了下一页的屏幕截图。

然后我还尝试插入自定义等待条件，并在下一页中搜索元素:

try:
    element = WebDriverWait(driver, 100).until(
        EC.presence_of_element_located((By.ID, "crnum"))
    )

但是， Selenium 给了我以下错误:

TimeoutException                          Traceback (most recent call last)
<ipython-input-39-16f1ee0dd7f3> in <module>()
     15 try:
     16     element = WebDriverWait(driver, 100).until(
---> 17         EC.presence_of_element_located((By.ID, "crnum"))
     18     )
     19 finally:

D:\anaconda\lib\site-packages\selenium\webdriver\support\wait.py in until(self, method, message)
     78             if time.time() > end_time:
     79                 break
---> 80         raise TimeoutException(message, screen, stacktrace)
     81 
     82     def until_not(self, method, message=''):

TimeoutException: Message: 

我检查了第二页，该 id 存在。我相信正在发生的事情是 selenium 仍在尝试搜索加载页面上的元素。

我通过监视其他页面手动验证，它在两秒内打开。

编辑:

Html 代码出现在第二页，我正在按 id 搜索:

<input type="text" id="crnum" name="number" title="Request Number" onfocus="this.className='focusField';" onblur="this.className='';" class="">

Answer 1

要获取下一页(新加载的页面)的屏幕截图，您必须诱导WebDriverWait，以使下一页上的已识别元素可单击，并且您可以使用以下代码块:

password.send_keys(Keys.RETURN)
# Wait for the element with text as Request Number to be clickable
WebDriverWait(driver, 20).until(EC.element_to_be_clickable((By.XPATH, "//input[@id='crnum' and @name='number' and @title='Request Number']")))
driver.save_screenshot('screenshot.png')
driver.quit()

<小时/>

引用

您可以在 How to take screenshot with Selenium WebDriver 中找到详细讨论