evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add ConfigEntry template function (#78030)

Browse source 
* Add ConfigEntry template function

* Remove looking up entry_id by entry title
This commit is contained in:
Rami Mosleh 2022-09-29 13:41:59 +03:00 • committed by GitHub
parent d0ac1073a0
commit e7764b8bf1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
homeassistant/helpers/template.py
View file

@ -1063,6 +1063,14 @@ def integration_entities(hass: HomeAssistant, entry_name: str) -> Iterable[str]:
]
def entry_id(hass: HomeAssistant, entity_id: str) -> str | None:
"""Get an entry ID from an entity ID."""
entity_reg = entity_registry.async_get(hass)
if entity := entity_reg.async_get(entity_id):
return entity.config_entry_id
return None
def device_id(hass: HomeAssistant, entity_id_or_device_name: str) -> str | None:
"""Get a device ID from an entity ID or device name."""
entity_reg = entity_registry.async_get(hass)
@ -2072,6 +2080,9 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
self.globals["device_attr"] = hassfunction(device_attr)
self.globals["is_device_attr"] = hassfunction(is_device_attr)
self.globals["entry_id"] = hassfunction(entry_id)
self.filters["entry_id"] = pass_context(self.globals["entry_id"])
self.globals["device_id"] = hassfunction(device_id)
self.filters["device_id"] = pass_context(self.globals["device_id"])

24
tests/helpers/test_template.py
View file

@ -2458,6 +2458,30 @@ async def test_integration_entities(hass):
assert info.rate_limit is None
async def test_entry_id(hass):
"""Test entry_id function."""
config_entry = MockConfigEntry(domain="light", title="Some integration")
config_entry.add_to_hass(hass)
entity_registry = mock_registry(hass)
entity_entry = entity_registry.async_get_or_create(
"sensor", "test", "test", suggested_object_id="test", config_entry=config_entry
)
info = render_to_info(hass, "{{ 'sensor.fail' | entry_id }}")
assert_result_info(info, None)
assert info.rate_limit is None
info = render_to_info(hass, "{{ 56 | entry_id }}")
assert_result_info(info, None)
info = render_to_info(hass, "{{ 'not_a_real_entity_id' | entry_id }}")
assert_result_info(info, None)
info = render_to_info(hass, f"{{{{ entry_id('{entity_entry.entity_id}') }}}}")
assert_result_info(info, config_entry.entry_id)
assert info.rate_limit is None
async def test_device_id(hass):
"""Test device_id function."""
config_entry = MockConfigEntry(domain="light")