Merge pull request #863 from kostrolenk/feature/tplink_power_strip

driver/power: add network power driver for TP-Link power strips

Author: Emantor

CHANGES.rst

@@ -3,7 +3,7 @@ Release 0.5.0 (unreleased)
 
 New Features in 0.5.0
 ~~~~~~~~~~~~~~~~~~~~~
-- Support for Eaton ePDU added, and can be used as a NetworkPowerPort.
+- Support for Eaton ePDU and TP-Link power strips added, either can be used as a NetworkPowerPort.
 - Consider a combination of multiple "lg_feature" markers instead of
   considering only the closest marker.
 - The labgrid client SSH command is now able to instantiate the SSHDriver when

dev-requirements.txt

@@ -16,3 +16,4 @@ psutil==5.8.0
 -r pyvisa-requirements.txt
 -r vxi11-requirements.txt
 -r mqtt-requirements.txt
+-r kasa-requirements.txt

kasa-requirements.txt

@@ -0,0 +1 @@
+python-kasa==0.4.0

labgrid/driver/power/tplink.py

@@ -0,0 +1,34 @@
+""" Tested with TP Link KP303, and should be compatible with any strip supported by kasa """
+
+import asyncio
+from kasa import SmartStrip
+
+
+async def _power_set(host, port, index, value):
+    """We embed the coroutines in an `async` function to minimise calls to `asyncio.run`"""
+    assert port is None
+    index = int(index)
+    strip = SmartStrip(host)
+    await strip.update()
+    assert (
+        len(strip.children) > index
+    ), "Trying to access non-existant plug socket on strip"
+    if value is True:
+        await strip.children[index].turn_on()
+    elif value is False:
+        await strip.children[index].turn_off()
+
+
+def power_set(host, port, index, value):
+    asyncio.run(_power_set(host, port, index, value))
+
+
+def power_get(host, port, index):
+    assert port is None
+    index = int(index)
+    strip = SmartStrip(host)
+    asyncio.run(strip.update())
+    assert (
+        len(strip.children) > index
+    ), "Trying to access non-existant plug socket on strip"
+    return strip.children[index].is_on

tests/test_powerdriver.py

@@ -166,6 +166,7 @@ def test_import_backends(self):
         import labgrid.driver.power.digipower
         import labgrid.driver.power.gude
         import labgrid.driver.power.gude24
+        import labgrid.driver.power.tplink
         import labgrid.driver.power.netio
         import labgrid.driver.power.netio_kshell
         import labgrid.driver.power.rest