爬虫的时候经常用到这个,找到了一个比较好用的方法
原文链接: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