检查页面元素是否存在

# -*- coding:utf-8 -*-

"""
检查元素是否存在
isElementPresent():检查页面的元素上是否存在
"""

from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.common.by import By

class IsElement(object):
    def __init__(self):
        self.driver = webdriver.Firefox()
        
    def isElementPresent(self, how, what):
        try:
            self.driver.find_element(how, what)
        except NoSuchElementException as e:
            return False        
        return True
    
    def testisElementPresent(self):
        self.driver.get('http://www.baidu.com')
        self.driver.find_element_by_id('kw').send_keys('selenium')
        
        if self.isElementPresent(By.ID, 'su'):
            self.driver.find_element(By.ID, 'su').click()
        else:
            raise '元素不存在'
        
if __name__ == '__main__':
    test = IsElement()
    test.testisElementPresent()