import xlrd
from QD_project.common.base_page import basepage
class Test_get_Cookies:
def Red_workbook(self):
self.drives = basepage.open_url()
workbook = xlrd.open_workbook('test_cookies.xls') #读取封装好的cookies信息
sheet = workbook.sheet_by_name('sheet1') #读取表头信息
cookies_list = [] #创建列表存放cookies信息
for row_num in range(1,sheet.nrows): #遍历列表行数
cookies_dict = {} #创建字典,遍历数据存放【以下是列数遍历】
cookies_dict['name'] = sheet.cell_value(row_num,0)
cookies_dict['value'] = sheet.cell_value(row_num, 1)
cookies_dict['path'] = sheet.cell_value(row_num, 2)
cookies_dict['domain'] = sheet.cell_value(row_num, 3)
cookies_dict['httpOnly'] = bool(sheet.cell_value(row_num, 4))
cookies_dict['secure'] = bool(sheet.cell_value(row_num, 5))
cookies_list.append(cookies_dict) #key:value数据存放列表
for cookie in cookies_list: #循环遍历存入cookies信息登陆
self.drives.add_cookie(cookie)
basepage.wait(5)
self.drives.refresh()
if __name__ == '__main__':
Test_get_Cookies().Red_workbook()