Selenium + Python 判断元素是否存在

1、判断元素是否存在# 定义元素是否存在函数，若不存在返回False，若存在返回True。（查找的ID=element）

2、调用定义好的函数

3、用例完整脚本如下：

import unittestfrom selenium 
import webdriverfrom selenium.common.exceptions 
import NoSuchElementException

class MyTestCase(unittest.TestCase):
	diver = webdriver.Firefox()    
	diver.get("https://www.baidu.com"
	def isElementExistence(self, element):
		try:
			element = self.diver.find_element_by_id(element)
		except NoSuchElementException as E:
			print("此路不通，换一条路")
			return False
		else:
			print("继续走")
			return True
	def test_isElementPresent1(self):
		query = self.isElementExistence("su0")        
    if query is True:            
    	print(u"所查找的元素存在于页面上")        
    else:            
    	print(u"页面未找到所需要的页面元素")

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

脚本执行结果：
.----------------------------------------------------------------------
Ran 1 test in 0.011s

OK
此路不通，换一条路
页面未找到所需要的页面元素