selenium页面跳转后成功与否的判断怎么实现

vincejiao

业务：注册一个账户，成功则出现一个提示框<div id="aaaa">，失败则出现另一个提示框<div id="bbbb">
如果我想通过判断跳转后的页面是否包含这两个原来作为跳转成功与否，该怎么实现。。。谢谢

实践过：chk1 = browser.find_element_by_id("aaaa")
            chk2 = browser.find_element_by_id("bbbb")

  if chk1>0:
     print success
if chk2>0:
    print fail

但是每次都会出现aaaa或者bbbb找不到的报错，就是说，当跳转到成功页面则不会出现bbbb；当失败了就不会出现aaaa的元素

求助其他办法来判断这个跳转是否成功，谢谢