bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-13 19:21:50 +02:00
Code Activity

fix(rpc_server): update _serialize_bec_connector to include wait parameter for registration check

Browse Source
This commit is contained in:
wakonig_k
2025-04-12 20:50:52 +02:00
parent 064343acf2
commit d6fccd10f5
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
bec_widgets/utils/rpc_server.py
View File

@ -11,6 +11,7 @@ from bec_lib.endpoints import MessageEndpoints
from bec_lib.logger import bec_logger from bec_lib.logger import bec_logger
from bec_lib.utils.import_utils import lazy_import from bec_lib.utils.import_utils import lazy_import
from qtpy.QtCore import QTimer from qtpy.QtCore import QTimer
from qtpy.QtWidgets import QApplication
from redis.exceptions import RedisError from redis.exceptions import RedisError
from bec_widgets.cli.rpc.rpc_register import RPCRegister from bec_widgets.cli.rpc.rpc_register import RPCRegister
@ -159,7 +160,7 @@ class RPCServer:
# Respect RPC = False # Respect RPC = False
if getattr(obj, "RPC", True) is False: if getattr(obj, "RPC", True) is False:
return None return None
return self._serialize_bec_connector(obj) return self._serialize_bec_connector(obj, wait=True)
def emit_heartbeat(self) -> None: def emit_heartbeat(self) -> None:
""" """
@ -196,15 +197,25 @@ class RPCServer:
max_size=1, max_size=1,
) )
def _serialize_bec_connector(self, connector: BECConnector) -> dict: def _serialize_bec_connector(self, connector: BECConnector, wait=False) -> dict:
""" """
Create the serialization dict for a single BECConnector, Create the serialization dict for a single BECConnector.
setting 'parent_id' via the real nearest BECConnector parent.
Args:
connector (BECConnector): The BECConnector to serialize.
wait (bool): If True, wait until the object is registered in the RPC register.
Returns:
dict: The serialized BECConnector object.
""" """
config_dict = connector.config.model_dump() config_dict = connector.config.model_dump()
config_dict["parent_id"] = getattr(connector, "parent_id", None) config_dict["parent_id"] = getattr(connector, "parent_id", None)
if wait:
while not self.rpc_register.object_is_registered(connector):
QApplication.processEvents()
return { return {
"gui_id": connector.gui_id, "gui_id": connector.gui_id,
"object_name": connector.object_name or connector.__class__.__name__, "object_name": connector.object_name or connector.__class__.__name__,