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

This commit is contained in:
2025-08-28 23:25:25 +02:00
parent 25f6ebbf59
commit 4496fc60aa

View File

@@ -130,6 +130,12 @@ class Connector:
self.socket.close()
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:
self.logger.debug("[.] Sending hello message...")
hello_message: Message = Message.hello(self.mac_address)
@@ -141,6 +147,7 @@ class Connector:
time.sleep(1)
data: bytes = self.receive()
if not data:
self.reconnect()
continue
message = Message.from_bytes(data.strip())
self.logger.info(f"[<] Message received: {message}")