jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-14 00:37:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Inline building entity registry dict (#74378)

Browse Source
This commit is contained in:
J. Nick Koston 2022-07-04 00:36:42 -05:00 committed by GitHub
parent b62c0dcb32
commit 810b2a2bd6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
homeassistant/components/config/entity_registry.py
View File

@ -29,7 +29,22 @@ def websocket_list_entities(hass, connection, msg):
registry = er.async_get(hass) registry = er.async_get(hass)
connection.send_message( connection.send_message(
websocket_api.result_message( websocket_api.result_message(
msg["id"], [_entry_dict(entry) for entry in registry.entities.values()] msg["id"],
[
{
"area_id": entry.area_id,
"config_entry_id": entry.config_entry_id,
"device_id": entry.device_id,
"disabled_by": entry.disabled_by,
"entity_category": entry.entity_category,
"entity_id": entry.entity_id,
"hidden_by": entry.hidden_by,
"icon": entry.icon,
"name": entry.name,
"platform": entry.platform,
}
for entry in registry.entities.values()
],
) )
) )
@ -196,7 +211,7 @@ def websocket_remove_entity(hass, connection, msg):
@callback @callback
def _entry_dict(entry): def _entry_ext_dict(entry):
"""Convert entry to API format.""" """Convert entry to API format."""
return { return {
"area_id": entry.area_id, "area_id": entry.area_id,
@ -209,19 +224,12 @@ def _entry_dict(entry):
"icon": entry.icon, "icon": entry.icon,
"name": entry.name, "name": entry.name,
"platform": entry.platform, "platform": entry.platform,
"capabilities": entry.capabilities,
"device_class": entry.device_class,
"has_entity_name": entry.has_entity_name,
"options": entry.options,
"original_device_class": entry.original_device_class,
"original_icon": entry.original_icon,
"original_name": entry.original_name,
"unique_id": entry.unique_id,
} }
@callback
def _entry_ext_dict(entry):
"""Convert entry to API format."""
data = _entry_dict(entry)
data["capabilities"] = entry.capabilities
data["device_class"] = entry.device_class
data["has_entity_name"] = entry.has_entity_name
data["options"] = entry.options
data["original_device_class"] = entry.original_device_class
data["original_icon"] = entry.original_icon
data["original_name"] = entry.original_name
data["unique_id"] = entry.unique_id
return data