Fix style of exception handler

[z_archive/twitter.git] / twitter / stream.py

diff --git a/twitter/stream.py b/twitter/stream.py
index 2f1e4f5f988baeaa5c5e08aacc1a6a08a3ad06f1..071f6b01bbbe1862d582521fc0928fffefce5bef 100644 (file)
--- a/twitter/stream.py
+++ b/twitter/stream.py
@@ -8,6 +8,7 @@ except ImportError:
 import json
 from ssl import SSLError
 import socket
+import sys
 
 from .api import TwitterCall, wrap_response
 
@@ -20,7 +21,10 @@ class TwitterJSONIter(object):
         self.block = block
 
     def __iter__(self):
-        sock = self.handle.fp._sock.fp._sock
+        if sys.version_info >= (3, 0):
+            sock = self.handle.fp.raw._sock
+        else:
+            sock = self.handle.fp._sock.fp._sock
         sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
         if not self.block:
             sock.setblocking(False)