原文链接: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
另一个大佬分享的:
def is_exist_element(elem):
s = driver.find_elements_by_css_selector("xxx")
if len(s) == 0:
print("不存在%s元素" % , elem)
return False
if len(s) == 1:
return True
else:
print ("存在%s个元素分别是%s" % , (len(s), elem))
return False