python selenium判斷元素是否存在的問題

爬蟲的時候經常用到這個，找到了一個比較好用的方法

原文鏈接：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("[e-mail]")
            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