Fixed saving profile to include devices and their configuration

1 files changed, 20 insertions, 5 deletions

diff --git a/devices/digital_io.py b/devices/digital_io.py
index 3d24192..96c6ca5 100644
--- a/devices/digital_io.py
+++ b/devices/digital_io.py
@@ -40,11 +40,13 @@ class DigitalIODevice(BaseDevice):
         ard_port:     str  = "COM3",
         ard_baud:     int  = 115200,
     ):
-        self.simulate  = simulate
-        self.backend   = backend
-        self._ni_device = ni_device
-        self._ard_port  = ard_port
-        self._ard_baud  = ard_baud
+        self.simulate     = simulate
+        self.backend      = backend
+        self._ni_device   = ni_device
+        self._ard_port    = ard_port
+        self._ard_baud    = ard_baud
+        self._num_inputs  = num_inputs
+        self._num_outputs = num_outputs
 
         channels = []
         for i in range(num_inputs):
@@ -252,6 +254,19 @@ class DigitalIODevice(BaseDevice):
         if was_running:
             self.connect()
 
+    def get_save_config(self) -> dict:
+        return {
+            "device_type": self.DEVICE_TYPE,
+            "device_id":   self.info.device_id,
+            "num_inputs":  self._num_inputs,
+            "num_outputs": self._num_outputs,
+            "simulate":    self.simulate,
+            "backend":     self.backend,
+            "ni_device":   self._ni_device,
+            "ard_port":    self._ard_port,
+            "ard_baud":    self._ard_baud,
+        }
+
     def get_config_widget(self) -> QWidget:
         return DigitalIOConfigWidget(self)