Improve integration type hints for diagnostics (#85526)

homeassistant/components/accuweather/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for AccuWeather."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE
@@ -14,7 +16,7 @@ TO_REDACT = {CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: AccuWeatherDataUpdateCoordinator = hass.data[DOMAIN][
         config_entry.entry_id

homeassistant/components/airly/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Airly."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
@@ -19,7 +21,7 @@ TO_REDACT = {CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE, CONF_UNIQUE_ID}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: AirlyDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
 

homeassistant/components/brother/diagnostics.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 from dataclasses import asdict
+from typing import Any
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
@@ -12,7 +13,7 @@ from .const import DATA_CONFIG_ENTRY, DOMAIN
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: BrotherDataUpdateCoordinator = hass.data[DOMAIN][DATA_CONFIG_ENTRY][
         config_entry.entry_id

homeassistant/components/co2signal/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for CO2Signal."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_API_KEY
@@ -13,7 +15,7 @@ TO_REDACT = {CONF_API_KEY}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: CO2SignalCoordinator = hass.data[DOMAIN][config_entry.entry_id]
 

homeassistant/components/coinbase/diagnostics.py

@@ -1,5 +1,7 @@
 """Diagnostics support for Coinbase."""
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_API_KEY, CONF_API_TOKEN, CONF_ID
@@ -20,7 +22,7 @@ TO_REDACT = {
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     instance: CoinbaseData = hass.data[DOMAIN][entry.entry_id]
 

homeassistant/components/environment_canada/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Environment Canada."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE
@@ -13,7 +15,7 @@ TO_REDACT = {CONF_LATITUDE, CONF_LONGITUDE}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinators = hass.data[DOMAIN][config_entry.entry_id]
     weather_coord = coordinators["weather_coordinator"]

homeassistant/components/evil_genius_labs/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Evil Genius Labs."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
@@ -13,7 +15,7 @@ TO_REDACT = {"wiFiSsidDefault", "wiFiSSID"}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: EvilGeniusUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
 

homeassistant/components/fritz/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for AVM FRITZ!Box."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
@@ -14,7 +16,7 @@ TO_REDACT = {CONF_USERNAME, CONF_PASSWORD}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     avm_wrapper: AvmWrapper = hass.data[DOMAIN][entry.entry_id]
 

homeassistant/components/fritzbox/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for AVM Fritz!Smarthome."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
@@ -14,7 +16,7 @@ TO_REDACT = {CONF_USERNAME, CONF_PASSWORD}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     data: dict = hass.data[DOMAIN][entry.entry_id]
     coordinator: FritzboxDataUpdateCoordinator = data[CONF_COORDINATOR]

homeassistant/components/gios/diagnostics.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 from dataclasses import asdict
+from typing import Any
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
@@ -12,7 +13,7 @@ from .const import DOMAIN
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: GiosDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
 

homeassistant/components/knx/diagnostics.py

@@ -29,7 +29,7 @@ TO_REDACT = {
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     diag: dict[str, Any] = {}
     knx_module = hass.data[DOMAIN]

homeassistant/components/nam/diagnostics.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 from dataclasses import asdict
+from typing import Any
 
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
@@ -16,7 +17,7 @@ TO_REDACT = {CONF_PASSWORD, CONF_USERNAME}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: NAMDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
 

homeassistant/components/nanoleaf/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Nanoleaf."""
 from __future__ import annotations
 
+from typing import Any
+
 from aionanoleaf import Nanoleaf
 
 from homeassistant.components.diagnostics import async_redact_data
@@ -14,7 +16,7 @@ from .const import DOMAIN
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant,
     config_entry: ConfigEntry,
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     device: Nanoleaf = hass.data[DOMAIN][config_entry.entry_id].device
 

homeassistant/components/nest/diagnostics.py

@@ -41,7 +41,7 @@ def _async_get_nest_devices(
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     nest_devices = _async_get_nest_devices(hass, config_entry)
     if not nest_devices:
@@ -64,7 +64,7 @@ async def async_get_device_diagnostics(
     hass: HomeAssistant,
     config_entry: ConfigEntry,
     device: DeviceEntry,
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a device."""
     nest_devices = _async_get_nest_devices(hass, config_entry)
     nest_device_id = next(iter(device.identifiers))[1]

homeassistant/components/netatmo/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Netatmo."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
@@ -30,7 +32,7 @@ TO_REDACT = {
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     data_handler: NetatmoDataHandler = hass.data[DOMAIN][config_entry.entry_id][
         DATA_HANDLER

homeassistant/components/nextdns/diagnostics.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 from dataclasses import asdict
+from typing import Any
 
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
@@ -24,7 +25,7 @@ TO_REDACT = {CONF_API_KEY, CONF_PROFILE_ID, CONF_UNIQUE_ID}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinators = hass.data[DOMAIN][config_entry.entry_id]
 

homeassistant/components/rainforest_eagle/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Eagle."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
@@ -13,7 +15,7 @@ TO_REDACT = {CONF_CLOUD_ID, CONF_INSTALL_CODE}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: EagleDataCoordinator = hass.data[DOMAIN][config_entry.entry_id]
 

homeassistant/components/screenlogic/diagnostics.py

@@ -1,5 +1,7 @@
 """Diagnostics for Screenlogic."""
 
+from typing import Any
+
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 
@@ -9,7 +11,7 @@ from .const import DOMAIN
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     coordinator: ScreenlogicDataUpdateCoordinator = hass.data[DOMAIN][
         config_entry.entry_id

homeassistant/components/shelly/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Shelly."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.bluetooth import async_scanner_by_source
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
@@ -15,7 +17,7 @@ TO_REDACT = {CONF_USERNAME, CONF_PASSWORD}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     shelly_entry_data = get_entry_data(hass)[entry.entry_id]
 

homeassistant/components/tibber/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Tibber."""
 from __future__ import annotations
 
+from typing import Any
+
 import tibber
 
 from homeassistant.config_entries import ConfigEntry
@@ -11,7 +13,7 @@ from .const import DOMAIN
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     tibber_connection: tibber.Tibber = hass.data[DOMAIN]
 

homeassistant/components/tractive/diagnostics.py

@@ -1,6 +1,8 @@
 """Diagnostics support for Tractive."""
 from __future__ import annotations
 
+from typing import Any
+
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
@@ -13,7 +15,7 @@ TO_REDACT = {CONF_PASSWORD, CONF_EMAIL, "title", "_id"}
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     trackables = hass.data[DOMAIN][config_entry.entry_id][TRACKABLES]
 

homeassistant/components/tradfri/diagnostics.py

@@ -1,7 +1,7 @@
 """Diagnostics support for IKEA Tradfri."""
 from __future__ import annotations
 
-from typing import cast
+from typing import Any, cast
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
@@ -12,7 +12,7 @@ from .const import CONF_GATEWAY_ID, COORDINATOR, COORDINATOR_LIST, DOMAIN
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics the Tradfri platform."""
     entry_data = hass.data[DOMAIN][entry.entry_id]
     coordinator_data = entry_data[COORDINATOR]

homeassistant/components/zha/diagnostics.py

@@ -69,7 +69,7 @@ def shallow_asdict(obj: Any) -> dict:
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     config: dict = hass.data[DATA_ZHA].get(DATA_ZHA_CONFIG, {})
     gateway: ZHAGateway = hass.data[DATA_ZHA][DATA_ZHA_GATEWAY]
@@ -94,7 +94,7 @@ async def async_get_config_entry_diagnostics(
 
 async def async_get_device_diagnostics(
     hass: HomeAssistant, config_entry: ConfigEntry, device: dr.DeviceEntry
-) -> dict:
+) -> dict[str, Any]:
     """Return diagnostics for a device."""
     zha_device: ZHADevice = async_get_zha_device(hass, device.id)
     device_info: dict[str, Any] = zha_device.zha_device_info