Python 2.6 之前版本中 urllib2.urlopen() 的超时

Question

urllib2 文档 表示 Python 2.6 中添加了 timeout 参数。不幸的是我的代码库一直在Python 2.5和2.4平台上运行。

有没有其他方法来模拟超时？我想做的就是允许代码在固定的时间内与远程服务器通信。

也许有任何替代的内置库？ （不想安装第 3 方，例如 pycurl）

Answer 1

您可以使用以下命令为所有套接字操作（包括 HTTP 请求）设置全局超时：

< code>socket.setdefaulttimeout()

像这样：

import urllib2
import socket
socket.setdefaulttimeout(30)
f = urllib2.urlopen('http://www.python.org/')

在这种情况下，您的 urllib2 请求将在 30 秒后超时并抛出套接字异常。 （这是在Python 2.3中添加的）

Answer 2

令人相当恼火的是，您可以重写 urllib2.HTTPHandler 使用的 httplib.HTTPConnection 类。

def urlopen_with_timeout(url, data=None, timeout=None):

  # Create these two helper classes fresh each time, since
  # timeout needs to be in the closure.
  class TimeoutHTTPConnection(httplib.HTTPConnection):
    def connect(self):
      """Connect to the host and port specified in __init__."""
      msg = "getaddrinfo returns an empty list"
      for res in socket.getaddrinfo(self.host, self.port, 0,
                      socket.SOCK_STREAM): 
        af, socktype, proto, canonname, sa = res
        try:
          self.sock = socket.socket(af, socktype, proto)
          if timeout is not None:
            self.sock.settimeout(timeout)
          if self.debuglevel > 0:
            print "connect: (%s, %s)" % (self.host, self.port)
          self.sock.connect(sa)
        except socket.error, msg:
          if self.debuglevel > 0:
            print 'connect fail:', (self.host, self.port)
          if self.sock:
            self.sock.close()
          self.sock = None
          continue
        break
      if not self.sock:
        raise socket.error, msg

  class TimeoutHTTPHandler(urllib2.HTTPHandler):
    http_request = urllib2.AbstractHTTPHandler.do_request_
    def http_open(self, req):
      return self.do_open(TimeoutHTTPConnection, req)

  opener = urllib2.build_opener(TimeoutHTTPHandler)
  opener.open(url, data)

Answer 3

我认为你最好的选择是用 2.6 维护分支的更改

该文件应位于 /usr/lib/python2.4/urllib2.py （在 Linux 和 2.4 上）

Answer 4

我使用标准库中的 httplib。它有一个极其简单的 API，但正如您可能猜到的那样，它只处理 http。 IIUC urllib 使用 httplib 来实现 http 的东西。

Answer 5

您必须在两个地方设置超时。

import urllib2
import socket

socket.setdefaulttimeout(30)
f = urllib2.urlopen('http://www.python.org/', timeout=30)

Answer 6

嗯，2.4 或 2.6 中处理超时的方式是相同的。如果你在 2.6 中打开 urllib2.py 文件，你会发现它需要一个额外的参数作为超时，并使用答案 1 中提到的 socket.defaulttimeout() 方法来处理它。

所以你真的不需要更新你的 urllib2.py案件。