selenium修改cookie

WebDriver操作cookie的方法：cookie以字典形式保存

• get_cookies()     获取所有的cookie
• get_cookie(name)   返回字典的key为"name"的cookie信息
• add_cookie(cookie_dict)   添加字典形式的cookie
• delete_cookie(name)   删除cookie
• delete_alla-cookies()：删除所有cookie信息

最近在写项目的登录功能时，需要把线上环境的token和测试环境的结合使用。所以需要对测试环境的cookie操作。

思路是：用selenium登录测试环境，同时也用进程在后台用headless方式登录线上环境。取到线上的token，再给测试环境用。

代码如下：

1、获取线上环境代码

from multiprocessing import Process
from pageobjects.production_login import Product_Login
from config.cfg import *
import time

class Test_Login:
    '''登录测试环境'''
    def __init__(self,driver):
        self.driver = driver

    def open_browser(self):
        self.driver.maximize_window()
        self.driver.implicitly_wait(10)  # 等待

    def login(self):
        '''登录测试环境，添加线上Token'''
        self.driver.get(test_url)
        self.driver.find_element_by_id("vcc_code").send_keys(vcc[1])
        self.driver.find_element_by_id("username").send_keys(username[1])
        self.driver.find_element_by_id("password").send_keys(password)
        self.driver.find_element_by_id('btn').click()
        self.driver.delete_cookie('access_token')  # 删除测试环境的token

    def add_cookie(self,product):
        '''读取线上token，写入cookie'''
        new_token = product.readtoken()[0]
        self.driver.add_cookie(new_token)

    def login_test(self):
        product = Product_Login()
        lastTime = product.readtoken()[1]   #获取日期
        if (time.strftime("%d")) == lastTime:  #如果日期是今天，则直接取token
            self.login()
            self.add_cookie(product)
        else:
            p = Process(target=product.login_production)   #运行线上环境进程
            p.start()
            self.login()
            p.join()   #等待
            self.add_cookie(product)

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

2、测试环境使用token

import time , unittest,json
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from config.cfg import *



class Product_Login:
    '''登录线上环境，把获取的token写入配置文件'''

    def login_production(self):
        '''selenium+chromeheadless获取线上环境Token'''
        day = time.strftime("%d")

        chrome_options = Options()
        chrome_options.add_argument('--headless')
        driver = webdriver.Chrome(options=chrome_options)
        driver.implicitly_wait(10)  # 等待
        driver.get(production_url)
        driver.find_element_by_id("vcc_code").send_keys(vcc[0])  # 登录
        driver.find_element_by_id("username").send_keys(username[0])
        driver.find_element_by_id("password").send_keys(password)
        driver.find_element_by_id('btn').click()

        manage_cookie = driver.get_cookie('access_token')
        driver.quit()
        manage_cookie = json.dumps(man_cookie)
        self.writetoken(day,man_cookie)

    def writetoken(self, nowTime,manage_cookie):  # 写线上token

        out = open(r'D:\config.txt', 'w')
        out.writelines(man_cookie)  #写入token
        out.writelines("\n")
        out.writelines(nowTime)  #写入日期
        out.writelines("\n")
        out.close()

    def readtoken(self):  # 读token

        out = open(r'D:\config.txt', encoding='utf-8')
        lines = out.readline()  #token
        if (lines.__contains__("\n")):
            lines = lines.split("\n")[0]
            lines = json.loads(lines)
        lines1 = out.readline()  #日期
        if (lines1.__contains__("\n")):
            lines1 = lines1.split("\n")[0]

        out.close()
        return lines,lines1