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

Don't try to re-add existing Z-Wave entities (#17995)

Browse Source 
* Keep track of created entities

* lint

* Update tests
This commit is contained in:
Charles Garwood 2018-11-01 14:38:23 -04:00 committed by Paulus Schoutsen
parent 23290fa6ee
commit bcea3a9cba
2 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
homeassistant/components/zwave/__init__.py
View File

@ -223,10 +223,11 @@ async def async_setup_platform(hass, config, async_add_entities,
if discovery_info is None or DATA_NETWORK not in hass.data: if discovery_info is None or DATA_NETWORK not in hass.data:
return False return False
device = hass.data[DATA_DEVICES].pop( device = hass.data[DATA_DEVICES].get(
discovery_info[const.DISCOVERY_DEVICE], None) discovery_info[const.DISCOVERY_DEVICE], None)
if device is None: if device is None:
return False return False
async_add_entities([device]) async_add_entities([device])
return True return True
@ -340,6 +341,9 @@ async def async_setup_entry(hass, config_entry):
entity = ZWaveNodeEntity(node, network) entity = ZWaveNodeEntity(node, network)
def _add_node_to_component(): def _add_node_to_component():
if hass.data[DATA_DEVICES].get(entity.unique_id):
return
name = node_name(node) name = node_name(node)
generated_id = generate_entity_id(DOMAIN + '.{}', name, []) generated_id = generate_entity_id(DOMAIN + '.{}', name, [])
node_config = device_config.get(generated_id) node_config = device_config.get(generated_id)
@ -348,6 +352,8 @@ async def async_setup_entry(hass, config_entry):
"Ignoring node entity %s due to device settings", "Ignoring node entity %s due to device settings",
generated_id) generated_id)
return return
hass.data[DATA_DEVICES][entity.unique_id] = entity
component.add_entities([entity]) component.add_entities([entity])
if entity.unique_id: if entity.unique_id:
@ -912,15 +918,12 @@ class ZWaveDeviceEntityValues():
self._entity = device self._entity = device
dict_id = id(self)
@callback @callback
def _on_ready(sec): def _on_ready(sec):
_LOGGER.info( _LOGGER.info(
"Z-Wave entity %s (node_id: %d) ready after %d seconds", "Z-Wave entity %s (node_id: %d) ready after %d seconds",
device.name, self._node.node_id, sec) device.name, self._node.node_id, sec)
self._hass.async_add_job(discover_device, component, device, self._hass.async_add_job(discover_device, component, device)
dict_id)
@callback @callback
def _on_timeout(sec): def _on_timeout(sec):
@ -928,22 +931,25 @@ class ZWaveDeviceEntityValues():
"Z-Wave entity %s (node_id: %d) not ready after %d seconds, " "Z-Wave entity %s (node_id: %d) not ready after %d seconds, "
"continuing anyway", "continuing anyway",
device.name, self._node.node_id, sec) device.name, self._node.node_id, sec)
self._hass.async_add_job(discover_device, component, device, self._hass.async_add_job(discover_device, component, device)
dict_id)
async def discover_device(component, device, dict_id): async def discover_device(component, device):
"""Put device in a dictionary and call discovery on it.""" """Put device in a dictionary and call discovery on it."""
self._hass.data[DATA_DEVICES][dict_id] = device if self._hass.data[DATA_DEVICES].get(device.unique_id):
return
self._hass.data[DATA_DEVICES][device.unique_id] = device
if component in SUPPORTED_PLATFORMS: if component in SUPPORTED_PLATFORMS:
async_dispatcher_send( async_dispatcher_send(
self._hass, 'zwave_new_{}'.format(component), device) self._hass, 'zwave_new_{}'.format(component), device)
else: else:
await discovery.async_load_platform( await discovery.async_load_platform(
self._hass, component, DOMAIN, self._hass, component, DOMAIN,
{const.DISCOVERY_DEVICE: dict_id}, self._zwave_config) {const.DISCOVERY_DEVICE: device.unique_id},
self._zwave_config)
if device.unique_id: if device.unique_id:
self._hass.add_job(discover_device, component, device, dict_id) self._hass.add_job(discover_device, component, device)
else: else:
self._hass.add_job(check_has_unique_id, device, _on_ready, self._hass.add_job(check_has_unique_id, device, _on_ready,
_on_timeout, self._hass.loop) _on_timeout, self._hass.loop)

4
tests/components/zwave/test_init.py
View File

@ -584,7 +584,7 @@ class TestZWaveDeviceEntityValues(unittest.TestCase):
assert args[0] == self.hass assert args[0] == self.hass
assert args[1] == 'mock_component' assert args[1] == 'mock_component'
assert args[2] == 'zwave' assert args[2] == 'zwave'
assert args[3] == {const.DISCOVERY_DEVICE: id(values)} assert args[3] == {const.DISCOVERY_DEVICE: mock_device.unique_id}
assert args[4] == self.zwave_config assert args[4] == self.zwave_config
discovery.async_load_platform.reset_mock() discovery.async_load_platform.reset_mock()
@ -646,7 +646,7 @@ class TestZWaveDeviceEntityValues(unittest.TestCase):
assert args[0] == self.hass assert args[0] == self.hass
assert args[1] == 'mock_component' assert args[1] == 'mock_component'
assert args[2] == 'zwave' assert args[2] == 'zwave'
assert args[3] == {const.DISCOVERY_DEVICE: id(values)} assert args[3] == {const.DISCOVERY_DEVICE: mock_device.unique_id}
assert args[4] == self.zwave_config assert args[4] == self.zwave_config
assert not self.primary.enable_poll.called assert not self.primary.enable_poll.called