update exceptions

Benzinga · Jul 9, 2020 · 23aa672 · 23aa672
1 parent a46f95d
commit 23aa672
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bztcp/client.py b/bztcp/client.py
@@ -119,7 +119,10 @@ def recv(self):
  return Message.from_bytes(line + BZ_EOT)
 
  # Buffer more data if not.
- buf = self._sock.recv(4096)
+ try:
+ buf = self._sock.recv(4096)
+ except ConnectionResetError:
+ return None
 
  if not buf:
  return None