diff --git a/src/packetserver/client/__init__.py b/src/packetserver/client/__init__.py index 0f4bd7d..511c2e7 100644 --- a/src/packetserver/client/__init__.py +++ b/src/packetserver/client/__init__.py @@ -25,6 +25,7 @@ class Client: self.started = False signal.signal(signal.SIGINT, self.exit_gracefully) signal.signal(signal.SIGTERM, self.exit_gracefully) + self.connection_map = None def exit_gracefully(self, signum, frame): self.stop() @@ -33,10 +34,12 @@ class Client: self.started = False self.clear_connections() self.app.stop() + self.connection_map = None def start(self): self.app.start(self.pe_server, self.pe_port) self.app.register_callsigns(self.callsign) + self.connection_map = self.app._engine._active_handler._handlers[1]._connection_map self.started = True def clear_connections(self): @@ -78,6 +81,9 @@ class Client: return Response(msg) return None + #def send_receive_callsign(self, req: Request, callsign: str, timeout: int = 300) -> Optional[Response]: + # for conn in self.connection_map + def single_connect_send_receive(self, dest: str, req: Request, timeout: int = 300) -> Optional[Response]: conn = self.new_connection(dest) logging.debug("Waiting for connection to be ready.")