Skip to content

Commit

Permalink
Improve Connection
Browse files Browse the repository at this point in the history
  • Loading branch information
realshouzy committed May 15, 2024
1 parent 7dce5d5 commit be5d0ab
Showing 1 changed file with 12 additions and 5 deletions.
17 changes: 12 additions & 5 deletions nrw/network/_connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,23 +33,30 @@ def __init__(self, server_ip: str, server_port: int) -> None:
socket.SOCK_STREAM,
)
self._socket.connect((server_ip, server_port))
self._to_server: TextIOWrapper | None = self._socket.makefile(mode="w")
self._from_server: TextIOWrapper | None = self._socket.makefile(mode="r")
self._to_server: TextIOWrapper | None = self._socket.makefile(
mode="w",
encoding="utf-8",
)
self._from_server: TextIOWrapper | None = self._socket.makefile(
mode="r",
encoding="utf-8",
)
except Exception: # pylint: disable=W0718
self._socket = None
self._to_server = None
self._from_server = None

def receive(self) -> str | None:
if self._from_server is not None:
with suppress(IOError):
with suppress(IOError, ValueError):
return self._from_server.readline()
return None

def send(self, message: str) -> None:
if self._to_server is not None:
self._to_server.write(f"{message}\n")
self._to_server.flush()
with suppress(ValueError):
self._to_server.write(f"{message}\n")
self._to_server.flush()

def close(self) -> None:
if self._socket is not None:
Expand Down

0 comments on commit be5d0ab

Please sign in to comment.