Skip to content

Commit 18f047e

Browse files
committed
Bring socket.timeout back
1 parent 79a17cb commit 18f047e

File tree

1 file changed

+7
-5
lines changed
  • mysql-connector-python/lib/mysql/connector

1 file changed

+7
-5
lines changed

mysql-connector-python/lib/mysql/connector/network.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@
2929
"""Module implementing low-level socket communication with MySQL servers.
3030
"""
3131

32+
# pylint: disable=overlapping-except
33+
3234
import socket
3335
import struct
3436
import warnings
@@ -162,7 +164,7 @@ def _send_pkt(self, sock: socket.socket, address: str, pkt: bytes) -> None:
162164
"""Write packet to the comm channel."""
163165
try:
164166
sock.sendall(pkt)
165-
except TimeoutError as err:
167+
except (socket.timeout, TimeoutError) as err:
166168
raise WriteTimeoutError(errno=3024) from err
167169
except IOError as err:
168170
raise OperationalError(
@@ -241,7 +243,7 @@ def recv(self, sock: socket.socket, address: str) -> bytearray:
241243

242244
# Read the payload, and return packet
243245
return header + self._recv_chunk(sock, size=payload_len)
244-
except TimeoutError as err:
246+
except (socket.timeout, TimeoutError) as err:
245247
raise ReadTimeoutError(errno=3024, msg=err.strerror) from err
246248
except IOError as err:
247249
raise OperationalError(
@@ -427,7 +429,7 @@ def recv(self, sock: socket.socket, address: str) -> bytearray:
427429
struct.unpack("<I", header[4:7] + b"\x00")[0],
428430
)
429431
self._recv_compressed_pkt(sock, compressed_pll, uncompressed_pll)
430-
except TimeoutError as err:
432+
except (socket.timeout, TimeoutError) as err:
431433
raise ReadTimeoutError(errno=3024) from err
432434
except IOError as err:
433435
raise OperationalError(
@@ -700,7 +702,7 @@ def open_connection(self) -> None:
700702
)
701703
self.sock.settimeout(self._connection_timeout)
702704
self.sock.connect(self.unix_socket)
703-
except TimeoutError as err:
705+
except (socket.timeout, TimeoutError) as err:
704706
raise ConnectionTimeoutError(
705707
errno=2002,
706708
values=(
@@ -795,7 +797,7 @@ def open_connection(self) -> None:
795797
self.sock = socket.socket(self._family, socktype, proto)
796798
self.sock.settimeout(self._connection_timeout)
797799
self.sock.connect(sockaddr)
798-
except TimeoutError as err:
800+
except (socket.timeout, TimeoutError) as err:
799801
raise ConnectionTimeoutError(
800802
errno=2003,
801803
values=(

0 commit comments

Comments
 (0)