ensure price state exists before doing calcs

rsnodgrass · Jan 19, 2022 · 8393639 · 8393639
1 parent a9e9792
commit 8393639
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/helium/sensor.py b/custom_components/helium/sensor.py
@@ -214,8 +214,10 @@ async def async_update(self):
             self._attrs[attr] = json[attr]
 
         # update USD price attribute for the current HNT value based on current Oracle HNT/USD price
-        self._attrs[CURRENCY_USD] = round(self._price_sensor.state * self._state, 2)
-
+        if self._state and self._price_sensor.state:
+            self._attrs[CURRENCY_USD] = round(self._price_sensor.state * self._state, 2)
+        else
+            self._attrs.pop(CURRENCY_USD)
 
     @property
     def extra_state_attributes(self):