add group_input

Author: inauenma

src/abb_robot_client/rws.py

@@ -470,6 +470,26 @@ def set_analog_io(self, signal: str, value: Union[int, float], network: str = "L
         payload = {"mode": "value", "lvalue": value}
         _res = self._do_post("rw/iosystem/signals/" + network + "/" + unit + "/" + signal + "?action=set", payload)
 
+    def get_group_io(self, signal: str, network: str = "", unit: str = "") -> int:
+        """
+        This is the same as get_digital_io(). Get the value of a group IO signal.
+        """
+        return self.get_digital_io(signal, network, unit)
+
+    def set_group_io(self, signal: str, value: int, network: str = "", unit: str = ""):
+        """
+        Set the value of an group IO signal.
+        WARNNING: This will just write an int to the group, make sure to convert the value to the correct bit pattern first.
+
+        :param value: The value of the signal.
+        :param signal: The name of the signal
+        :param network: The network the signal is on. The default `Local` will work for most signals.
+        :param unit: The drive unit of the signal. The default `DRV_1` will work for most signals.
+        """
+        lvalue = str(value)
+        payload = {"lvalue": lvalue}
+        _res = self._do_post("rw/iosystem/signals/" + network + "/" + unit + "/" + signal + "?action=set", payload)   
+
     def get_rapid_variables(self, task: str = "T_ROB1") -> List[str]:
         """
         Get a list of the persistent variables in a task

src/abb_robot_client/rws2.py

@@ -408,6 +408,30 @@ def set_analog_io(self, signal: str, value: Union[int, float], network: str = ""
         payload = {"mode": "value", "lvalue": value}
         _res = self._do_post(f"rw/iosystem/signals/{network}/{unit}/{signal}/set-value?mastership=implicit", payload)
 
+    def get_group_io(self, signal: str, network: str = "", unit: str = "") -> int:
+        """
+        This is the same as get_digital_io(). Get the value of a group IO signal.
+        """
+        return self.get_digital_io(signal, network, unit)
+
+    def set_group_io(self, signal: str, value: int, network: str = "", unit: str = ""):
+        """
+        Set the value of an group IO signal.
+        WARNNING: This will just write an int to the group, make sure to convert the value to the correct bit pattern first.
+
+        :param value: The value of the signal.
+        :param signal: The name of the signal
+        :param network: The network the signal is on. The default `Local` will work for most signals.
+        :param unit: The drive unit of the signal. The default `DRV_1` will work for most signals.
+        """
+        lvalue = value
+        payload = {"lvalue": lvalue}
+        if network and unit:
+            url = f"rw/iosystem/signals/{network}/{unit}/{signal}/set-value?mastership=implicit"
+        else:
+            url = f"rw/iosystem/signals/{signal}/set-value?mastership=implicit"
+        _res = self._do_post(url, payload)    
+    
     # def get_rapid_variables(self, task: str="T_ROB1") -> List[str]:
     #     """
     #     Get a list of the persistent variables in a task

src/abb_robot_client/rws_mock.py

@@ -164,7 +164,13 @@ def get_analog_io(self, signal: str, network: str = "", unit: str = "") -> float
 
     def set_analog_io(self, signal: str, value: int | float, network: str = "", unit: str = "") -> None:
         self._aio[signal] = float(value)
+    
+    def get_group_io(self, signal: str, network: str = "", unit: str = "") -> int:
+        return int(self._dio.get(signal, 0))
 
+    def set_group_io(self, signal: str, value: bool | int, network: str = "", unit: str = "") -> None:
+        self._dio[signal] = value
+        
     # RAPID variables
     def _qualify_var(self, var: str, task: str) -> str:
         return f"{task}/{var}" if task else var