fix: fixed imports to comply with bec_lib refactoring

This commit is contained in:
appel_c 2023-11-13 13:50:19 +01:00
parent 7fefb44462
commit 79cfaf6dc0

View File

@ -9,7 +9,7 @@ from ophyd_devices.utils import bec_utils
from bec_lib.bec_service import SERVICE_CONFIG from bec_lib.bec_service import SERVICE_CONFIG
from bec_lib.file_utils import FileWriterMixin from bec_lib.file_utils import FileWriterMixin
from bec_lib.messages import BECMessage, MessageEndpoints from bec_lib import messages, MessageEndpoints
from ophyd_devices.epics.devices.bec_scaninfo_mixin import BecScaninfoMixin from ophyd_devices.epics.devices.bec_scaninfo_mixin import BecScaninfoMixin
@ -257,9 +257,9 @@ class SLSDetectorBase(ABC, Device):
""" """
pipe = self._producer.pipeline() pipe = self._producer.pipeline()
if successful is None: if successful is None:
msg = BECMessage.FileMessage(file_path=self.filepath, done=done) msg = messages.FileMessage(file_path=self.filepath, done=done)
else: else:
msg = BECMessage.FileMessage(file_path=self.filepath, done=done, successful=successful) msg = messages.FileMessage(file_path=self.filepath, done=done, successful=successful)
self._producer.set_and_publish( self._producer.set_and_publish(
MessageEndpoints.public_file(self.scaninfo.scanID, self.name), msg.dumps(), pipe=pipe MessageEndpoints.public_file(self.scaninfo.scanID, self.name), msg.dumps(), pipe=pipe
) )