爬虫的时候经常用到这个,找到了一个比较好用的方法

原文链接:http://blog.csdn.net/u012189659/article/details/36391837

背景:selenium+python处于学习阶段,功能实现之后开始整理之前写的代码,突然发现一个功能没有做判断,比如,点击“请登录”,需要判断一下这个弹出框到底弹了没有,这样就需要判断某个元素是否存在,使用is_displayed函数只能用于该元素存在去判断此元素是否出现,而不能判断某个元素是否存在,所以就自己写了个方法

 

 

代码:

复制代码
from selenium import webdriver
import unittest
from selenium.webdriver.common.action_chains import ActionChains
class test1(unittest.TestCase):
    
    def setUp(self):
        self.driver=webdriver.Firefox()
        self.baseurl="http://www.xebest.com"
#        self.driver.maximize_window()
        
        
        
        
        
    def dengLu(self):
        browser=self.driver
        
        browser.get(self.baseurl)
        
        browser.find_element_by_link_text(u"请登录").click()
        #调用isElementExist方法,判断元素是否存在
        flag=test1.isElementExist(self, "div.popup-content")         
        
        if flag:
            
            browser.find_element_by_id("userName").send_keys("w74581@163.com")
            browser.find_element_by_id("password").send_keys("w123456")
            browser.find_element_by_id("imgLogin").click()
            print(browser.switch_to_alert().text)
            browser.switch_to_alert().accept()
            
            
        else:
            print("没有弹框")
        
        
        
        
#   该方法用来确认元素是否存在,如果存在返回flag=true,否则返回false        
    def isElementExist(self,element):
        flag=True
        browser=self.driver
        try:
            browser.find_element_by_css_selector(element)
            return flag
        
        except:
            flag=False
            return flag
复制代码