python-kasa/devtools/check_readme_vs_fixtures.py

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

with open("README.md") as f:
    readme = f.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)}")