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

feat(SafeSlot): slot parameters can be overridden with kwarg; add option to raise

Browse Source
This commit is contained in:
wakonig_k
2025-05-02 16:30:49 +02:00
committed by appel_c
parent 94463afdba
commit 9387275851
1 changed files with 21 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
bec_widgets/utils/error_popups.py
View File

@ -99,16 +99,30 @@ def SafeSlot(*slot_args, **slot_kwargs): # pylint: disable=invalid-name
'verify_sender' keyword argument can be passed with boolean value if the sender should be verified 'verify_sender' keyword argument can be passed with boolean value if the sender should be verified
before executing the slot. If True, the slot will only execute if the sender is a QObject. This is before executing the slot. If True, the slot will only execute if the sender is a QObject. This is
useful to prevent function calls from already deleted objects. useful to prevent function calls from already deleted objects.
'raise_error' keyword argument can be passed with boolean value if the error should be raised
after the error is displayed. This is useful to propagate the error to the caller but should be used
with great care to avoid segfaults.
The keywords above are stored in a container which can be overridden by passing
'_override_slot_params' keyword argument with a dictionary containing the keywords to override.
This is useful to override the default behavior of the decorator for a specific function call.
""" """
popup_error = bool(slot_kwargs.pop("popup_error", False)) _slot_params = {
verify_sender = bool(slot_kwargs.pop("verify_sender", False)) "popup_error": bool(slot_kwargs.pop("popup_error", False)),
"verify_sender": bool(slot_kwargs.pop("verify_sender", False)),
"raise_error": bool(slot_kwargs.pop("raise_error", False)),
}
def error_managed(method): def error_managed(method):
@Slot(*slot_args, **slot_kwargs) @Slot(*slot_args, **slot_kwargs)
@functools.wraps(method) @functools.wraps(method)
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
_override_slot_params = kwargs.pop("_override_slot_params", {})
_slot_params.update(_override_slot_params)
try: try:
if not verify_sender or len(args) == 0: if not _slot_params["verify_sender"] or len(args) == 0:
return method(*args, **kwargs) return method(*args, **kwargs)
_instance = args[0] _instance = args[0]
@ -126,11 +140,11 @@ def SafeSlot(*slot_args, **slot_kwargs): # pylint: disable=invalid-name
except Exception: except Exception:
slot_name = f"{method.__module__}.{method.__qualname__}" slot_name = f"{method.__module__}.{method.__qualname__}"
error_msg = traceback.format_exc() error_msg = traceback.format_exc()
if popup_error: if _slot_params["popup_error"]:
ErrorPopupUtility().custom_exception_hook( ErrorPopupUtility().custom_exception_hook(*sys.exc_info(), popup_error=True)
*sys.exc_info(), popup_error=popup_error
)
logger.error(f"SafeSlot error in slot '{slot_name}':\n{error_msg}") logger.error(f"SafeSlot error in slot '{slot_name}':\n{error_msg}")
if _slot_params["raise_error"]:
raise
return wrapper return wrapper