Allow storing device options in DeviceEntry

home-assistant · Nov 14, 2021 · 2b07378 · 2b07378
1 parent c323815
commit 2b07378
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/homeassistant/helpers/device_registry.py b/homeassistant/helpers/device_registry.py
@@ -75,6 +75,7 @@ class DeviceEntry:
     model: str | None = attr.ib(default=None)
     name_by_user: str | None = attr.ib(default=None)
     name: str | None = attr.ib(default=None)
+    options: dict[str, Any] = attr.ib(factory=dict)
     suggested_area: str | None = attr.ib(default=None)
     sw_version: str | None = attr.ib(default=None)
     via_device_id: str | None = attr.ib(default=None)
@@ -257,6 +258,7 @@ def async_get_or_create(
         manufacturer: str | None | UndefinedType = UNDEFINED,
         model: str | None | UndefinedType = UNDEFINED,
         name: str | None | UndefinedType = UNDEFINED,
+        options: dict[str, Any] | UndefinedType = UNDEFINED,
         suggested_area: str | None | UndefinedType = UNDEFINED,
         sw_version: str | None | UndefinedType = UNDEFINED,
         via_device: tuple[str, str] | None = None,
@@ -312,6 +314,7 @@ def async_get_or_create(
             merge_identifiers=identifiers or UNDEFINED,
             model=model,
             name=name,
+            options=options,
             suggested_area=suggested_area,
             sw_version=sw_version,
             via_device_id=via_device_id,
@@ -336,6 +339,7 @@ def async_update_device(
         name_by_user: str | None | UndefinedType = UNDEFINED,
         name: str | None | UndefinedType = UNDEFINED,
         new_identifiers: set[tuple[str, str]] | UndefinedType = UNDEFINED,
+        options: dict[str, Any] | UndefinedType = UNDEFINED,
         remove_config_entry_id: str | UndefinedType = UNDEFINED,
         suggested_area: str | None | UndefinedType = UNDEFINED,
         sw_version: str | None | UndefinedType = UNDEFINED,
@@ -353,6 +357,7 @@ def async_update_device(
             name_by_user=name_by_user,
             name=name,
             new_identifiers=new_identifiers,
+            options=options,
             remove_config_entry_id=remove_config_entry_id,
             suggested_area=suggested_area,
             sw_version=sw_version,
@@ -376,6 +381,7 @@ def _async_update_device(
         name_by_user: str | None | UndefinedType = UNDEFINED,
         name: str | None | UndefinedType = UNDEFINED,
         new_identifiers: set[tuple[str, str]] | UndefinedType = UNDEFINED,
+        options: dict[str, Any] | UndefinedType = UNDEFINED,
         remove_config_entry_id: str | UndefinedType = UNDEFINED,
         suggested_area: str | None | UndefinedType = UNDEFINED,
         sw_version: str | None | UndefinedType = UNDEFINED,
@@ -436,6 +442,7 @@ def _async_update_device(
             ("manufacturer", manufacturer),
             ("model", model),
             ("name", name),
+            ("options", options),
             ("suggested_area", suggested_area),
             ("sw_version", sw_version),
             ("via_device_id", via_device_id),