Python:Urllib2检查head中缺少的contentlength

Question

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

该脚本的基本功能是在主机web前端执行HTTP/URL请求，并返回状态代码和内容大小（即content-length）。我在解决如何在将content-lenght分配给变量之前对其进行检查有困难。目前，我有以下几点：

#... 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的子类，因此有一个（有限的）^{}-like interface。您可以使用对get的2参数调用来提供默认值。在

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