Fixed crash when pool goes offline, fixes #152

tomaae · Apr 25, 2024 · 2c5ed51 · 2c5ed51
1 parent a59ea1e
commit 2c5ed51
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/custom_components/truenas/coordinator.py b/custom_components/truenas/coordinator.py
@@ -782,13 +782,16 @@ def get_pool(self) -> None:
 
                 self.ds["pool"][uid].pop("root_dataset")
 
-            self.ds["pool"][uid]["usage"] = round(
-                (
-                    self.ds["pool"][uid]["available_gib"]
-                    / self.ds["pool"][uid]["total_gib"]
+            if self.ds["pool"][uid]["total_gib"] > 0:
+                self.ds["pool"][uid]["usage"] = round(
+                    (
+                        self.ds["pool"][uid]["available_gib"]
+                        / self.ds["pool"][uid]["total_gib"]
+                    )
+                    * 100
                 )
-                * 100
-            )
+            else:
+                self.ds["pool"][uid]["usage"] = 0
 
     # ---------------------------
     #   get_dataset