refactor: fix bec_lib imports

bec_widgets/examples/extreme/extreme.py

@@ -19,7 +19,7 @@ from pyqtgraph import mkBrush, mkPen
 from pyqtgraph.Qt import QtCore, uic
 from pyqtgraph.Qt import QtWidgets
 
-from bec_lib.core import MessageEndpoints
+from bec_lib import MessageEndpoints
 from bec_widgets.qt_utils import Crosshair, Colors
 
 

bec_widgets/examples/mca_readout/mca_plot.py

@@ -8,7 +8,7 @@ from PyQt5.QtWidgets import (
     QWidget,
 )
 
-from bec_lib.core import MessageEndpoints, BECMessage
+from bec_lib import MessageEndpoints, messages
 
 
 class StreamApp(QWidget):
@@ -102,7 +102,7 @@ class StreamApp(QWidget):
 
     @staticmethod
     def _streamer_cb(msg, *, parent, **_kwargs) -> None:
-        msgMCS = BECMessage.DeviceMessage.loads(msg.value)
+        msgMCS = messages.DeviceMessage.loads(msg.value)
         print(msgMCS)
         row = msgMCS.content["signals"][parent.sub_device]
         metadata = msgMCS.metadata
@@ -123,7 +123,7 @@ class StreamApp(QWidget):
     def _device_cv(msg, *, parent, **_kwargs) -> None:
         print("Getting ScanID")
 
-        msgDEV = BECMessage.ScanStatusMessage.loads(msg.value)
+        msgDEV = messages.ScanStatusMessage.loads(msg.value)
 
         current_scanID = msgDEV.content["scanID"]
 
@@ -143,7 +143,7 @@ class StreamApp(QWidget):
 
 if __name__ == "__main__":
     import argparse
-    from bec_lib.core import RedisConnector
+    from bec_lib import RedisConnector
 
     parser = argparse.ArgumentParser(description="Stream App.")
     parser.add_argument("--port", type=str, default="pc15543:6379", help="Port for RedisConnector")

bec_widgets/examples/mca_readout/mca_sim.py

@@ -1,4 +1,4 @@
-from bec_lib.core import BECMessage, MessageEndpoints, RedisConnector
+from bec_lib import messages, MessageEndpoints, RedisConnector
 import time
 
 connector = RedisConnector("localhost:6379")
@@ -15,7 +15,7 @@ metadata.update(
 )
 for ii in range(20):
     data = {"mca1": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10], "mca2": [10, 9, 8, 7, 6, 5, 4, 3, 2, 1]}
-    msg = BECMessage.DeviceMessage(
+    msg = messages.DeviceMessage(
         signals=data,
         metadata=metadata,
     ).dumps()

bec_widgets/examples/motor_movement/motor_example.py

@@ -24,7 +24,7 @@ from PyQt5.QtWidgets import QMessageBox
 from PyQt5.QtWidgets import QShortcut
 from pyqtgraph.Qt import QtWidgets, uic, QtCore
 
-from bec_lib.core import MessageEndpoints, BECMessage
+from bec_lib import MessageEndpoints, messages
 from bec_widgets.qt_utils import DoubleValidationDelegate
 
 
@@ -1295,7 +1295,7 @@ class MotorControl(QThread):
 
     @staticmethod
     def _device_status_callback_motors(msg, *, parent, **_kwargs) -> None:
-        deviceMSG = BECMessage.DeviceMessage.loads(msg.value)
+        deviceMSG = messages.DeviceMessage.loads(msg.value)
         if parent.motor_x.name in deviceMSG.content["signals"]:
             parent.current_x = deviceMSG.content["signals"][parent.motor_x.name]["value"]
         elif parent.motor_y.name in deviceMSG.content["signals"]:
@@ -1307,9 +1307,7 @@ if __name__ == "__main__":
     import yaml
     import argparse
 
-    from bec_lib import BECClient
-
-    from bec_lib.core import ServiceConfig
+    from bec_lib import BECClient, ServiceConfig
 
     parser = argparse.ArgumentParser(description="Motor App")
 

bec_widgets/examples/oneplot/oneplot.py

@@ -3,7 +3,7 @@ import os
 import numpy as np
 import PyQt5.QtWidgets
 import pyqtgraph as pg
-from bec_lib.core import MessageEndpoints
+from bec_lib import MessageEndpoints
 from PyQt5.QtCore import pyqtSignal, pyqtSlot
 from PyQt5.QtWidgets import QApplication, QTableWidgetItem, QWidget
 from pyqtgraph import mkBrush, mkColor, mkPen

bec_widgets/examples/stream_plot/stream_plot.py

@@ -7,8 +7,8 @@ from typing import Any
 import numpy as np
 import pyqtgraph
 import pyqtgraph as pg
-from bec_lib.core import BECMessage, MessageEndpoints
-from bec_lib.core.redis_connector import MessageObject, RedisConnector
+from bec_lib import messages, MessageEndpoints
+from bec_lib.redis_connector import MessageObject, RedisConnector
 from PyQt5.QtCore import pyqtSlot
 from PyQt5.QtWidgets import QCheckBox, QTableWidgetItem
 from pyqtgraph import mkBrush, mkColor, mkPen
@@ -210,7 +210,7 @@ class StreamPlot(QtWidgets.QWidget):
                 np.where(self.plotter_data_x[0] < 10 ** region[1])[0][-1],
             ]
         }
-        msg = BECMessage.DeviceMessage(signals=return_dict).dumps()
+        msg = messages.DeviceMessage(signals=return_dict).dumps()
         self.producer.set_and_publish("px_stream/gui_event", msg=msg)
         self.roi_signal.emit(region)
 
@@ -268,7 +268,7 @@ class StreamPlot(QtWidgets.QWidget):
                 continue
             endpoint = f"px_stream/projection_{self._current_proj}/data"
             msgs = self.client.producer.lrange(topic=endpoint, start=-1, end=-1)
-            data = [BECMessage.DeviceMessage.loads(msg) for msg in msgs]
+            data = [messages.DeviceMessage.loads(msg) for msg in msgs]
             if not data:
                 continue
             with np.errstate(divide="ignore", invalid="ignore"):
@@ -293,7 +293,7 @@ class StreamPlot(QtWidgets.QWidget):
         proj_nr = content["signals"]["proj_nr"]
         endpoint = f"px_stream/projection_{proj_nr}/metadata"
         msg_raw = self.client.producer.get(topic=endpoint)
-        msg = BECMessage.DeviceMessage.loads(msg_raw)
+        msg = messages.DeviceMessage.loads(msg_raw)
         self._current_q = msg.content["signals"]["q"]
         self._current_norm = msg.content["signals"]["norm_sum"]
         self._current_metadata = msg.content["signals"]["metadata"]