首页 > 技术文章 > 【Selenium10】处理对话框

june2014 2014-04-08 07:01 原文

页面上弹出的对话框若是基于iframe,需要进行switch_to_frame操作,若是div形式,则处理更简单

操作有:打开对话框、关闭对话框、操作对话框中的元素

用到的HTML文档
    <html>
        <head>
            <meta http-equiv="content-type" content="text/html;charset=utf-8" />
            <title>modal</title>        
            <script type="text/javascript" async="" src="http://ajax.googleapis.com/ajax/libs/jquery/1.9.1/jquery.min.js"></script>
            <link href="http://netdna.bootstrapcdn.com/twitter-bootstrap/2.3.2/css/bootstrap-combined.min.css" rel="stylesheet" />      
            <script type="text/javascript">
                $(document).ready(function(){
                    $('#click').click(function(){
                        $(this).parent().find('p').text('try watir-webdriver right now!');
                    });
                });
            </script>
        </head>

        <body>
            <h3>modal</h3>
            <div class="row-fluid">
                <div class="span6">     
                    <!-- Button to trigger modal -->
                    <a href="#myModal" role="button" class="btn btn-primary" data-toggle="modal" id="show_modal">Click</a>

                    <!-- Modal -->
                    <div id="myModal" class="modal hide fade" tabindex="-1" role="dialog" aria-labelledby="myModalLabel" aria-hidden="true">
                      <div class="modal-header">
                        <button type="button" class="close" data-dismiss="modal" aria-hidden="true">×</button>
                        <h3 id="myModalLabel">Modal header</h3>
                      </div>
                      <div class="modal-body">
                        <p>watir-webdriver is better than slenium-webdriver</p>
                        <a href="#" id="click">click me</a>
                      </div>
                      <div class="modal-footer">
                        <button class="btn" data-dismiss="modal" aria-hidden="true">Close</button>
                        <button class="btn btn-primary">Save changes</button>
                      </div>
                    </div>
                </div>      
            </div>      
        </body>
        <script src="http://netdna.bootstrapcdn.com/twitter-bootstrap/2.3.2/js/bootstrap.min.js"></script>
    </html>

代码
from selenium import webdriver
from selenium.webdriver.support.ui import ui
import time
import os
dr=webdriver.Firefox()
file_path='file:///'+os.path.abspath('modal.html')
dr.get(file_path)
time.sleep(2)
dr.find_element_by_id('show_modal').click()   #打开对话框
wait=ui.WebDriverWait(dr,10)
wait.until(lambda dr: dr.find_element_by_id('myModal').is_displayed()) #myModal是点击按钮引发的下拉列表
#点击对话框中的链接,由于对话框中的元素被蒙版遮挡,直接点击会报错,因此用js模拟click
link=dr.find_element_by_id('myModal').find_element_by_id('click')
dr.execute_script('$(arguments[0]).click()',link) #对link元素执行js脚本
time.sleep(2)
buttons=dr.find_element_by_class_name('modal-footer').find_elements_by_tag_name('button')
buttons[0].click() #关闭对话框
dr.quit()

推荐阅读