tuya-local/util/match_against.py

"""Find matching devices for the supplied dp list"""

import json
import sys

from custom_components.tuya_local.helpers.device_config import _typematch

from .common_funcs import FakeDevice, load_config, make_sample_dps


def main() -> int:
    rest = " ".join(sys.argv[2:])
    config = load_config(rest)
    if config is None:
        dps = json.loads(rest)
    else:
        dps = make_sample_dps(config)

    config = load_config(sys.argv[1])
    if config is None:
        print(f"No config could be loaded for {sys.argv[1]}")
        return 1
    device = FakeDevice(dps)
    unseen = set(dps.keys())
    for entity in config.all_entities():
        print(f"{entity.config_id}:")
        for dp in entity.dps():
            unseen.discard(dp.id)
            if dp.id not in dps.keys():
                print(f"   {dp.name} missing from data")
                if not dp.optional:
                    print(f">> dp {dp.id} is REQUIRED!!!!")
            elif not _typematch(dp.type, dps.get(dp.id)):
                print(
                    f">> {dp.name} type MISMATCH, expected {dp.type.__name__}, got {dps.get(dp.id)}!!!"
                )
            else:
                values = dp.values(device)
                if values:
                    values = f" from {values}"
                else:
                    values = ""
                print(f"   {dp.name}: {dp.get_value(device)}{values}")
    for dp in unseen:
        print(f"   Extra dp {dp} in data")


if __name__ == "__main__":
    sys.exit(main())