diff --git a/src/aare/daq/spreadsheetupdater.py b/src/aare/daq/spreadsheetupdater.py index 79e98f5e..e8abdf5b 100644 --- a/src/aare/daq/spreadsheetupdater.py +++ b/src/aare/daq/spreadsheetupdater.py @@ -5,7 +5,7 @@ import time from aareDBclient.models import PuckWithTellPosition from aare.common.models import SampleShortInfoList, SampleShortInfo, DewarAddress from config import BeamlineConfig -from aare.common.beamline import MXBeamline +from aare.common.beamline import MXBeamline, mx_beamline SLOT_IDENTIFIER = "X10SA" WS_URL = f"wss://mx-db-01.psi.ch/dispatcher/protected_router/tell_runner/ws/samples-spreadsheet/{SLOT_IDENTIFIER}" @@ -16,7 +16,7 @@ if not password: raise ValueError("The AAREDB_SHARED_PASSWORD environment variable is not set.") WS_HEADERS = [f"X-Shared-Password: {password}"] -beamline = MXBeamline.X10SA +beamline = mx_beamline().value() config = BeamlineConfig(bl=beamline) # Cache the current spreadsheet for change detection diff --git a/src/aare/daq/tellupdater.py b/src/aare/daq/tellupdater.py index a32678a8..cf9239c6 100644 --- a/src/aare/daq/tellupdater.py +++ b/src/aare/daq/tellupdater.py @@ -20,7 +20,7 @@ WS_HEADERS = [f"X-Shared-Password: {os.getenv('AAREDB_SHARED_PASSWORD')}"] print(WS_HEADERS) # Initialize TELL client and DB wrapper -beamline = mx_beamline() # Use your beamline enum/value +beamline = mx_beamline().value() # Use your beamline enum/value tell_client = TellClient(bl=beamline) aare_db = AareWrapper(bl=beamline)