"""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)}")