Fix device info

ElD4n1 · Dec 4, 2024 · 5ccc519 · 5ccc519
1 parent 0287241
commit 5ccc519
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/custom_components/bmstools/device.py b/custom_components/bmstools/device.py
@@ -46,11 +46,17 @@ def unique_id(self) -> str | None:
     @property
     def device_info(self) -> DeviceInfo:
         """Return device specific attributes."""
-        return {
+        device_info = {
             ATTR_IDENTIFIERS: {(DOMAIN, self._config_entry_data[ATTR_SERIAL_NUMBER])},
             ATTR_MANUFACTURER: MANUFACTURER,
             ATTR_MODEL: self._config_entry_data[ATTR_MODEL],
             ATTR_SW_VERSION: self._config_entry_data[ATTR_SW_VERSION],
             ATTR_HW_VERSION: self._config_entry_data[ATTR_HW_VERSION],
-            ATTR_DEFAULT_NAME: DEFAULT_DEVICE_NAME,
+            ATTR_MODEL_ID: self._config_entry_data[ATTR_MODEL],
+            ATTR_NAME: f"{MANUFACTURER} {DEFAULT_DEVICE_NAME} {self._config_entry_data[ATTR_MODEL]}",
         }
+
+        if ATTR_SERIAL_NUMBER in self._config_entry_data:
+            device_info.ATTR_SERIAL_NUMBER = self._config_entry_data[ATTR_SERIAL_NUMBER]
+
+        return device_info
diff --git a/custom_components/bmstools/manifest.json b/custom_components/bmstools/manifest.json
@@ -6,7 +6,7 @@
   "documentation": "https://github.com/ElD4n1/home-assistant-bms-tools-integration",
   "issue_tracker": "https://github.com/ElD4n1/home-assistant-bms-tools-integration/issues",
   "requirements": [
-    "bmstools==1.2.0rc1"
+    "bmstools==1.2.1-beta.1"
   ],
   "ssdp": [],
   "zeroconf": [],