selenium --- Cookie复用

为什么要使用Cookie自动化登录

  • 放在setup里面,可以让每次测试用例执行前,都能复用cookie,这样就不需要重新去登录网站了
  • 复用浏览器,仍然在每次用例开始都需要人为介入
  • 若用例需要经常执行,复用浏览器则不是一个好的选择
  • 大部分cookie的时效性都很长,扫码一次可以使用多次

思路

1、打开浏览器,扫码登陆
2、确保登陆之后,通过get_cookies获取cookie
3、检查本地文件是否已经获取成功
4、再次打开浏览器,通过cookie直接进入主页

常见问题

1、企业微信cookie有互踢机制,在获取cookie成功之后,不要再进行扫码登陆操作
2、获取cookie的时候,即执行代码get_cookies()时,一定要确保已经登陆
3、植入cookie之后需要进入登陆页面,刷新验证是否自动登陆成功

使用cookie登陆

获取cookie

driver.get_cookies

def test_get_cookies(self):
        """
        获取cookie信息,并且存入到本地
        1、访问登录页
        2、等登录成功后再获取cookie信息
        3、将cookie信息存入一个可持久化存储的地方(文件or数据库)
        :return:
        """
        # 访问企业微信登录页
        self.driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome")
        # 等待20秒,人工扫码操作
        sleep(20)
        # 等成功登陆后,获取cookie信息
        cookie = self.driver.get_cookies()
        print(cookie)
        # 将cookie存入一个可持久存储的地方:文件或者数据库
        # 打开文件的时候添加写入权限
        with open("cookie.yaml", "w", encoding="utf8") as f:
            # 第一个参数是要写入的数据
            yaml.safe_dump(cookie, f)

添加cookie

driver.add_cookie(cookie)

 def test_add_cookie(self):
        """
        植入cookie信息,不扫码也能直接登录界面
        1、访问登录页:https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome
        2、从本地保存的cookie信息导出
        3、植入cookie到driver实例对象中
        4、无需扫码也能自动登录成功
        :return:
        """
        # 访问企业微信主页面
        self.driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome")
        # 定义cookie,cookie信息从已经写入的cookie文件中获取
        cookie = yaml.safe_load(open("cookie.yaml"))
        # 植入cookie
        for c in cookie:
            self.driver.add_cookie(c)
        time.sleep(3)
        # 再次访问企业微信页面,发现无需扫码,自动登陆
        self.driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome")