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