Merge pull request #183 from laserson/compat

fix set_timeout backwards compatibility
Thriftpy · Feb 5, 2016 · 010717e · 010717e
2 parents 6b342c7 + 4db14a7
commit 010717e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/thriftpy/transport/socket.py b/thriftpy/transport/socket.py
@@ -75,9 +75,8 @@ def set_timeout(self, ms):
         """Backward compat api, will bind the timeout to both connect_timeout
         and socket_timeout.
         """
-        if ms and ms > 0:
-            self.socket_timeout = ms / 1000
-            self.connect_timeout = self.socket_timeout
+        self.socket_timeout = ms / 1000 if (ms and ms > 0) else None
+        self.connect_timeout = self.socket_timeout
 
     def is_open(self):
         return bool(self.sock)