# 該方法用來確認元素是否存在,如果存在返回flag=true,否則返回false def isElementExist(self,element): flag=True try: driver.find_element_by_css_selector(element) return flag except: flag=False return flag driver=webdriver.Firefox() driver.maximize_window() driver.get("http://www.baidu.com") driver.find_element_by_link_text(u"請登錄").click()#調用isElementExist方法,判斷元素是否存在 flag=test1.isElementExist(self, "div.p") if flag: driver.find_element_by_id("username").send_keys("11111") driver.find_element_by_id("password").send_keys("222222") driver.find_element_by_id("login").click() else: print("登陸失敗")