starred/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-09-24 23:41:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use get_device_and_config_entry_for_domain in octoprint (#180102)

Browse Source
This commit is contained in:
Erik Montnemery
2026-08-25 03:13:36 -04:00
committed by GitHub
parent b1106a56d2
commit 89893cd160
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
homeassistant/components/octoprint/services.py
+5 -6
View File
@@ -29,13 +29,12 @@ def async_get_client_for_service_call(
) -> OctoprintClient:
"""Get the client related to a service call (by device ID)."""
device_id = call.data[CONF_DEVICE_ID]
device_registry = dr.async_get(hass)
if device_entry := device_registry.async_get(device_id):
for entry_id in device_entry.config_entries:
if entry := hass.config_entries.async_get_entry(entry_id):
if entry.domain == DOMAIN and entry.state is ConfigEntryState.LOADED:
return cast(OctoprintConfigEntry, entry).runtime_data.octoprint
_, config_entry = dr.async_get_device_and_config_entry_for_domain(
hass, device_id, domain=DOMAIN
)
if config_entry is not None and config_entry.state is ConfigEntryState.LOADED:
return cast(OctoprintConfigEntry, config_entry).runtime_data.octoprint
raise ServiceValidationError(
translation_domain=DOMAIN,