python-addcookie

1. cookie文件：

{"name":"keyname", "value":"keyvalue"}

 

2. 遇到提示cookie domain不对的问题

解决方法：在添加cookie前，先访问一次网站

 

 

import shelve

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.support.wait import WebDriverWait


# /Applications/Google\ Chrome.app/Contents/MacOS/Google\ Chrome --remotedebugging-port=9222

class TestWX:
    def setup(self):

        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.driver.implicitly_wait(5)
        self.driver.get("https://work.weixin.qq.com/")

        with open("cookies.txt", "r+") as f:
            cookie_str = f.read()
        aa = {cookie_str.split(":")[0]: cookie_str.split(":")[1]}
        for i in aa["cookie"].split(";"):
            cookie = {}
            cookie["name"] = i.strip().split("=")[0]
            cookie["value"] = i.strip().split("=")[1]
            if "expiry" in cookie.keys():
                # dict支持pop的删除函数
                cookie.pop("expiry")
            self.driver.add_cookie(cookie)

    def teardown(self):
        self.driver.quit()

    def test_wx(self):

        self.driver.get("https://work.weixin.qq.com/wework_admin/frame#index")
        self.driver.find_element(By.ID, 'menu_contacts').click()  # 点击通讯录