Fixed saving profile to include devices and their configuration

1 files changed, 21 insertions, 6 deletions

diff --git a/devices/analog_input.py b/devices/analog_input.py
index e1b8b09..c9e5d2a 100644
--- a/devices/analog_input.py
+++ b/devices/analog_input.py
@@ -73,12 +73,13 @@ class AnalogInputDevice(BaseDevice):
         )
         super().__init__(info)
 
-        self._ni_device  = ni_device
-        self._ni_min_v   = ni_min_v
-        self._ni_max_v   = ni_max_v
-        self._ard_port   = ard_port
-        self._ard_baud   = ard_baud
-        self._last_error = ""
+        self._ni_device    = ni_device
+        self._ni_min_v     = ni_min_v
+        self._ni_max_v     = ni_max_v
+        self._ard_port     = ard_port
+        self._ard_baud     = ard_baud
+        self._num_channels = num_channels
+        self._last_error   = ""
 
         self._layer = self._make_layer(
             backend, simulate, ni_device, ni_min_v, ni_max_v,
@@ -170,6 +171,20 @@ class AnalogInputDevice(BaseDevice):
         # Named parameter / setpoint
         return self._layer.set_parameter(ch, value)
 
+    def get_save_config(self) -> dict:
+        return {
+            "device_type": self.DEVICE_TYPE,
+            "device_id":   self.info.device_id,
+            "num_channels": self._num_channels,
+            "simulate":    self.simulate,
+            "backend":     self.backend,
+            "ni_device":   self._ni_device,
+            "ni_min_v":    self._ni_min_v,
+            "ni_max_v":    self._ni_max_v,
+            "ard_port":    self._ard_port,
+            "ard_baud":    self._ard_baud,
+        }
+
     def get_config_widget(self) -> QWidget:
         return AnalogInputConfigWidget(self)