Merge pull request #1645 from lsa-cs/add-tcpdu-upstream

driver/power: Add support for tinycontrol tcPDU

Author: Bastian-Krause

doc/configuration.rst

@@ -248,6 +248,10 @@ Currently available are:
   It was tested on the *6G10A v2* model.
   `Manual <https://tinycontrol.pl/media/documents/manual_IP_Power_Socket__6G10A_v2_LANLIS-010-015_En-1.pdf>`__
 
+``tinycontrol_tcpdu``
+  Controls a Tinycontrol tcPDU via HTTP.
+  See the `documentation <https://docs.tinycontrol.pl/en/tcpdu/api/commands/>`__
+
 ``poe_mib``
   Controls PoE switches using the PoE SNMP administration MiBs.
 

labgrid/driver/power/tinycontrol_tcpdu.py

@@ -0,0 +1,33 @@
+""" A driver to control the Tinycontrol tcPDU
+    Reference: https://docs.tinycontrol.pl/en/tcpdu/api/commands
+
+    Example configuration to use port #3 on a device with URL 'http://172.17.180.53/'
+
+    NetworkPowerPort:
+      model: tinycontrol_tcpdu
+      host: 'http://172.17.180.53'
+      index: 3
+"""
+
+from urllib.parse import urljoin
+
+import requests
+
+
+def power_set(host, port, index, value):
+    assert port is None
+
+    index = int(index)
+    value = 1 if value else 0
+    r = requests.get(urljoin(host, f"/api/v1/save/?out{index}={value}"))
+    r.raise_for_status()
+
+
+def power_get(host, port, index):
+    assert port is None
+
+    index = int(index)
+    r = requests.get(urljoin(host, "/api/v1/read/status/?outValues"))
+    r.raise_for_status()
+    json_decoded = r.json()
+    return json_decoded[f'out{index}'] == 1

tests/test_powerdriver.py

@@ -290,6 +290,7 @@ def test_import_backends(self):
         import labgrid.driver.power.eg_pms2_network
         import labgrid.driver.power.shelly_gen1
         import labgrid.driver.power.ubus
+        import labgrid.driver.power.tinycontrol_tcpdu
 
     def test_import_backend_eaton(self):
         pytest.importorskip("pysnmp")