Python如何保存cookie

一、使用requests库、保存cookie到文件、使用pickle模块保存

在Python中，保存cookie主要通过使用requests库来实现。requests库是Python中非常流行的一个HTTP库，它提供了非常方便的API来处理HTTP请求和响应。保存cookie最常用的方法是使用requests.Session()对象。这个对象可以在多个请求之间保持cookie，并且可以很方便地将cookie保存到文件中。另外，我们还可以使用pickle模块将cookie对象序列化保存到文件，方便后续使用。

使用requests.Session()保存cookie到文件：首先，我们可以通过创建一个requests.Session()对象来实现cookie的管理。这个对象会自动为我们处理cookie的保存与发送。当我们需要将cookie保存到文件中时，可以通过访问session.cookies属性来获取cookie对象，并将其写入文件。这样，下次请求时可以从文件中读取cookie对象并加载到session中，继续维持会话状态。

import requests
import pickle
创建一个Session对象
session = requests.Session()
进行登录请求，假设登录的URL为'https://example.com/login'
login_url = 'https://example.com/login'
login_data = {
    'username': 'your_username',
    'password': 'your_password'
}
response = session.post(login_url, data=login_data)
检查登录是否成功
if response.ok:
    # 将cookie保存到文件
    with open('cookies.pkl', 'wb') as f:
        pickle.dump(session.cookies, f)

二、从文件加载cookie、使用cookieJar对象

在保存cookie之后，我们可以在需要时从文件中加载cookie。这样可以避免重新登录，继续维持会话状态。

从文件加载cookie：我们可以使用pickle模块从文件中加载cookie对象，并将其加载到requests.Session()中继续使用。这种方式能够非常方便地进行会话的恢复和管理。

import requests
import pickle
创建一个新的Session对象
session = requests.Session()
从文件中加载cookie
with open('cookies.pkl', 'rb') as f:
    cookies = pickle.load(f)
    session.cookies.update(cookies)
使用加载的cookie进行请求
response = session.get('https://example.com/protected_page')
检查请求是否成功
if response.ok:
    print(response.text)

使用cookieJar对象管理cookie：在requests库中，session.cookies是一个RequestsCookieJar对象，我们可以通过它来管理和操作cookie。RequestsCookieJar提供了许多便捷的方法，比如set()、get()等，可以对cookie进行增删改查。

三、使用http.cookiejar模块、与requests结合使用

Python的http.cookiejar模块提供了对cookie的管理功能，我们可以结合requests库使用http.cookiejar来实现更加复杂的cookie管理。

使用http.cookiejar.LWPCookieJar：LWPCookieJar是http.cookiejar模块中的一个类，它可以将cookie保存到文件中，并支持以Netscape格式存储和读取cookie。我们可以将其与requests库结合使用，实现cookie的管理和持久化。

import requests
import http.cookiejar as cookielib
创建一个LWPCookieJar对象
cookie_jar = cookielib.LWPCookieJar('cookies.txt')
创建一个Session对象
session = requests.Session()
将cookie_jar与session关联
session.cookies = cookie_jar
进行登录请求，假设登录的URL为'https://example.com/login'
login_url = 'https://example.com/login'
login_data = {
    'username': 'your_username',
    'password': 'your_password'
}
response = session.post(login_url, data=login_data)
检查登录是否成功
if response.ok:
    # 将cookie保存到文件
    cookie_jar.save(ignore_discard=True, ignore_expires=True)

从文件加载cookie并使用：在需要使用已保存的cookie时，可以从文件中加载LWPCookieJar对象并将其与requests.Session()关联。这样可以方便地实现会话恢复和管理。

import requests
import http.cookiejar as cookielib
创建一个LWPCookieJar对象
cookie_jar = cookielib.LWPCookieJar()
从文件中加载cookie
cookie_jar.load('cookies.txt', ignore_discard=True, ignore_expires=True)
创建一个Session对象
session = requests.Session()
将cookie_jar与session关联
session.cookies = cookie_jar
使用加载的cookie进行请求
response = session.get('https://example.com/protected_page')
检查请求是否成功
if response.ok:
    print(response.text)

四、使用第三方库browsercookie、管理浏览器的cookie

除了requests库和http.cookiejar模块，Python中还有一些第三方库可以帮助我们管理浏览器的cookie。browsercookie库就是一个非常方便的工具，它可以直接从浏览器中读取cookie并与requests.Session()结合使用。

使用browsercookie读取浏览器cookie：browsercookie库支持从多个浏览器中读取cookie，包括Chrome、Firefox、Edge等。我们可以将读取到的cookie与requests.Session()关联，实现对浏览器会话的模拟和管理。

import requests
import browsercookie
读取Chrome浏览器的cookie
cj = browsercookie.chrome()
创建一个Session对象
session = requests.Session()
将浏览器cookie与session关联
session.cookies = cj
使用浏览器的cookie进行请求
response = session.get('https://example.com/protected_page')
检查请求是否成功
if response.ok:
    print(response.text)