selenium复用浏览器及cookie登录

import time
import yaml
from selenium import webdriver


class TestDemo():
    def setup_method(self,method):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(10)
    def teardown_methed(self,method):
        self.driver.quit()
    def test_demo(self):
        self.driver.get("https://www.baidu.com")
        print("jjj")
        self.driver.find_element_by_id("kw")
        self.driver.find_element_by_id("kw").send_keys("测试人生多好啊")
        self.driver.find_element_by_id("su").click()
    def test_work(self):
        #调用chromeoptions方法
        opt = webdriver.ChromeOptions()
        #设置复用浏览器的地址
        opt.debugger_address = "127.0.0.1:9222"
        driver = webdriver.Chrome(options=opt)
        driver.implicitly_wait(10)
        driver.get("https://work.weixin.qq.com/wework_admin/frame")
        driver.find_element_by_id("menu_contacts").click()
        time.sleep(2)
    #获取cookie
    def test_get_cookie(self):
        # 调用chromeoptions方法
        opt = webdriver.ChromeOptions()
        # 设置复用浏览器的地址
        opt.debugger_address = "127.0.0.1:9222"
        driver = webdriver.Chrome(options=opt)
        driver.implicitly_wait(10)
        # print(driver.get_cookies())
        cookies = driver.get_cookies()
        with open("data.yaml","w",encoding="utf-8") as f:
            #快捷键导包，alt+enter
            #序列化cookie，存入yaml文件
            yaml.dump(cookies,f)
    # 从yaml里读取cookie登录
    def test_login(self):
        driver=webdriver.Chrome()
        driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome")
        #读取cookies
        with open("data.yaml","r",encoding="utf-8") as f:
            cookies = yaml.safe_load(f)
        for cookie in cookies:
            #把cookie传给driver
            driver.add_cookie(cookie)
        driver.get("https://work.weixin.qq.com/wework_admin/frame")