Selenium如何获取cookies保存起来用于下次访问

Selenium如何获取cookies保存起来用于下次访问

一、获取cookie

获取到cookie后，保存到文件中

from selenium import webdriver
import time
import json
from selenium.webdriver.common.by import By

#创建webdriver 对象，指明使用chrome 浏览器驱动
wd = webdriver.Chrome()

wd.implicitly_wait(10)

#调用webdriver 对象的get方法，可以让浏览器打开指定网址
wd.get('https://zhidao.baidu.com/activity/iknowduck/level?actId=47')
input('网页端登录百度账号后，请按回车键')
cookie= wd.get_cookies()
# #将获得cookie 的信息打印
print(cookie)
with open('baiducookies.txt','w') as f:
    # 将cookies保存为json格式
    f.write(json.dumps(wd.get_cookies()))
    f.close()

二、加载cookie

通过读取txt文件，添加到浏览器中

wd.add_cookie(cookie)

完整的读取cookie的流程

from selenium import webdriver
import time
import json
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.options import Options
from chatgpt_demo import chatgpt

#创建webdriver 对象，指明使用chrome 浏览器驱动
wd= webdriver.Chrome()
wd.implicitly_wait(10)

#调用webdriver 对象的get方法，可以让浏览器打开指定网址
wd.get('https://zhidao.baidu.com/activity/iknowduck/level?actId=47')
# 首先清除由于浏览器打开已有的cookies
wd.delete_all_cookies()
time.sleep(10)
#打开cookie文本，使用已保存的cookie登录
with open('baiducookies.txt','r') as f:
    # 使用json读取cookies 注意读取的是文件 所以用load而不是loads
    cookies_list = json.load(f)
    for cookie in cookies_list:
        wd.add_cookie(cookie)
wd.refresh() #刷新页面
wd.refresh() #刷新页面
time.sleep(6)
#获取当前窗口的title
First_handle = wd.current_window_handle
j = 0
while j < 60:
#try:
    for i in range(9):
        n = 1
        #点击第一个问题
        questions = wd.find_elements(By.CSS_SELECTOR, 'div.q-item > div.q-title > span:nth-child(2) ')
        for question in questions:
            print('第{}个问题：'.format(n), question.text)
            n += 1

            #将第一个问题，输入catgpt在线智能回答
            message = chatgpt(question.text)
            print(message)
            #with open("./output/{}.doc".format(prompt), "w") as of:
            #    of.write(message)
            titles = question.text + '_百度知道'
            time.sleep(1)
            #点击“回答”，按钮
            # answer = wd.find_element(By.CSS_SELECTOR,'.goto-anwser-btn')
            # print('点击:',answer.text)
            wd.add_cookie(cookie) #带cookie
            question.click() #点击问题
            time.sleep(3)
            #切换窗口，点击“去回答”按钮时，打开了新的窗口，但WebDriver对象对应的 还是老窗口。这里要跟随跳转
            for handle in wd.window_handles:
                # 先切换到该窗口
                wd.switch_to.window(handle)
                # 得到该窗口的标题栏字符串，判断是不是我们要操作的那个窗口
                if 'titles' in wd.title:
                    # 如果是，那么这时候WebDriver对象就是对应的该该窗口，正好，跳出循环，
                    break
            #print(wd.title)

            #新窗口下，不管是否已有其他回答，都点击“我来答按钮”
            element = wd.find_element(By.CSS_SELECTOR, '#answer-bar')
            element.click()#点击“我来答按钮”
            #输入文本
            #新窗口，切换到第一级iframe框下，正文内容
            wd.switch_to.frame('ueditor_0')
            element = wd.find_element(By.CSS_SELECTOR,'body > p')
            print('正在输入答案')
            element.send_keys(message)
            #先返回到主html,点击提交按钮
            wd.switch_to.default_content()
            time.sleep(50)
            element = wd.find_element(By.CSS_SELECTOR,'div.addons.line > a')
            print(element.text)
            wd.add_cookie(cookie)  # 带cookie
            element.click()
            time.sleep(5)
            #做完一系列操作后关闭school_handle
            wd.close()
            # 切换窗口会第一个窗口
            wd.switch_to.window(First_handle)
#except:
    print("出错{}次,正在重新运行程序。".format(j))
    j += 1
    wd.switch_to.window(First_handle)
    time.sleep(5)
    wd.refresh()  # 刷新页面
else:
    print('出错太多次啦，程序已结束')