Fix Habitica gems/hourglass sensors (home-assistant#135323)

tdelaux · Jan 10, 2025 · bf747bb · bf747bb
1 parent 560d15e
commit bf747bb
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/homeassistant/components/habitica/sensor.py b/homeassistant/components/habitica/sensor.py
@@ -135,14 +135,14 @@ class HabiticaSensorEntity(StrEnum):
     HabiticaSensorEntityDescription(
         key=HabiticaSensorEntity.GEMS,
         translation_key=HabiticaSensorEntity.GEMS,
-        value_fn=lambda user, _: round(user.balance * 4) if user.balance else None,
+        value_fn=lambda user, _: None if (b := user.balance) is None else round(b * 4),
         suggested_display_precision=0,
         entity_picture="shop_gem.png",
     ),
     HabiticaSensorEntityDescription(
         key=HabiticaSensorEntity.TRINKETS,
         translation_key=HabiticaSensorEntity.TRINKETS,
-        value_fn=lambda user, _: user.purchased.plan.consecutive.trinkets or 0,
+        value_fn=lambda user, _: user.purchased.plan.consecutive.trinkets,
         suggested_display_precision=0,
         native_unit_of_measurement="⧖",
         entity_picture="notif_subscriber_reward.png",

diff --git a/tests/components/habitica/fixtures/user.json b/tests/components/habitica/fixtures/user.json
@@ -114,6 +114,13 @@
         }
       }
     },
-    "balance": 10
+    "balance": 10,
+    "purchased": {
+      "plan": {
+        "consecutive": {
+          "trinkets": 0
+        }
+      }
+    }
   }
 }