用 python 编写自动化测试脚本，我像根据某界面元素是否存在（假设为 button1）来判断执行某流程，如：

button1 = self.driver.find_element_by_name("登录")

if not button1 is None:
# 执行用户未登录时的流程

# 执行用户已登录时的流程

但是这样写代码，运行的时候第一句会报 “an element could not be located by give para.......” 元素定位不到的错误
这个是正常报错，因为当用户已经登录的时候，界面上不会有 “登录” 按钮

我的问题是：有没有什么好方法能实现我的这个目的的？