Python使用cookielib模块操作cookie的实例教程

Python提供了cookielib模块，用于管理cookie。cookie是HTTP协议中的一种状态管理机制，主要用于Web应用的用户身份认证、记住用户在应用中的行为等。Python的cookielib模块提供了一系列方法，可以方便地对cookie进行操作。

导入模块

在使用cookielib模块之前，需要先导入相关模块：

import cookielib
import urllib2

创建cookie

cookielib.Cookie()用于创建cookie，它的参数包含以下项：

• Name：cookie名字。
• Value：cookie值。
• Domain：cookie所在域名。
• Path：cookie所在路径。
• Expires：cookie的失效时间。
• Secure：是否只在HTTPS安全连接中传输。

示例代码：

import cookielib
import urllib2

cookie = cookielib.Cookie(
    version=0, 
    name='name', 
    value='value', 
    port=None, 
    port_specified=False, 
    domain='example.com', 
    domain_specified=False, 
    domain_initial_dot=False, 
    path='/', 
    path_specified=True, 
    secure=False, 
    expires=None, 
    discard=False, 
    comment=None, 
    comment_url=None, 
    rest=None
)

创建cookie处理器

创建cookie后，需要将cookie添加到cookie处理器中。cookie处理器是urllib2库中的一种处理器，用于处理HTTP Cookie，它提供了一个可以帮助我们自动保存和发送cookie的机制。创建cookie处理器的代码如下：

cookie_jar = cookielib.CookieJar()
cookie_processor = urllib2.HTTPCookieProcessor(cookie_jar)
opener = urllib2.build_opener(cookie_processor, urllib2.HTTPHandler)
urllib2.install_opener(opener)

发送cookie

发送已经创建的cookie给服务器的代码如下：

url = 'http://example.com'
response = urllib2.urlopen(url)
html = response.read()

保存cookie

当客户端收到服务器返回的cookie后，需要将cookie保存到本地，方便以后使用。CookieJar类提供了save()方法，可以将cookie保存到本地文件。代码如下：

cookie_jar.save('cookie.txt')

读取cookie

在下次请求服务器时，可以将保存在本地的cookie读取出来，使用CookieJar类的load()方法即可。代码如下：

cookie_jar2 = cookielib.CookieJar()
cookie_jar2.load('cookie.txt')
cookie_processor2 = urllib2.HTTPCookieProcessor(cookie_jar2)
opener2 = urllib2.build_opener(cookie_processor2, urllib2.HTTPHandler)
urllib2.install_opener(opener2)

示例说明

下面是两条示例说明：

示例1

代码如下，我们将ua字段的cookie从服务器中获取出来，然后输出到控制台上：

import cookielib
import urllib2

cookie_jar = cookielib.CookieJar()
cookie_processor = urllib2.HTTPCookieProcessor(cookie_jar)
opener = urllib2.build_opener(cookie_processor, urllib2.HTTPHandler)
urllib2.install_opener(opener)

url = 'http://example.com/login'
req = urllib2.Request(url)

ua_cookie = None
for cookie in cookie_jar:
    if cookie.name == 'ua':
        ua_cookie = cookie
        break

if ua_cookie is not None:
    print 'ua cookie: ', ua_cookie
else:
    print 'ua cookie not found'

示例2

代码如下，我们将name字段的cookie数据写到cookie.txt文件中，再读入cookie.txt文件并输出到控制台上：

import cookielib
import urllib2

cookie_jar = cookielib.CookieJar()
cookie_processor = urllib2.HTTPCookieProcessor(cookie_jar)
opener = urllib2.build_opener(cookie_processor, urllib2.HTTPHandler)
urllib2.install_opener(opener)

url = 'http://example.com/login'
req = urllib2.Request(url)

name_cookie = cookielib.Cookie(
    version=0, 
    name='name', 
    value='value', 
    port=None, 
    port_specified=False, 
    domain='example.com', 
    domain_specified=False, 
    domain_initial_dot=False, 
    path='/', 
    path_specified=True, 
    secure=False, 
    expires=None, 
    discard=False, 
    comment=None, 
    comment_url=None, 
    rest=None
)

cookie_jar.set_cookie(name_cookie)
cookie_jar.save('cookie.txt')

cookie_jar2 = cookielib.CookieJar()
cookie_jar2.load('cookie.txt')
cookie_processor2 = urllib2.HTTPCookieProcessor(cookie_jar2)
opener2 = urllib2.build_opener(cookie_processor2, urllib2.HTTPHandler)
urllib2.install_opener(opener2)

for cookie in cookie_jar2:
    print cookie