如何确定我的Selenium脚本是否能够登录？

Question

如何让这个硒脚本知道它是否能够登录？目前我正在使用is_displayed()来查看登录表单是否仍在页面上。有没有更好的办法？

driver.find_element_by_id("uname").send_keys("admin") 
driver.find_element_by_id("pwd").send_keys("admin") 
driver.find_element_by_xpath('//*[@id="login_form"]/div/button').click() 

# below xpath is code for invalid user id or password message 
if driver.find_element_by_xpath('//*[@id="login_form"]/div[1]').is_displayed(): 
    output=open("OutputAdmin","w+") 
    output.write("Login unsuccessful") 
    output.close() 
else: 
    output = open("OutputAdmin", "w+") 
    output.write("Login successful") 
    output.close() 
driver.quit() 

现在，如果用户ID或密码不正确，这是工作正常，但如果他们是正确的它不执行其他代码。

Answer 1

try: 
    LoginFailMessage = driver.find_element_by_xpath('//*[@id="login_form"]/div[1]') 
    if LoginFailMessage.is_displayed(): 
     output=open("OutputAdmin","w+") 
     output.write("Login unsuccessful") 
     output.close() 
except: 
    output = open("OutputAdmin", "w+") 
    output.write("Login successful") 
    output.close() 

我已经解决了这个问题。感谢那些试图解决它的人。