diff --git a/apns.py b/apns.py index d044d34..f29ec2f 100644 --- a/apns.py +++ b/apns.py @@ -27,7 +27,7 @@ from datetime import datetime, timedelta from time import mktime from random import getrandbits -from socket import socket, AF_INET, SOCK_STREAM, timeout +from socket import socket, AF_INET, SOCK_STREAM, timeout, error from struct import pack, unpack import os @@ -338,7 +338,7 @@ def send_notification(self, token_hex, payload, expiry=None): try: # Connection might have been closed self.write(self._get_notification(token_hex, payload, identifier, expiry)) - except SSLError: + except (SSLError, error): # We were disconnected or timeout. # Prepare to reconnect. self._disconnect() @@ -365,4 +365,4 @@ def send_notification(self, token_hex, payload, expiry=None): pass # Timeouts are OK - don't reconnect except APNResponseError: self._disconnect() # Make sure we're ready for next send. - raise \ No newline at end of file + raise