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.