judas/judas_client
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

feat(connector.py): add reconnect() method to simplify reconnecting to server

Browse Source
This commit is contained in:
Artur Borecki
2025-08-28 23:25:25 +02:00
parent 25f6ebbf59
commit 4496fc60aa
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/judas_client/connector.py
View File

@@ -130,6 +130,12 @@ class Connector:
self.socket.close() self.socket.close()
self.logger.info("Connection closed.") self.logger.info("Connection closed.")
def reconnect(self) -> None:
self.logger.debug("Reconnecting...")
self.close()
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.connect()
def send_hello(self) -> None: def send_hello(self) -> None:
self.logger.debug("[.] Sending hello message...") self.logger.debug("[.] Sending hello message...")
hello_message: Message = Message.hello(self.mac_address) hello_message: Message = Message.hello(self.mac_address)
@@ -141,6 +147,7 @@ class Connector:
time.sleep(1) time.sleep(1)
data: bytes = self.receive() data: bytes = self.receive()
if not data: if not data:
self.reconnect()
continue continue
message = Message.from_bytes(data.strip()) message = Message.from_bytes(data.strip())
self.logger.info(f"[<] Message received: {message}") self.logger.info(f"[<] Message received: {message}")