Skip to content

Commit c151007

Browse files
author
Luke Bakken
committed
Make execption handler Python 2 compatible
1 parent 1502977 commit c151007

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

riak/transports/tcp/connection.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import errno
12
import logging
23
import socket
34
import struct
@@ -54,8 +55,11 @@ def _non_connect_send_msg(self, msg_code, data):
5455
"""
5556
try:
5657
self._socket.sendall(self._encode_msg(msg_code, data))
57-
except BrokenPipeError as e:
58-
raise ConnectionClosed(e)
58+
except (IOError, socket.error) as e:
59+
if e.errno == errno.EPIPE:
60+
raise ConnectionClosed(e)
61+
else:
62+
raise
5963

6064
def _send_msg(self, msg_code, data):
6165
self._connect()

0 commit comments

Comments
 (0)