innocuous-symmetry/core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow loading UniFi entities on config options change (#88762)

Browse Source 
Co-authored-by: Franck Nijhof <frenck@frenck.nl>
This commit is contained in:
Robert Svensson
2023-03-06 16:08:14 +01:00
committed by GitHub
parent ee6f969c2a
commit 9ff45ca013
4 changed files with 108 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
homeassistant/components/unifi/controller.py
View File

@@ -31,7 +31,10 @@ from homeassistant.helpers import (
entity_registry as er,
)
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.dispatcher import (
async_dispatcher_connect,
async_dispatcher_send,
)
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.entity_registry import async_entries_for_config_entry
from homeassistant.helpers.event import async_track_time_interval
@@ -108,6 +111,7 @@ class UniFiController:
self.load_config_entry_options()
self.entities = {}
self.known_objects: set[tuple[str, str]] = set()
def load_config_entry_options(self):
"""Store attributes to avoid property call overhead since they are called frequently."""
@@ -207,6 +211,7 @@ class UniFiController:
[
unifi_platform_entity(obj_id, self, description)
for obj_id in obj_ids
if (description.key, obj_id) not in self.known_objects
if description.allowed_fn(self, obj_id)
if description.supported_fn(self, obj_id)
]
@@ -221,6 +226,17 @@ class UniFiController:
api_handler.subscribe(async_create_entity, ItemEvent.ADDED)
@callback
def async_options_updated() -> None:
"""Load new entities based on changed options."""
async_add_unifi_entity(list(api_handler))
self.config_entry.async_on_unload(
async_dispatcher_connect(
self.hass, self.signal_options_update, async_options_updated
)
)
for description in descriptions:
async_load_entities(description)