Adjust diagnostics return types (#85525)

* Adjust diagnostics return types

* Replace dict with Mapping
This commit is contained in:
epenet 2023-01-13 09:17:54 +01:00 committed by GitHub
parent 9927bb3330
commit 1deb4c68f3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 8 deletions

View file

@ -1,7 +1,7 @@
"""The Diagnostics integration.""" """The Diagnostics integration."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Callable, Coroutine from collections.abc import Callable, Coroutine, Mapping
from dataclasses import dataclass, field from dataclasses import dataclass, field
from http import HTTPStatus from http import HTTPStatus
import json import json
@ -38,10 +38,11 @@ class DiagnosticsPlatformData:
"""Diagnostic platform data.""" """Diagnostic platform data."""
config_entry_diagnostics: Callable[ config_entry_diagnostics: Callable[
[HomeAssistant, ConfigEntry], Coroutine[Any, Any, Any] [HomeAssistant, ConfigEntry], Coroutine[Any, Any, Mapping[str, Any]]
] | None ] | None
device_diagnostics: Callable[ device_diagnostics: Callable[
[HomeAssistant, ConfigEntry, DeviceEntry], Coroutine[Any, Any, Any] [HomeAssistant, ConfigEntry, DeviceEntry],
Coroutine[Any, Any, Mapping[str, Any]],
] | None ] | None
@ -72,12 +73,12 @@ class DiagnosticsProtocol(Protocol):
async def async_get_config_entry_diagnostics( async def async_get_config_entry_diagnostics(
self, hass: HomeAssistant, config_entry: ConfigEntry self, hass: HomeAssistant, config_entry: ConfigEntry
) -> Any: ) -> Mapping[str, Any]:
"""Return diagnostics for a config entry.""" """Return diagnostics for a config entry."""
async def async_get_device_diagnostics( async def async_get_device_diagnostics(
self, hass: HomeAssistant, config_entry: ConfigEntry, device: DeviceEntry self, hass: HomeAssistant, config_entry: ConfigEntry, device: DeviceEntry
) -> Any: ) -> Mapping[str, Any]:
"""Return diagnostics for a device.""" """Return diagnostics for a device."""
@ -148,7 +149,7 @@ def handle_get(
async def _async_get_json_file_response( async def _async_get_json_file_response(
hass: HomeAssistant, hass: HomeAssistant,
data: Any, data: Mapping[str, Any],
filename: str, filename: str,
domain: str, domain: str,
d_id: str, d_id: str,

View file

@ -375,7 +375,7 @@ _FUNCTION_MATCH: dict[str, list[TypeHintMatch]] = {
0: "HomeAssistant", 0: "HomeAssistant",
1: "ConfigEntry", 1: "ConfigEntry",
}, },
return_type=_Special.UNDEFINED, return_type="Mapping[str, Any]",
), ),
TypeHintMatch( TypeHintMatch(
function_name="async_get_device_diagnostics", function_name="async_get_device_diagnostics",
@ -384,7 +384,7 @@ _FUNCTION_MATCH: dict[str, list[TypeHintMatch]] = {
1: "ConfigEntry", 1: "ConfigEntry",
2: "DeviceEntry", 2: "DeviceEntry",
}, },
return_type=_Special.UNDEFINED, return_type="Mapping[str, Any]",
), ),
], ],
"notify": [ "notify": [