Set after server dies raises AttributeError.

2 files changed, 5 insertions, 2 deletions

diff --git a/ChangeLog b/ChangeLog
index 798d0e2..eb9c8e1 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,6 @@
+   *  Doing a "set" after server goes away, raised AttributeError:
+      'NoneType' object has no attribute 'sendall'.  Patch by Ken Lalonde
+
    *  incr/decr return None instead of 0 on server connection failure.
       Suggested by Ivan Virabyan
 
diff --git a/memcache.py b/memcache.py
index f674dae..d0d6186 100644
--- a/memcache.py
+++ b/memcache.py
@@ -816,8 +816,8 @@ class Client(local):
         except _ConnectionDeadError:
             # retry once
             try:
-                server._get_socket()
-                return _unsafe_set()
+                if server._get_socket():
+                    return _unsafe_set()
             except (_ConnectionDeadError, socket.error), msg:
                 server.mark_dead(msg)
             return 0