From c1e2d055c522a25a486e7b478becd43869dbdd86 Mon Sep 17 00:00:00 2001 From: Richard Neumann Date: Sun, 21 Aug 2022 15:08:06 +0200 Subject: [PATCH] Outsource sending of packets into send() --- rcon/source/client.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/rcon/source/client.py b/rcon/source/client.py index cff09a3..53d33ef 100644 --- a/rcon/source/client.py +++ b/rcon/source/client.py @@ -34,22 +34,24 @@ class Client(BaseClient, socket_type=SOCK_STREAM): def communicate(self, packet: Packet) -> Packet: """Send and receive a packet.""" - with self._socket.makefile('wb') as file: - file.write(bytes(packet)) - + self.send(packet) response = self.read() if len(response.payload) < self.frag_threshold: return response - with self._socket.makefile('wb') as file: - file.write(bytes(Packet.make_command(self.frag_detect_cmd))) + self.send(Packet.make_command(self.frag_detect_cmd)) while (successor := self.read()).id == response.id: response += successor return response + def send(self, packet: Packet) -> None: + """Send a packet to the server.""" + with self._socket.makefile('wb') as file: + file.write(bytes(packet)) + def read(self) -> Packet: """Read a packet.""" with self._socket.makefile('rb') as file: