From 82b82659b7919b15d629375866302624b5b6e457 Mon Sep 17 00:00:00 2001 From: wakonig_k Date: Sat, 12 Apr 2025 19:35:52 +0200 Subject: [PATCH] fix(rpc_register): change add_rpc parameter type to BECConnector and add object_is_registered method --- bec_widgets/cli/rpc/rpc_register.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/bec_widgets/cli/rpc/rpc_register.py b/bec_widgets/cli/rpc/rpc_register.py index ee9a69a5..b984e333 100644 --- a/bec_widgets/cli/rpc/rpc_register.py +++ b/bec_widgets/cli/rpc/rpc_register.py @@ -65,7 +65,7 @@ class RPCRegister: return register._broadcast_on_hold @broadcast_update - def add_rpc(self, rpc: QObject): + def add_rpc(self, rpc: BECConnector): """ Add an RPC object to the register. @@ -136,6 +136,18 @@ class RPCRegister: for callback in self.callbacks: callback(connections) + def object_is_registered(self, obj: BECConnector) -> bool: + """ + Check if an object is registered in the RPC register. + + Args: + obj(QObject): The object to check. + + Returns: + bool: True if the object is registered, False otherwise. + """ + return obj.gui_id in self._rpc_register + def add_callback(self, callback: Callable[[dict], None]): """ Add a callback that will be called whenever the registry is updated.