如何在urllib.urlretrieve中捕获404错误

时间:2009-08-20 20:14:40

标签: python http url urllib

背景:我正在使用urllib.urlretrieve,而不是urllib*模块中的任何其他函数,因为钩子函数支持(参见下面的reporthook)..这用于显示文本进度条。这是Python> = 2.6。

>>> urllib.urlretrieve(url[, filename[, reporthook[, data]]])

然而,urlretrieve是如此愚蠢,以至于它无法检测HTTP请求的状态(例如:是404还是200?)。

>>> fn, h = urllib.urlretrieve('http://google.com/foo/bar')
>>> h.items() 
[('date', 'Thu, 20 Aug 2009 20:07:40 GMT'),
 ('expires', '-1'),
 ('content-type', 'text/html; charset=ISO-8859-1'),
 ('server', 'gws'),
 ('cache-control', 'private, max-age=0')]
>>> h.status
''
>>>

下载具有类似钩子支持的远程HTTP文件(显示进度条)和一个不错的HTTP错误处理的最有名的方法是什么?

3 个答案:

答案 0 :(得分:28)

查看urllib.urlretrieve的完整代码:

def urlretrieve(url, filename=None, reporthook=None, data=None):
  global _urlopener
  if not _urlopener:
    _urlopener = FancyURLopener()
  return _urlopener.retrieve(url, filename, reporthook, data)

换句话说,您可以使用urllib.FancyURLopener(它是公共urllib API的一部分)。您可以覆盖http_error_default以检测404:

class MyURLopener(urllib.FancyURLopener):
  def http_error_default(self, url, fp, errcode, errmsg, headers):
    # handle errors the way you'd like to

fn, h = MyURLopener().retrieve(url, reporthook=my_report_hook)

答案 1 :(得分:14)

您应该使用:

import urllib2

try:
    resp = urllib2.urlopen("http://www.google.com/this-gives-a-404/")
except urllib2.URLError, e:
    if not hasattr(e, "code"):
        raise
    resp = e

print "Gave", resp.code, resp.msg
print "=" * 80
print resp.read(80)

编辑:这里的理由是,除非你期望异常状态,否则它会发生异常,你可能甚至没想过 - 所以不要让你代码在不成功时继续运行,默认行为 - 非常合理 - 禁止其执行。

答案 2 :(得分:2)

URL Opener对象的“retreive”方法支持reporthook并在404上抛出异常。

http://docs.python.org/library/urllib.html#url-opener-objects