bec-ipython-client 3.79.0__tar.gz → 3.81.1__tar.gz

{bec_ipython_client-3.79.0 → bec_ipython_client-3.81.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bec_ipython_client
-Version: 3.79.0
+Version: 3.81.1
 Summary: BEC IPython client
 Project-URL: Bug Tracker, https://github.com/bec-project/bec/issues
 Project-URL: Homepage, https://github.com/bec-project/bec

{bec_ipython_client-3.79.0 → bec_ipython_client-3.81.1}/bec_ipython_client/main.py

@@ -13,6 +13,7 @@ import redis
 import redis.exceptions
 from IPython.terminal.ipapp import TerminalIPythonApp
 from IPython.terminal.prompts import Prompts, Token
+from pydantic import ValidationError
 from rich.console import Console
 from rich.panel import Panel
 from rich.text import Text
@@ -30,6 +31,7 @@ from bec_lib.client import BECClient
 from bec_lib.logger import bec_logger
 from bec_lib.redis_connector import RedisConnector
 from bec_lib.service_config import ServiceConfig
+from bec_lib.utils.pydantic_pretty_print import pretty_print_pydantic_validation_error
 
 logger = bec_logger.logger
 
@@ -213,16 +215,16 @@ class BECIPythonClient:
         header.append("Alarm Raised\n", style="bold red")
         header.append(f"Severity: {alarm.severity.name}\n", style="bold")
         header.append(f"Type: {alarm.alarm_type}\n", style="bold")
-        if alarm.alarm.source.get("device"):
-            header.append(f"Device: {alarm.alarm.source['device']}\n", style="bold")
+        if alarm.alarm.info.device:
+            header.append(f"Device: {alarm.alarm.info.device}\n", style="bold")
 
         console.print(Panel(header, title="Alarm Info", border_style="red", expand=False))
 
         # --- SHOW SUMMARY
-        if alarm.alarm.compact_msg:
+        if alarm.alarm.info.compact_error_message:
             console.print(
                 Panel(
-                    Text(alarm.alarm.compact_msg, style="yellow"),
+                    Text(alarm.alarm.info.compact_error_message, style="yellow"),
                     title="Summary",
                     border_style="yellow",
                     expand=False,
@@ -230,7 +232,7 @@ class BECIPythonClient:
             )
 
         # --- SHOW FULL TRACEBACK
-        tb_str = alarm.alarm.msg
+        tb_str = alarm.alarm.info.error_message
         if tb_str:
             try:
                 console.print(tb_str)
@@ -248,6 +250,9 @@ def _ip_exception_handler(
         evalue.pretty_print()
         print("For more details, use 'bec.show_last_alarm()'")
         return
+    if issubclass(etype, ValidationError):
+        pretty_print_pydantic_validation_error(evalue)
+        return
     if issubclass(etype, (ScanInterruption, DeviceConfigError)):
         print(f"\x1b[31m {evalue.__class__.__name__}:\x1b[0m {evalue}")
         return

{bec_ipython_client-3.79.0 → bec_ipython_client-3.81.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "bec_ipython_client"
-version = "3.79.0"
+version = "3.81.1"
 description = "BEC IPython client"
 requires-python = ">=3.11"
 classifiers = [

{bec_ipython_client-3.79.0 → bec_ipython_client-3.81.1}/tests/client_tests/test_bec_client.py

@@ -216,13 +216,13 @@ def test_bec_ipython_client_property_access(ipython_client):
 
 def test_bec_ipython_client_show_last_alarm(ipython_client, capsys):
     client = ipython_client
-    alarm_msg = messages.AlarmMessage(
-        severity=Alarms.MAJOR,
-        alarm_type="TestAlarm",
-        source={},
-        msg="This is a test alarm",
-        compact_msg="Test alarm",
+    error_info = messages.ErrorInfo(
+        error_message="This is a test alarm",
+        compact_error_message="Test alarm",
+        exception_type="TestAlarm",
+        device=None,
     )
+    alarm_msg = messages.AlarmMessage(severity=Alarms.MAJOR, info=error_info)
     client.alarm_handler = AlarmHandler(connector=mock.MagicMock())
     client.alarm_handler.add_alarm(alarm_msg)
     client._alarm_history.append(

{bec_ipython_client-3.79.0 → bec_ipython_client-3.81.1}/tests/end-2-end/test_scans_e2e.py

@@ -737,12 +737,12 @@ def test_update_config(bec_ipython_client_fixture):
     bec = bec_ipython_client_fixture
     bec.metadata.update({"unit_test": "test_update_config"})
     demo_config_path = os.path.join(os.path.dirname(configs.__file__), "demo_config.yaml")
-    config = bec.config._load_config_from_file(demo_config_path)
+    config = bec.device_manager.config_helper._load_config_from_file(demo_config_path)
     config.pop("samx")
-    bec.config.send_config_request(action="set", config=config)
+    bec.device_manager.config_helper.send_config_request(action="set", config=config)
     assert "samx" not in bec.device_manager.devices
-    config = bec.config._load_config_from_file(demo_config_path)
-    bec.config.send_config_request(action="set", config=config)
+    config = bec.device_manager.config_helper._load_config_from_file(demo_config_path)
+    bec.device_manager.config_helper.send_config_request(action="set", config=config)
 
 
 @pytest.mark.timeout(100)

{bec_ipython_client-3.79.0 → bec_ipython_client-3.81.1}/tests/end-2-end/test_scans_lib_e2e.py

@@ -322,7 +322,9 @@ def test_config_reload(
     num_devices = len(bec.device_manager.devices)
     if raises_error:
         with pytest.raises(DeviceConfigError):
-            bec.config.update_session_with_file(runtime_config_file_path)
+            bec.config.update_session_with_file(
+                runtime_config_file_path, force=True, validate=False
+            )
         if deletes_config:
             assert len(bec.device_manager.devices) == 0
         elif disabled_device:
@@ -330,7 +332,7 @@ def test_config_reload(
         else:
             assert len(bec.device_manager.devices) == num_devices
     else:
-        bec.config.update_session_with_file(runtime_config_file_path)
+        bec.config.update_session_with_file(runtime_config_file_path, force=True, validate=False)
         assert len(bec.device_manager.devices) == 2
     for dev in disabled_device:
         assert bec.device_manager.devices[dev].enabled is False
@@ -375,7 +377,7 @@ def test_config_reload_with_describe_failure(bec_test_config_file_path, bec_clie
         f.write(yaml.dump(config))
 
     with pytest.raises(DeviceConfigError):
-        bec.config.update_session_with_file(runtime_config_file_path)
+        bec.config.update_session_with_file(runtime_config_file_path, force=True, validate=False)
 
     assert len(bec.device_manager.devices) == 2
     assert bec.device_manager.devices["eyefoc"].enabled is True
@@ -386,7 +388,7 @@ def test_config_reload_with_describe_failure(bec_test_config_file_path, bec_clie
         f"e2e_test_hexapod_fail", messages.DeviceStatusMessage(device="hexapod", status=0)
     )
 
-    bec.config.update_session_with_file(runtime_config_file_path)
+    bec.config.update_session_with_file(runtime_config_file_path, force=True)
     assert len(bec.device_manager.devices) == 2
     assert bec.device_manager.devices["eyefoc"].enabled is True
     assert bec.device_manager.devices["hexapod"].enabled is True
@@ -412,13 +414,13 @@ def test_config_add_remove_device(bec_client_lib):
             "readOnly": False,
         }
     }
-    bec.config.send_config_request(action="add", config=config)
+    bec.device_manager.config_helper.send_config_request(action="add", config=config)
     with pytest.raises(DeviceConfigError) as config_error:
-        bec.config.send_config_request(action="add", config=config)
+        bec.device_manager.config_helper.send_config_request(action="add", config=config)
     assert config_error.match("Device new_device already exists")
     assert "new_device" in dev
 
-    bec.config.send_config_request(action="remove", config={"new_device": {}})
+    bec.device_manager.config_helper.send_config_request(action="remove", config={"new_device": {}})
     assert "new_device" not in dev
 
     device_config_msg = bec.connector.get(MessageEndpoints.device_config())
@@ -429,7 +431,7 @@ def test_config_add_remove_device(bec_client_lib):
 
     config["new_device"]["deviceClass"] = "ophyd_devices.doesnt_exist"
     with pytest.raises(DeviceConfigError) as config_error:
-        bec.config.send_config_request(action="add", config=config)
+        bec.device_manager.config_helper.send_config_request(action="add", config=config)
     assert config_error.match("module 'ophyd_devices' has no attribute 'doesnt_exist'")
     assert "new_device" not in dev
     assert "samx" in dev