Fixes update sensor after startup, fixes #174

tomaae · Nov 14, 2024 · ed432b9 · ed432b9
1 parent 025045a
commit ed432b9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/custom_components/truenas/coordinator.py b/custom_components/truenas/coordinator.py
@@ -292,14 +292,20 @@ def get_updatecheck(self) -> None:
                 {
                     "name": "update_version",
                     "source": "version",
-                    "default": self.ds["system_info"]["version"],
+                    "default": "unknown",
                 },
             ],
         )
 
         if not self.api.connected():
             return
 
+        if (
+            self.ds["system_info"]["update_version"] == "unknown"
+            and self.ds["system_info"]["version"]
+        ):
+            self.ds["system_info"]["update_version"] = self.ds["system_info"]["version"]
+
         self.ds["system_info"]["update_available"] = (
             self.ds["system_info"]["update_status"] == "AVAILABLE"
         )