Make Store a generic class (#74617)

2022-07-09 22:32:57 +02:00
parent d37ad20894
commit 16900dcef1
27 changed files with 106 additions and 97 deletions
--- a/homeassistant/helpers/device_registry.py
+++ b/homeassistant/helpers/device_registry.py
@@ -164,7 +164,7 @@ def _async_get_device_id_from_index(
    return None


-class DeviceRegistryStore(storage.Store):
+class DeviceRegistryStore(storage.Store[dict[str, list[dict[str, Any]]]]):
    """Store entity registry data."""

    async def _async_migrate_func(
@@ -569,7 +569,6 @@ class DeviceRegistry:
        deleted_devices = OrderedDict()

        if data is not None:
-            data = cast("dict[str, Any]", data)
            for device in data["devices"]:
                devices[device["id"]] = DeviceEntry(
                    area_id=device["area_id"],