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"))
    )

然而，硒给了我以下错误：

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()

如何截取新加载的网页的截图

1 个答案: