Followup to r86383: it seems that in some cases (buildbots), the server

closes the connection before we can call shutdown().
This commit is contained in:
Antoine Pitrou 2010-11-10 08:59:25 +00:00
parent 1790ed2b2e
commit 81c87c5e9a
1 changed files with 9 additions and 3 deletions

View File

@ -22,7 +22,7 @@ Public functions: Internaldate2tuple
__version__ = "2.58"
import binascii, random, re, socket, subprocess, sys, time
import binascii, errno, random, re, socket, subprocess, sys, time
__all__ = ["IMAP4", "IMAP4_stream", "Internaldate2tuple",
"Int2AP", "ParseFlags", "Time2Internaldate"]
@ -260,8 +260,14 @@ class IMAP4:
def shutdown(self):
"""Close I/O established in "open"."""
self.file.close()
self.sock.shutdown(socket.SHUT_RDWR)
self.sock.close()
try:
self.sock.shutdown(socket.SHUT_RDWR)
except socket.error as e:
# The server might already have closed the connection
if e.errno != errno.ENOTCONN:
raise
finally:
self.sock.close()
def socket(self):