version 3.3.1, fixed a regression involving SSL and non-blocking sockets

Fixes #1197

Author: andymccurdy

CHANGES

@@ -1,3 +1,6 @@
+* 3.3.1
+    * Fixed a regression introduced in 3.3.1 involving SSL and non-blocking
+      sockets. #1197
 * 3.3.0
     * Resolve a race condition with the PubSubWorkerThread. #1150
     * Cleanup socket read error messages. Thanks Vic Yu. #1159

redis/__init__.py

@@ -29,7 +29,7 @@ def int_or_str(value):
         return value
 
 
-__version__ = '3.3.0'
+__version__ = '3.3.1'
 VERSION = tuple(map(int_or_str, __version__.split('.')))
 
 __all__ = [

redis/connection.py

@@ -10,12 +10,6 @@
 import threading
 import warnings
 
-try:
-    import ssl
-    ssl_available = True
-except ImportError:
-    ssl_available = False
-
 from redis._compat import (xrange, imap, byte_to_chr, unicode, long,
                            nativestr, basestring, iteritems,
                            LifoQueue, Empty, Full, urlparse, parse_qs,
@@ -34,6 +28,23 @@
     TimeoutError,
 )
 from redis.utils import HIREDIS_AVAILABLE
+
+try:
+    import ssl
+    ssl_available = True
+except ImportError:
+    ssl_available = False
+
+if ssl_available:
+    blocking_exceptions = (
+        BlockingIOError,
+        ssl.SSLWantReadError,
+        ssl.SSLWantWriteError
+    )
+else:
+    blocking_exceptions = (BlockingIOError,)
+
+
 if HIREDIS_AVAILABLE:
     import hiredis
 
@@ -168,7 +179,7 @@ def _read_from_socket(self, length=None, timeout=SENTINEL,
                 if length is not None and length > marker:
                     continue
                 return True
-        except BlockingIOError as ex:
+        except blocking_exceptions as ex:
             # if we're in nonblocking mode and the recv raises a
             # blocking error, simply return False indicating that
             # there's no data to be read. otherwise raise the
@@ -397,7 +408,7 @@ def read_from_socket(self, timeout=SENTINEL, raise_on_timeout=True):
             # data was read from the socket and added to the buffer.
             # return True to indicate that data was read.
             return True
-        except BlockingIOError as ex:
+        except blocking_exceptions as ex:
             # if we're in nonblocking mode and the recv raises a
             # blocking error, simply return False indicating that
             # there's no data to be read. otherwise raise the