Add script to check supported devices, update README (#242)

rytilahti · web-flow · commit 8a4068c62393 · 2021-10-25T10:17:35.000+03:00
* check_readme_vs_fixtures.py checks if a device with a fixture is listed in the README
* Add missing entries to README.md
diff --git a/README.md b/README.md
@@ -119,10 +119,13 @@ or the `parse_pcap.py` script contained inside the `devtools` directory.
 * HS105
 * HS107
 * HS110
+* KP105
 * KP115
+* KP401
 
 ### Power Strips
 
+* EP40
 * HS300
 * KP303
 * KP400
@@ -135,11 +138,13 @@ or the `parse_pcap.py` script contained inside the `devtools` directory.
 
 ### Bulbs
 
+* EP40
 * LB100
 * LB110
 * LB120
 * LB130
 * LB230
+* KL50
 * KL60
 * KL110
 * KL120
diff --git a/devtools/check_readme_vs_fixtures.py b/devtools/check_readme_vs_fixtures.py
@@ -0,0 +1,25 @@
+"""Script that checks if README.md is missing devices that have fixtures."""
+from kasa.tests.conftest import ALL_DEVICES, BULBS, DIMMERS, LIGHT_STRIPS, PLUGS, STRIPS
+
+readme = open("README.md").read()
+
+typemap = {
+    "light strips": LIGHT_STRIPS,
+    "bulbs": BULBS,
+    "plugs": PLUGS,
+    "strips": STRIPS,
+    "dimmers": DIMMERS,
+}
+
+
+def _get_device_type(dev, typemap):
+    for typename, devs in typemap.items():
+        if dev in devs:
+            return typename
+    else:
+        return "Unknown type"
+
+
+for dev in ALL_DEVICES:
+    if dev not in readme:
+        print(f"{dev} not listed in {_get_device_type(dev, typemap)}")
