在前一节中,我们了解了 requests 的基本用法,如基本的 GET、POST 请求以及Response
对象。本节中,我们再来了解下 requests 的一些高级用法,如文件上传、cookie 设置、代理设置等。
1. 文件上传
我们知道 requests 可以模拟提交一些数据。假如有的网站需要上传文件,我们也可以用它来实现,这非常简单,示例如下:
1 |
import requests |
在前一节中我们保存了一个文件 favicon.ico,这次用它来模拟文件上传的过程。需要注意的是,favicon.ico 需要和当前脚本在同一目录下。如果有其他文件,当然也可以使用其他文件来上传,更改下代码即可。
运行结果如下:
1 |
{ |
以上省略部分内容,这个网站会返回响应,里面包含files
这个字段,而form
字段是空的,这证明文件上传部分会单独有一个files
字段来标识。
2. Cookies
前面我们使用 urllib 处理过 Cookies,写法比较复杂,而有了 requests,获取和设置 Cookies 只需一步即可完成。
我们先用一个实例看一下获取 Cookies 的过程:
1 |
import requests |
运行结果如下:
1 |
<RequestsCookieJar > |
这里我们首先调用cookies
属性即可成功得到 Cookies,可以发现它是RequestCookieJar
类型。然后用items()
方法将其转化为元组组成的列表,遍历输出每一个 Cookie 的名称和值,实现 Cookie 的遍历解析。
当然,我们也可以直接用 Cookie 来维持登录状态,下面以知乎为例来说明。首先登录知乎,将Headers
中的Cookie
内容复制下来,如图 3-6 所示。
图 3-6 Cookie
这里可以替换成你自己的Cookie
,将其设置到Headers
里面,然后发送请求,示例如下:
1 |
import requests |
我们发现,结果中包含了登录后的结果,如图 3-7 所示,这证明登录成功。
图 3-7 运行结果
当然,你也可以通过cookies
参数来设置,不过这样就需要构造RequestsCookieJar
对象,而且需要分割一下cookies
。这相对烦琐,不过效果是相同的,示例如下:
1 |
import requests |
这里我们首先新建了一个RequestCookieJar
对象,然后将复制下来的cookies
利用split()
方法分割,接着利用set()
方法设置好每个 Cookie 的key
和value
,然后通过调用 requests 的get()
方法并传递给cookies
参数即可。当然,由于知乎本身的限制,headers
参数也不能少,只不过不需要在原来的headers
参数里面设置cookie
字段了。
测试后,发现同样可以正常登录知乎。
3. 会话维持
在 requests 中,如果直接利用get()
或post()
等方法的确可以做到模拟网页的请求,但是这实际上是相当于不同的会话,也就是说相当于你用了两个浏览器打开了不同的页面。
设想这样一个场景,第一个请求利用post()
方法登录了某个网站,第二次想获取成功登录后的自己的个人信息,你又用了一次get()
方法去请求个人信息页面。实际上,这相当于打开了两个浏览器,是两个完全不相关的会话,能成功获取个人信息吗?那当然不能。
有小伙伴可能说了,我在两次请求时设置一样的cookies
不就行了?可以,但这样做起来显得很烦琐,我们有更简单的解决方法。
其实解决这个问题的主要方法就是维持同一个会话,也就是相当于打开一个新的浏览器选项卡而不是新开一个浏览器。但是我又不想每次设置cookies
,那该怎么办呢?这时候就有了新的利器——Session
对象。
利用它,我们可以方便地维护一个会话,而且不用担心cookies
的问题,它会帮我们自动处理好。示例如下:
1 |
import requests |
这里我们请求了一个测试网址http://httpbin.org/cookies/set/number/123456789。请求这个网址时,可以设置一个 cookie,名称叫作 number,内容是 123456789,随后又请求了http://httpbin.org/cookies,此网址可以获取当前的 Cookies。
这样能成功获取到设置的 Cookies 吗?试试看。
运行结果如下:
1 |
{ |
这并不行。我们再用Session
试试看:
1 |
import requests |
再看下运行结果:
1 |
{ |
成功获取!这下能体会到同一个会话和不同会话的区别了吧!
所以,利用Session
,可以做到模拟同一个会话而不用担心 Cookies 的问题。它通常用于模拟登录成功之后再进行下一步的操作。
Session
在平常用得非常广泛,可以用于模拟在一个浏览器中打开同一站点的不同页面,后面会有专门的章节来讲解这部分内容。
4. SSL 证书验证
此外,requests 还提供了证书验证的功能。当发送 HTTP 请求的时候,它会检查 SSL 证书,我们可以使用verify
参数控制是否检查此证书。其实如果不加verify
参数的话,默认是True
,会自动验证。
前面我们提到过,12306 的证书没有被官方 CA 机构信任,会出现证书验证错误的结果。我们现在访问它,都可以看到一个证书问题的页面,如图 3-8 所示。
图 3-8 错误页面
现在我们用 requests 来测试一下:
1 |
import requests |
运行结果如下:
1 |
requests.exceptions.SSLError: ("bad handshake: Error([('SSL routines', 'tls_process_server_certificate', 'certificate verify failed')],)",) |
这里提示一个错误SSLError
,表示证书验证错误。所以,如果请求一个 HTTPS 站点,但是证书验证错误的页面时,就会报这样的错误,那么如何避免这个错误呢?很简单,把verify
参数设置为False
即可。相关代码如下:
1 |
import requests |
这样就会打印出请求成功的状态码:
1 |
/usr/local/lib/python3.6/site-packages/urllib3/connectionpool.py:852: InsecureRequestWarning: Unverified HTTPS request is being made. Adding certificate verification is strongly advised. See: https://urllib3.readthedocs.io/en/latest/advanced-usage.html#ssl-warnings |
不过我们发现报了一个警告,它建议我们给它指定证书。我们可以通过设置忽略警告的方式来屏蔽这个警告:
1 |
import requests |
或者通过捕获警告到日志的方式忽略警告:
1 |
import logging |
当然,我们也可以指定一个本地证书用作客户端证书,这可以是单个文件(包含密钥和证书)或一个包含两个文件路径的元组:
1 |
import requests |
当然,上面的代码是演示实例,我们需要有 crt 和 key 文件,并且指定它们的路径。注意,本地私有证书的key
必须是解密状态,加密状态的key
是不支持的。
5. 代理设置
对于某些网站,在测试的时候请求几次,能正常获取内容。但是一旦开始大规模爬取,对于大规模且频繁的请求,网站可能会弹出验证码,或者跳转到登录认证页面,更甚者可能会直接封禁客户端的 IP,导致一定时间段内无法访问。
那么,为了防止这种情况发生,我们需要设置代理来解决这个问题,这就需要用到proxies
参数。可以用这样的方式设置:
1 |
import requests |
当然,直接运行这个实例可能不行,因为这个代理可能是无效的,请换成自己的有效代理试验一下。
若代理需要使用 HTTP Basic Auth,可以使用类似 http://user:password@host:port 这样的语法来设置代理,示例如下:
1 |
import requests |
除了基本的 HTTP 代理外,requests 还支持 SOCKS 协议的代理。
首先,需要安装 socks 这个库:
1 |
pip3 install 'requests[socks]' |
然后就可以使用 SOCKS 协议代理了,示例如下:
1 |
import requests |
6. 超时设置
在本机网络状况不好或者服务器网络响应太慢甚至无响应时,我们可能会等待特别久的时间才可能收到响应,甚至到最后收不到响应而报错。为了防止服务器不能及时响应,应该设置一个超时时间,即超过了这个时间还没有得到响应,那就报错。这需要用到timeout
参数。这个时间的计算是发出请求到服务器返回响应的时间。示例如下:
1 |
import requests |
通过这样的方式,我们可以将超时时间设置为 1 秒,如果 1 秒内没有响应,那就抛出异常。
实际上,请求分为两个阶段,即连接(connect)和读取(read)。
上面设置的timeout
将用作连接和读取这二者的timeout
总和。
如果要分别指定,就可以传入一个元组:
1 |
r = requests.get('https://www.taobao.com', timeout=(5,11, 30)) |
如果想永久等待,可以直接将timeout
设置为None
,或者不设置直接留空,因为默认是None
。这样的话,如果服务器还在运行,但是响应特别慢,那就慢慢等吧,它永远不会返回超时错误的。其用法如下:
1 |
r = requests.get('https://www.taobao.com', timeout=None) |
或直接不加参数:
1 |
r = requests.get('https://www.taobao.com') |
7. 身份认证
在访问网站时,我们可能会遇到这样的认证页面,如图 3-9 所示。
图 3-9 认证页面
此时可以使用 requests 自带的身份认证功能,示例如下:
1 |
import requests |
如果用户名和密码正确的话,请求时就会自动认证成功,会返回 200 状态码,如果认证失败,则返回 401 状态码。
当然,如果参数都传一个HTTPBasicAuth
类,就显得有点烦琐了,所以 requests 提供了一个更简单的写法,可以直接传一个元组,它会默认使用HTTPBasicAuth
这个类来认证。
所以上面的代码可以直接简写如下:
1 |
import requests |
此外,requests 还提供了其他认证方式,如 OAuth 认证,不过此时需要安装 oauth 包,安装命令如下:
1 |
pip3 install requests_oauthlib |
使用 OAuth1 认证的方法如下:
1 |
import requests |
更多详细的功能可以参考 requests_oauthlib 的官方文档https://requests-oauthlib.readthedocs.org/,在此不再赘述了。
8. Prepared Request
前面介绍 urllib 时,我们可以将请求表示为数据结构,其中各个参数都可以通过一个Request
对象来表示。这在 requests 里同样可以做到,这个数据结构就叫 Prepared Request。我们用实例看一下:
1 |
from requests import Request, Session |
这里我们引入了Request
,然后用url
、data
和headers
参数构造了一个Request
对象,这时需要再调用Session
的prepare_request()
方法将其转换为一个 Prepared Request 对象,然后调用send()
方法发送即可,运行结果如下:
1 |
{ |
可以看到,我们达到了同样的 POST 请求效果。
有了Request
这个对象,就可以将请求当作独立的对象来看待,这样在进行队列调度时会非常方便。后面我们会用它来构造一个Request
队列。
本节讲解了 requests 的一些高级用法,这些用法在后面实战部分会经常用到,需要熟练掌握。更多的用法可以参考 requests 的官方文档:http://docs.python-requests.org/。