无法使用python在selenium中获取警报框的截图

Question

我对selenium是新的，每当警报弹出时，我想拍警报框的截图。我编写的代码如下：

import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.common.exceptions import NoAlertPresentException
import time


class SearchXSS(unittest.TestCase):

def setUp(self):
    self.driver = webdriver.Firefox()

def test_search_in_python_org(self):
    driver = self.driver
    driver.get("http://testfire.net/")
    ele = driver.find_element_by_name("txtSearch")
    ele.send_keys("<script>alert(document.cookie)</script>")
    ele.send_keys(Keys.RETURN)
    try:
        time.sleep(1)
        alert = driver.switch_to_alert()
        driver.get_screenshot_as_file('screenshot.png')
        alert.accept()
    except NoAlertPresentException as e:
        print "no alert to accept "
    fo.close()


def tearDown(self):
    self.driver.quit()

if __name__=="__main__":
    unittest.main()

我在拍摄截图时遇到了这个问题。错误信息如下所示

======================================================================================================================================================================================================="，第22行，在test_search_in_python_org driver.get_screenshot_as_file('screenshot.png')文件“/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py”中，第798行，在get_screenshot_as_file png=self.get_screenshot_as_file png=self.get_screenshot_as_png()文件“('ASCII'))
文件”/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py“，第827行，在get_screenshot_as_base64返回self.execute(command.screenshot)['value']文件”/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py“，第236行，在execute self.error_handler.check_response（响应）文件“/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/errorhandler.py”，第191行，在check_response中引发exception_class（消息，屏幕，stacktrace,value['aler'].get('text'))expectiontedalertpresentexception:Alert text:amsessionid=92632533494消息：意外模式对话框（文本:amsessionid=92632533494)stacktrace:at e.js:4745)位于（http://testfire.net/search.aspx？txtsearch=%3cscript%3ealer%28document.cookie%29%3c%2fscript%3e:80)

-----------------------------------------------------------------------------------------------------------------------------------------------------------

失败（错误=1)[在10.4s内完成，退出代码为1]

帮我解决这个问题。

Answer 1

您不能使用警报框进行屏幕截图，您需要首先处理警报表示接受或拒绝然后进行屏幕截图，因为当selenium进行带有警报的屏幕截图时，它总是抛出ExpectiondalertPresentException，因此不可能使用selenium使用alert框进行屏幕截图。因此，您需要如下所示：-

alert = driver.switch_to_alert()
alert.accept()
driver.get_screenshot_as_file('screenshot.png')

如果您想用alert进行屏幕截图，您应该用其他库尝试一些不同的方法，在java中有robot类，可以在这种情况下进行screenshot，但我不确定在python中有什么等效的方法。