Python:Urllib2 检查 header 中是否缺少内容长度

Question

我有一个脚本“似乎”可以在我们的测试环境中工作，因为缺乏可见性。然而，我们现在正在预生产环境中进行测试，我们注意到以下问题......

该脚本的基本功能是在主机 Web 前端执行 HTTP/URL 请求并返回状态代码和内容大小(即 content-length)。我在解决如何在将内容分配给变量之前检查内容长度时遇到问题。目前我有以下内容:

#... Code ...
try:
  conn = urllib2.urlopen(line, timeout=1)
  contentLen = conn.headers['content-length']
#... Code ...

但是，对于缺少此内容长度的 URL，如果缺少内容长度，它只会破坏脚本。我不想引发异常，而是想在分配 contentLen 之前简单地检查它是否存在，然后如果存在，则分配实际的 content-length 值(如果不存在)然后分配值“NULL”。因此可以将其打印为“Null”。

提前致谢。

Answer 1

conn.headers 是一个 httplib.HTTPMessage，它是 rfc822.Message 的子类，因此有一个(有限的) dict-like interface 。您可以使用对 get 的 2 参数调用来提供默认值。

contentLen = conn.headers.get('content-length','NULL')