From 1fd09bee82dc9ff6be4717782a275e3f394ddea6 Mon Sep 17 00:00:00 2001
From: CPD <CPD@TUZEWSI-2LN203M.ads.mwn.de>
Date: Mon, 17 Mar 2025 18:15:20 +0100
Subject: [PATCH] Add support for a usb power switch

---
 cpdctrl/power_switch_device/__init__.py       |  0
 cpdctrl/power_switch_device/base.py           | 17 ++++++++
 cpdctrl/power_switch_device/impl/__init__.py  |  0
 .../impl/cleware_switch.py                    | 40 +++++++++++++++++++
 4 files changed, 57 insertions(+)
 create mode 100644 cpdctrl/power_switch_device/__init__.py
 create mode 100644 cpdctrl/power_switch_device/base.py
 create mode 100644 cpdctrl/power_switch_device/impl/__init__.py
 create mode 100644 cpdctrl/power_switch_device/impl/cleware_switch.py

diff --git a/cpdctrl/power_switch_device/__init__.py b/cpdctrl/power_switch_device/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/cpdctrl/power_switch_device/base.py b/cpdctrl/power_switch_device/base.py
new file mode 100644
index 0000000..058f699
--- /dev/null
+++ b/cpdctrl/power_switch_device/base.py
@@ -0,0 +1,17 @@
+from abc import ABC, abstractmethod
+
+class PowerSwitchDevice(ABC):
+    @abstractmethod
+    def status(self) -> bool:
+        pass
+    @abstractmethod   
+    def on(self):
+        pass
+
+    @abstractmethod
+    def off(self):
+        pass
+    
+    @abstractmethod
+    def __str__(self):
+        pass
\ No newline at end of file
diff --git a/cpdctrl/power_switch_device/impl/__init__.py b/cpdctrl/power_switch_device/impl/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/cpdctrl/power_switch_device/impl/cleware_switch.py b/cpdctrl/power_switch_device/impl/cleware_switch.py
new file mode 100644
index 0000000..218ee1f
--- /dev/null
+++ b/cpdctrl/power_switch_device/impl/cleware_switch.py
@@ -0,0 +1,40 @@
+from ..base import PowerSwitchDevice
+import os
+from subprocess import Popen, PIPE
+
+class ClewareSwitch(PowerSwitchDevice):
+    def __init__(self, usbswitchcmd_exe_path: str):
+        super().__init__()
+        self.exe = usbswitchcmd_exe_path
+
+    def run_cmd(self, cmd) -> (int, str, str):
+        args = [self.exe, cmd]
+        p = Popen(args, stdout=PIPE, stderr=PIPE)
+        p.wait()
+        return p.returncode, p.stdout.read().decode("utf-8").strip("\n"), p.stderr.read().decode("utf-8").strip("\n")
+
+    def status(self) -> bool:
+        cmd = "-r"
+        ret, out, err = self.run_cmd(cmd)
+        print(f"status: ret: {ret}, out: '{out}', err: '{err}'")
+        if out == "1":
+            return True
+        elif out == "0":
+            return False
+        else:
+            raise RuntimeError(f"Failed to read USB switch status.\nCommand: '{cmd}'\nCode: {ret}, stdout: '{out}', stderr: '{err}'")
+
+    def on(self):
+        cmd = "1"
+        ret, out, err = self.run_cmd(cmd)
+        if ret != 1 or out or err:  # yes, it actually returns 1 when it turns on ._.
+            raise RuntimeError(f"Failed to turn on USB switch.\nCommand: '{cmd}'\nCode: {ret}, stdout: '{out}', stderr: '{err}'")
+
+    def off(self):
+        cmd = "0"
+        ret, out, err = self.run_cmd(cmd)
+        if ret != 0 or out or err:
+            raise RuntimeError(f"Failed to turn on USB switch.\nCommand: '{cmd}'\nCode: {ret}, stdout: '{out}', stderr: '{err}'")
+
+    def __str__(self):
+        return f"Cleware USB Switch"
\ No newline at end of file