Add icons

cyr-ius · May 8, 2024 · f6bc05a · f6bc05a
1 parent 453ed4c
commit f6bc05a
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/custom_components/livebox/binary_sensor.py b/custom_components/livebox/binary_sensor.py
@@ -17,7 +17,7 @@
 from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from .const import DOMAIN, MISSED_ICON
+from .const import DDNS_ICON, DOMAIN, MISSED_ICON, RA_ICON
 from .coordinator import LiveboxDataUpdateCoordinator
 from .entity import LiveboxEntity
 
@@ -69,7 +69,9 @@ class LiveboxBinarySensorEntityDescription(BinarySensorEntityDescription):
     LiveboxBinarySensorEntityDescription(
         key="remote_access",
         name="Remote Access",
+        icon=RA_ICON,
         value_fn=lambda x: x.get("remote_access"),
+        attrs={},
         translation_key="remote_access",
     ),
 )
@@ -90,6 +92,7 @@ async def async_setup_entry(
         description = LiveboxBinarySensorEntityDescription(
             key=f"ddns_{idx}",
             index=idx,
+            icon=DDNS_ICON,
             device_class=BinarySensorDeviceClass.PROBLEM,
             name=f"Dynamic DNS ({item.get('service')})",
             value_fn=lambda x, y: x["ddns"][y].get("status", "").lower() != "updated",

diff --git a/custom_components/livebox/const.py b/custom_components/livebox/const.py
@@ -26,3 +26,5 @@
 RING_ICON = "mdi:phone-classic"
 GUESTWIFI_ICON = "mdi:wifi-lock-open"
 DEVICE_WANACCESS_ICON = "mdi:wan"
+RA_ICON = "mdi:remote-desktop"
+DDNS_ICON = "mdi:dns"
diff --git a/custom_components/livebox/sensor.py b/custom_components/livebox/sensor.py
@@ -162,6 +162,7 @@ class LiveboxSensorEntityDescription(SensorEntityDescription):
     LiveboxSensorEntityDescription(
         key="fiber_tx",
         name="Fiber Tx",
+        icon=UPLOAD_ICON,
         value_fn=lambda x: round(
             x.get("fiber_stats", {}).get("TxBytes", 0) / 1048576, 2
         ),
@@ -173,6 +174,7 @@ class LiveboxSensorEntityDescription(SensorEntityDescription):
     LiveboxSensorEntityDescription(
         key="fiber_rx",
         name="Fiber Rx",
+        icon=DOWNLOAD_ICON,
         value_fn=lambda x: round(
             x.get("fiber_stats", {}).get("RxBytes", 0) / 1048576, 2
         ),