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