From 1007283da5072b9f44fe3d6debd45af162028258 Mon Sep 17 00:00:00 2001 From: Robert Svensson Date: Sun, 23 Feb 2020 19:17:26 +0100 Subject: [PATCH] UniFi - Add unit of measurement to bandwidth sensors (#32114) --- homeassistant/components/unifi/sensor.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/unifi/sensor.py b/homeassistant/components/unifi/sensor.py index 942b0ef6779..1b6667f2e80 100644 --- a/homeassistant/components/unifi/sensor.py +++ b/homeassistant/components/unifi/sensor.py @@ -2,6 +2,7 @@ import logging from homeassistant.components.unifi.config_flow import get_controller_from_config_entry +from homeassistant.const import DATA_BYTES from homeassistant.core import callback from homeassistant.helpers.dispatcher import async_dispatcher_connect @@ -9,9 +10,6 @@ from .unifi_client import UniFiClient LOGGER = logging.getLogger(__name__) -ATTR_RECEIVING = "receiving" -ATTR_TRANSMITTING = "transmitting" - async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): """Sensor platform doesn't support configuration through configuration.yaml.""" @@ -115,6 +113,11 @@ class UniFiRxBandwidthSensor(UniFiClient): """Return a unique identifier for this bandwidth sensor.""" return f"rx-{self.client.mac}" + @property + def unit_of_measurement(self): + """Return the unit of measurement of this entity.""" + return DATA_BYTES + class UniFiTxBandwidthSensor(UniFiRxBandwidthSensor): """Transmitting bandwidth sensor."""