selenium使用cookie登录企业微信

皮不动

已于 2023-08-31 20:35:23 修改

阅读量316

点赞数

文章标签： python selenium 测试工具企业微信

于 2023-08-31 00:41:10 首次发布

本文链接：https://blog.csdn.net/qq_41904572/article/details/132593951

版权

指定端口启动浏览器

chrome_options = Options()
chrome_options.debugger_address = "127.0.0.1:9223"
"""chrome浏览器"""
# self.driver = webdriver.Chrome(options=chrome_options)

"""edge浏览器"""
self.driver = webdriver.Edge(options=chrome_options)

移除cookie

 cookies.pop("cookie_id")

将cookies存入shelve（想当于小型db）

# 获取cookies
db = shelve.open("cookies")
# 将数据保存到db中
db['cookie'] = self.driver.get_cookies()
# 获取登录后的cookie
cookies = db['cookie']

使用cookie登录

# 获取登录后的cookie
cookies = self.driver.get_cookies()
print(cookies)
for cookie in cookies:
    self.driver.add_cookie(cookie)
self.driver.get("https://work.weixin.qq.com/wework_admin/frame")

完整代码：

import shelve
from time import sleep

from selenium import webdriver

"""chrome.options"""
# from selenium.webdriver.chrome.options import Options

"""edge.options"""
from selenium.webdriver.edge.options import Options


class TestWeChatDemo():

    def setup(self):
        chrome_options = Options()
        chrome_options.debugger_address = "127.0.0.1:9223"
        """chrome浏览器"""
        # self.driver = webdriver.Chrome()
        # self.driver = webdriver.Chrome(options=chrome_options)

        """edge浏览器"""
        self.driver = webdriver.Edge()
        # self.driver = webdriver.Edge(options=chrome_options)
        # self.driver.maximize_window()
        self.driver.implicitly_wait(2)

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

    def test_wechat_demo(self):
        self.driver.get("https://work.weixin.qq.com/wework_admin/frame")
        db = shelve.open("cookies")
        # db['cookie'] = self.driver.get_cookies()  # 可以将数据保存到db中

        # 获取登录后的cookie
        cookies = db['cookie']
        for cookie in cookies:
            # 移除cookie中的元素
            # if "expiry" in cookie.keys():
            #     cookie.pop("expiry")
            self.driver.add_cookie(cookie)
        self.driver.get("https://work.weixin.qq.com/wework_admin/frame")

        sleep(2)
        db.close()