diff --git a/src/aare/daq/aaredb.py b/src/aare/daq/aaredb.py index 295c8f4e..460ba927 100644 --- a/src/aare/daq/aaredb.py +++ b/src/aare/daq/aaredb.py @@ -4,11 +4,11 @@ import json import os from typing import List, Optional -import aareDBclient +import aareDB import cv2 import numpy as np import requests -from aareDBclient import ( +from aareDB import ( SetTellPosition, SampleEventCreate, SampleEventType, @@ -44,16 +44,16 @@ class AareWrapper: bl: MXBeamline, host: str = "https://mx-db-01.psi.ch/dispatcher", ): - configuration = aareDBclient.Configuration(host=host) + configuration = aareDB.Configuration(host=host) configuration.verify_ssl = False # Disable SSL verification - self.client = aareDBclient.ApiClient(configuration) + self.client = aareDB.ApiClient(configuration) self.client.default_headers["X-Shared-Password"] = os.getenv("AAREDB_SHARED_PASSWORD") self.__host = host - self.__tell_api = aareDBclient.TellsRunnerApi(self.client) - self.__sample_api = aareDBclient.SamplesRunnerApi(self.client) - self.__proc_api = aareDBclient.ProcessingsRunnerApi(self.client) - self.__raster_api = aareDBclient.GridscanRunnerApi(self.client) + self.__tell_api = aareDB.TellsRunnerApi(self.client) + self.__sample_api = aareDB.SamplesRunnerApi(self.client) + self.__proc_api = aareDB.ProcessingsRunnerApi(self.client) + self.__raster_api = aareDB.GridscanRunnerApi(self.client) self.__bl = bl def set_pucks_beamline(self, input_list: List[PuckLoadedInfo]): @@ -73,7 +73,7 @@ class AareWrapper: print(ret) def create_manual_sample(self, s: SampleShortInfo): - from aareDBclient.models import ManualSampleCreate + from aareDB.models import ManualSampleCreate manual_sample = ManualSampleCreate( pgroup=s.user, diff --git a/src/aare/devices/tell_client.py b/src/aare/devices/tell_client.py index e047fc37..8f469639 100755 --- a/src/aare/devices/tell_client.py +++ b/src/aare/devices/tell_client.py @@ -15,9 +15,8 @@ from aare.common.models import ( SampleDewarAddress, #PuckInfo, ) -from aareDBclient import ( - PuckWithTellPosition, -) +from aareDB import PuckWithTellPosition + from aare.common.beamline import MXBeamline # noqa: F401 from aare.devices.pshell_client import PShellClient diff --git a/src/aare/gui/panels/manual_sample_panel.py b/src/aare/gui/panels/manual_sample_panel.py index fd39da58..bb0bf81a 100644 --- a/src/aare/gui/panels/manual_sample_panel.py +++ b/src/aare/gui/panels/manual_sample_panel.py @@ -1,6 +1,6 @@ from PySide6.QtCore import Signal, Slot from PySide6.QtWidgets import QWidget, QGridLayout, QLabel, QPushButton, QCheckBox, QLineEdit -from aareDBclient import DataCollectionParameters +from aareDB import DataCollectionParameters from aare.common.models import SampleShortInfo, DAQStatusModel from aare.gui.widgets.number_line_edit import NumberLineEdit