fix: remove orphaned "Balance" entity left over from the 0.3.0 sensor split

0.3.0 replaced the single "Balance" sensor with separate "You Owe"/"You
Are Owed" sensors, but Home Assistant doesn't automatically remove
entities an integration stops creating - anyone who upgraded was left
with a permanently-unavailable "Balance" entity in their entity registry.
async_setup_entry now removes it if present.

Also updated images/dash.png to show the current two-sensor dashboard.

Author: sriramsv

custom_components/splitwise/__init__.py

@@ -8,7 +8,11 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_PLATFORM, Platform
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers import config_entry_oauth2_flow, issue_registry as ir
+from homeassistant.helpers import (
+    config_entry_oauth2_flow,
+    entity_registry as er,
+    issue_registry as ir,
+)
 from homeassistant.helpers.typing import ConfigType
 from splitwise import Splitwise
 
@@ -42,6 +46,8 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up Splitwise from a config entry."""
+    _async_remove_stale_entities(hass, entry)
+
     implementation = (
         await config_entry_oauth2_flow.async_get_config_entry_implementation(
             hass, entry
@@ -64,6 +70,22 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     return True
 
 
+def _async_remove_stale_entities(hass: HomeAssistant, entry: ConfigEntry) -> None:
+    """Remove entities from unique_ids no longer created by this integration.
+
+    0.3.0 replaced the single "Balance" sensor (unique_id "<entry_id>_balance")
+    with separate "You Owe"/"You Are Owed" sensors. Home Assistant doesn't
+    remove entities an integration stops creating on its own, so anyone
+    upgrading would otherwise be left with a permanently-unavailable orphan.
+    """
+    ent_reg = er.async_get(hass)
+    stale_unique_id = f"{entry.entry_id}_balance"
+    if entity_id := ent_reg.async_get_entity_id(
+        Platform.SENSOR, DOMAIN, stale_unique_id
+    ):
+        ent_reg.async_remove(entity_id)
+
+
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Unload a Splitwise config entry."""
     unloaded = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)

custom_components/splitwise/manifest.json

@@ -8,5 +8,5 @@
     "iot_class": "cloud_polling",
     "issue_tracker": "https://github.com/sriramsv/custom_component_splitwise/issues",
     "requirements": ["splitwise==3.0.0"],
-    "version": "0.3.2"
+    "version": "0.3.3"
   }