1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
"""Process the device status JSON file to generate a tree of a device status."""
import json
from pathlib import Path
import re
import sys
from pysmartthings import CAPABILITY_ATTRIBUTES, Attribute, Capability
from treelib import Tree
def main() -> int: # noqa: PLR0912
"""Run the script."""
if len(sys.argv) != 2:
print("Usage: python process_device_status.py <filename>")
return 1
filename = sys.argv[1]
print(f"Processing {filename}")
with Path(filename).open(encoding="utf-8") as file:
data = json.load(file)
components = data["components"]
tree = Tree()
found_capabilities = {}
found_attributes = {}
missing_attribute_mapping: dict[str, list[str]] = {}
tree.create_node(filename, "root")
for component_name, capabilities in components.items(): # pylint: disable=too-many-nested-blocks
tree.create_node(component_name, component_name, parent="root")
for capability_name, attributes in capabilities.items():
if capability_name not in Capability:
found_capabilities[capability_name] = (
re.sub(r"(?<!^)(?=[A-Z])", "_", capability_name)
.upper()
.replace(".", "_")
.replace("SAMSUNGCE", "SAMSUNG_CE")
)
missing_attribute_mapping[capability_name] = []
tree.create_node(
capability_name,
f"{component_name}-{capability_name}",
parent=component_name,
)
for attribute in attributes:
attribute_name: str
if attribute not in Attribute:
found_attributes[attribute] = attribute_name = re.sub(
r"(?<!^)(?=[A-Z])", "_", attribute
).upper()
else:
attribute_name = Attribute(attribute).name
attribute_name = f"Attribute.{attribute_name}"
if capability_name in CAPABILITY_ATTRIBUTES:
if attribute not in CAPABILITY_ATTRIBUTES[capability_name]:
if capability_name not in missing_attribute_mapping:
missing_attribute_mapping[capability_name] = []
if (
attribute_name
not in missing_attribute_mapping[capability_name]
):
missing_attribute_mapping[capability_name].append(
attribute_name
)
else:
if capability_name not in missing_attribute_mapping:
missing_attribute_mapping[capability_name] = []
if attribute_name not in missing_attribute_mapping[capability_name]:
missing_attribute_mapping[capability_name].append(
attribute_name
)
tree.create_node(
attribute,
f"{component_name}-{capability_name}-{attribute}",
parent=f"{component_name}-{capability_name}",
)
print(tree.show(stdout=False))
if found_capabilities:
print("\nFound capabilities:")
for capability_name, slug in found_capabilities.items():
print(f'{slug} = "{capability_name}"')
if found_attributes:
print("\nFound attributes:")
for attribute_name, slug in found_attributes.items():
print(f'{slug} = "{attribute_name}"')
print(json.dumps(missing_attribute_mapping, indent=4))
return 0
if __name__ == "__main__":
sys.exit(main())
|
