import xlwt
import xlrd
from selenium import webdriver
import time
from selenium.webdriver.support.select import  Select

driver = webdriver.Chrome()
driver.get('http://127.0.0.1/zentao/my/')
driver.get('http://127.0.0.1/zentao/my-profile.html')

#——————一、cookie基础操作————————————
# 输出cookie
# cookies = driver.get_cookies()
# for cookie in cookies:
#     print(cookie)
# #获取cookie值
# cookie_value = driver.get_cookies('theme')
# print(cookie_value)


#——————二、获取cookie的方法————————————
#1、浏览器获取cookie
#2、打开验证码页——>等待30s——>手动输入用户名、密码、验证码、点击提交或登陆 ——>打印所有的cookie

#——————三、cookie登陆操作————————————
# #方法一：代码直接填入
# driver.add_cookie({'domain': '127.0.0.1', 'httpOnly': False, 'name': 'windowHeight', 'path': '/zentao', 'secure': False, 'value': '577'})
# driver.add_cookie({'domain': '127.0.0.1', 'expiry': 1616159946, 'httpOnly': False, 'name': 'theme', 'path': '/zentao/', 'secure': False, 'value': 'default'})
# driver.add_cookie({'domain': '127.0.0.1', 'httpOnly': False, 'name': 'windowHeight', 'path': '/zentao/my', 'secure': False, 'value': '577'})
# driver.add_cookie({'domain': '127.0.0.1', 'expiry': 1616159946, 'httpOnly': True, 'name': 'device', 'path': '/zentao/', 'secure': False, 'value': 'desktop'})
# driver.add_cookie({'domain': '127.0.0.1', 'httpOnly': False, 'name': 'windowWidth', 'path': '/zentao', 'secure': False, 'value': '1034'})
# driver.add_cookie({'domain': '127.0.0.1', 'expiry': 1616159946, 'httpOnly': False, 'name': 'lang', 'path': '/zentao/', 'secure': False, 'value': 'zh-cn'})
# driver.add_cookie({'domain': '127.0.0.1', 'httpOnly': True, 'name': 'zentaosid', 'path': '/zentao/', 'secure': False, 'value': '3df928fecb2bf07d678d4e112fa4ebf5'})
# driver.add_cookie({'domain': '127.0.0.1', 'httpOnly': False, 'name': 'windowWidth', 'path': '/zentao/my', 'secure': False, 'value': '1034'})
#
# time.sleep(2)
# driver.refresh()#刷新

##方法二、excel +Cookie操作
##步骤1__创建文件并保存cookie
# time.sleep(20) #停留20s手动登陆获取cookie
# cookies = driver.get_cookies()
# # 导入xlwt
# workbook = xlwt.Workbook(encoding='utf-8')#创建workbook对象
# workbsheet = workbook.add_sheet('test_cookie_sheet')#创建工作表sheet
# workbsheet.write(0,0,'name')#往表中写内容，第一各参数 行，第二个参数列，第三个参数内容
# workbsheet.write(0,1,'value')
# workbsheet.write(0,2,'path')
# workbsheet.write(0,3,'domain')
# workbsheet.write(0,4,'httpOnly')
# workbsheet.write(0,5,'secure')
# for i in range(1,len(cookies)+1):
#     workbsheet.write(i, 0, cookies[i-1]['name'])  # 往表中写内容，第一各参数 行，第二个参数列，第三个参数内容
#     workbsheet.write(i, 1, cookies[i-1]['value'])
#     workbsheet.write(i, 2, cookies[i-1]['path'])
#     workbsheet.write(i, 3, cookies[i-1]['domain'])
#     workbsheet.write(i, 4, cookies[i-1]['httpOnly'])
#     workbsheet.write(i, 5,cookies[i-1]['secure'] )
# workbook.save('20210217_cookie.xlsx')#保存表为20210217_cookie.xlsx

##步骤2_读取本地excel文件登陆
workbook = xlrd.open_workbook('20210217_cookie.xlsx')
sheet = workbook.sheet_by_name('test_cookie_sheet')
cookie_list = []
for row_num in range(1,sheet.nrows):
    cookie_dict = {}
    cookie_dict['name'] = sheet.cell_value(row_num,0)
    cookie_dict['value'] = sheet.cell_value(row_num, 1)
    cookie_dict['path'] = sheet.cell_value(row_num, 2)
    cookie_dict['domain'] = sheet.cell_value(row_num, 3)
    cookie_dict['httpOnly'] = bool(sheet.cell_value(row_num, 4))
    cookie_dict['secure'] = bool(sheet.cell_value(row_num, 5))
    cookie_list.append(cookie_dict)

for cookie in cookie_list:
    driver.add_cookie(cookie)
time.sleep(3)
driver.refresh()