From 265ebd17a3f17ed8dc1e9bdede03ac8e323f1ab1 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sun, 16 Jan 2022 19:56:35 +0100
Subject: [PATCH] Move list to dict in iperf3 discovery (#64204)

---
 homeassistant/components/iperf3/__init__.py |  6 +++++-
 homeassistant/components/iperf3/sensor.py   | 20 ++++++++++++++++----
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/homeassistant/components/iperf3/__init__.py b/homeassistant/components/iperf3/__init__.py
index 8e7e365813e..ad7c6775107 100644
--- a/homeassistant/components/iperf3/__init__.py
+++ b/homeassistant/components/iperf3/__init__.py
@@ -118,7 +118,11 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
 
     hass.async_create_task(
         async_load_platform(
-            hass, SENSOR_DOMAIN, DOMAIN, conf[CONF_MONITORED_CONDITIONS], config
+            hass,
+            SENSOR_DOMAIN,
+            DOMAIN,
+            {CONF_MONITORED_CONDITIONS: conf[CONF_MONITORED_CONDITIONS]},
+            config,
         )
     )
 
diff --git a/homeassistant/components/iperf3/sensor.py b/homeassistant/components/iperf3/sensor.py
index 4e102b69782..efdaea5b4f5 100644
--- a/homeassistant/components/iperf3/sensor.py
+++ b/homeassistant/components/iperf3/sensor.py
@@ -1,9 +1,13 @@
 """Support for Iperf3 sensors."""
+from __future__ import annotations
+
 from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
-from homeassistant.const import ATTR_ATTRIBUTION
-from homeassistant.core import callback
+from homeassistant.const import ATTR_ATTRIBUTION, CONF_MONITORED_CONDITIONS
+from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.restore_state import RestoreEntity
+from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
 from . import ATTR_VERSION, DATA_UPDATED, DOMAIN as IPERF3_DOMAIN, SENSOR_TYPES
 
@@ -16,13 +20,21 @@ ATTR_REMOTE_HOST = "Remote Server"
 ATTR_REMOTE_PORT = "Remote Port"
 
 
-async def async_setup_platform(hass, config, async_add_entities, discovery_info):
+async def async_setup_platform(
+    hass: HomeAssistant,
+    config: ConfigType,
+    async_add_entities: AddEntitiesCallback,
+    discovery_info: DiscoveryInfoType | None = None,
+) -> None:
     """Set up the Iperf3 sensor."""
+    if not discovery_info:
+        return
+
     entities = [
         Iperf3Sensor(iperf3_host, description)
         for iperf3_host in hass.data[IPERF3_DOMAIN].values()
         for description in SENSOR_TYPES
-        if description.key in discovery_info
+        if description.key in discovery_info[CONF_MONITORED_CONDITIONS]
     ]
     async_add_entities(entities, True)