Python请求 - 从iter_content退出清理

时间:2012-04-13 14:36:27

标签: python python-requests

我正在使用Python Requests模块来使用Twitter流API,这是我的代码:

self.conn = self.session.post(url, data=self.parameters, headers=self.headers)
    print >> sys.stderr, 'Connected to Twitter Streaming API'
    try:
        for line in self.conn.iter_content(self.ITER_BYTES):
            self.status += line
            if line.endswith(self.DIVIDER) and self.status.strip():
                self.handler.handle(self.status)
                self.status = ""
    except Exception as e:
        pass

当我使用KeyboardInterrupt结束此脚本或通过传递终止信号时,我收到以下堆栈错误:

^CTraceback (most recent call last):
  File "python-test.py", line 18, in <module>
    api.start()
  File "bin/polygraph/api/twitter/streaming.py", line 95, in start
    self.conn = self.session.post(url, data=self.parameters, headers=self.headers)
  File "venv/lib/python2.7/site-packages/requests/sessions.py", line 258, in post
    return self.request('post', url, data=data, **kwargs)
  File "venv/lib/python2.7/site-packages/requests/sessions.py", line 208, in request
    r.send(prefetch=prefetch)
  File "venv/lib/python2.7/site-packages/requests/models.py", line 575, in send
    timeout=self.timeout,
  File "venv/lib/python2.7/site-packages/requests/packages/urllib3/connectionpool.py", line 383, in urlopen
    body=body, headers=headers)
  File "venv/lib/python2.7/site-packages/requests/packages/urllib3/connectionpool.py", line 261, in _make_request
    httplib_response = conn.getresponse()
  File "/System/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/httplib.py", line 1013, in getresponse
    response.begin()
  File "/System/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/httplib.py", line 402, in begin
    version, status, reason = self._read_status()
  File "/System/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/httplib.py", line 360, in _read_status
    line = self.fp.readline()
  File "/System/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/socket.py", line 430, in readline
    data = recv(1)
  File "/System/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/ssl.py", line 219, in recv
    return self.read(buflen)
  File "/System/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/ssl.py", line 138, in read
    return self._sslobj.read(len)
KeyboardInterrupt

有没有办法避免这种情况或优雅地退出连接?

1 个答案:

答案 0 :(得分:2)

您只需捕获KeyboardInterrupt异常即可。这does not inherit from Exception,因此您目前无法抓住它。

try:
   do_something()
except KeyboardInterrupt:
   cleanup()

这是为什么except Exception是个坏主意的另一个例子。它不仅可以捕捉到您不想要的东西,而且可能无法捕获您想要的东西。抓住明确的例外情况。