Adjust config entry state checks in reolink (#138909)

pull/138916/head
Erik Montnemery 2025-02-20 08:58:37 +01:00 committed by GitHub
parent 0949f7d0ba
commit 2f7a8b4d9d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 5 deletions

View File

@ -18,7 +18,6 @@ from homeassistant.components.media_source import (
Unresolvable, Unresolvable,
) )
from homeassistant.components.stream import create_stream from homeassistant.components.stream import create_stream
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
@ -151,9 +150,7 @@ class ReolinkVODMediaSource(MediaSource):
entity_reg = er.async_get(self.hass) entity_reg = er.async_get(self.hass)
device_reg = dr.async_get(self.hass) device_reg = dr.async_get(self.hass)
for config_entry in self.hass.config_entries.async_entries(DOMAIN): for config_entry in self.hass.config_entries.async_loaded_entries(DOMAIN):
if config_entry.state != ConfigEntryState.LOADED:
continue
channels: list[str] = [] channels: list[str] = []
host = config_entry.runtime_data.host host = config_entry.runtime_data.host
entities = er.async_entries_for_config_entry( entities = er.async_entries_for_config_entry(

View File

@ -40,7 +40,7 @@ def async_setup_services(hass: HomeAssistant) -> None:
if ( if (
config_entry is None config_entry is None
or device is None or device is None
or config_entry.state == ConfigEntryState.NOT_LOADED or config_entry.state != ConfigEntryState.LOADED
): ):
raise ServiceValidationError( raise ServiceValidationError(
translation_domain=DOMAIN, translation_domain=DOMAIN,