Add UniFi Uptime sensor

homeassistant/components/unifi/config_flow.py

@@ -16,6 +16,7 @@
 
 from .const import (
     CONF_ALLOW_BANDWIDTH_SENSORS,
+    CONF_ALLOW_UPTIME_SENSORS,
     CONF_BLOCK_CLIENT,
     CONF_CONTROLLER,
     CONF_DETECTION_TIME,
@@ -312,7 +313,11 @@ async def async_step_statistics_sensors(self, user_input=None):
                     vol.Optional(
                         CONF_ALLOW_BANDWIDTH_SENSORS,
                         default=self.controller.option_allow_bandwidth_sensors,
-                    ): bool
+                    ): bool,
+                    vol.Optional(
+                        CONF_ALLOW_UPTIME_SENSORS,
+                        default=self.controller.option_allow_uptime_sensors,
+                    ): bool,
                 }
             ),
         )

homeassistant/components/unifi/const.py

@@ -12,6 +12,7 @@
 UNIFI_WIRELESS_CLIENTS = "unifi_wireless_clients"
 
 CONF_ALLOW_BANDWIDTH_SENSORS = "allow_bandwidth_sensors"
+CONF_ALLOW_UPTIME_SENSORS = "allow_uptime_sensors"
 CONF_BLOCK_CLIENT = "block_client"
 CONF_DETECTION_TIME = "detection_time"
 CONF_IGNORE_WIRED_BUG = "ignore_wired_bug"
@@ -22,6 +23,7 @@
 CONF_SSID_FILTER = "ssid_filter"
 
 DEFAULT_ALLOW_BANDWIDTH_SENSORS = False
+DEFAULT_ALLOW_UPTIME_SENSORS = False
 DEFAULT_IGNORE_WIRED_BUG = False
 DEFAULT_POE_CLIENTS = True
 DEFAULT_TRACK_CLIENTS = True

homeassistant/components/unifi/controller.py

@@ -33,6 +33,7 @@
 
 from .const import (
     CONF_ALLOW_BANDWIDTH_SENSORS,
+    CONF_ALLOW_UPTIME_SENSORS,
     CONF_BLOCK_CLIENT,
     CONF_CONTROLLER,
     CONF_DETECTION_TIME,
@@ -45,6 +46,7 @@
     CONF_TRACK_WIRED_CLIENTS,
     CONTROLLER_ID,
     DEFAULT_ALLOW_BANDWIDTH_SENSORS,
+    DEFAULT_ALLOW_UPTIME_SENSORS,
     DEFAULT_DETECTION_TIME,
     DEFAULT_IGNORE_WIRED_BUG,
     DEFAULT_POE_CLIENTS,
@@ -184,6 +186,13 @@ def option_allow_bandwidth_sensors(self):
             CONF_ALLOW_BANDWIDTH_SENSORS, DEFAULT_ALLOW_BANDWIDTH_SENSORS
         )
 
+    @property
+    def option_allow_uptime_sensors(self):
+        """Config entry option to allow uptime sensors."""
+        return self.config_entry.options.get(
+            CONF_ALLOW_UPTIME_SENSORS, DEFAULT_ALLOW_UPTIME_SENSORS
+        )
+
     @callback
     def async_unifi_signalling_callback(self, signal, data):
         """Handle messages back from UniFi library."""

homeassistant/components/unifi/sensor.py

@@ -1,10 +1,11 @@
 """Support for bandwidth sensors with UniFi clients."""
 import logging
 
-from homeassistant.components.sensor import DOMAIN
+from homeassistant.components.sensor import DEVICE_CLASS_TIMESTAMP, DOMAIN
 from homeassistant.const import DATA_MEGABYTES
 from homeassistant.core import callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
+import homeassistant.util.dt as dt_util
 
 from .const import DOMAIN as UNIFI_DOMAIN
 from .unifi_client import UniFiClient
@@ -13,6 +14,7 @@
 
 RX_SENSOR = "rx"
 TX_SENSOR = "tx"
+UPTIME_SENSOR = "uptime"
 
 
 async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
@@ -22,15 +24,22 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
 async def async_setup_entry(hass, config_entry, async_add_entities):
     """Set up sensors for UniFi integration."""
     controller = hass.data[UNIFI_DOMAIN][config_entry.entry_id]
-    controller.entities[DOMAIN] = {RX_SENSOR: set(), TX_SENSOR: set()}
+    controller.entities[DOMAIN] = {
+        RX_SENSOR: set(),
+        TX_SENSOR: set(),
+        UPTIME_SENSOR: set(),
+    }
 
     @callback
     def items_added(
         clients: set = controller.api.clients, devices: set = controller.api.devices
     ) -> None:
         """Update the values of the controller."""
         if controller.option_allow_bandwidth_sensors:
-            add_entities(controller, async_add_entities, clients)
+            add_bandwith_entities(controller, async_add_entities, clients)
+
+        if controller.option_allow_uptime_sensors:
+            add_uptime_entities(controller, async_add_entities, clients)
 
     for signal in (controller.signal_update, controller.signal_options_update):
         controller.listeners.append(async_dispatcher_connect(hass, signal, items_added))
@@ -39,7 +48,7 @@ def items_added(
 
 
 @callback
-def add_entities(controller, async_add_entities, clients):
+def add_bandwith_entities(controller, async_add_entities, clients):
     """Add new sensor entities from the controller."""
     sensors = []
 
@@ -55,6 +64,22 @@ def add_entities(controller, async_add_entities, clients):
         async_add_entities(sensors)
 
 
+@callback
+def add_uptime_entities(controller, async_add_entities, clients):
+    """Add new sensor entities from the controller."""
+    sensors = []
+
+    for mac in clients:
+        if mac in controller.entities[DOMAIN][UniFiUpTimeSensor.TYPE]:
+            continue
+
+        client = controller.api.clients[mac]
+        sensors.append(UniFiUpTimeSensor(client, controller))
+
+    if sensors:
+        async_add_entities(sensors)
+
+
 class UniFiBandwidthSensor(UniFiClient):
     """UniFi bandwidth sensor base class."""
 
@@ -100,3 +125,30 @@ def state(self) -> int:
         if self._is_wired:
             return self.client.wired_tx_bytes / 1000000
         return self.client.tx_bytes / 1000000
+
+
+class UniFiUpTimeSensor(UniFiClient):
+    """UniFi uptime sensor."""
+
+    DOMAIN = DOMAIN
+    TYPE = UPTIME_SENSOR
+
+    @property
+    def device_class(self) -> str:
+        """Return device class."""
+        return DEVICE_CLASS_TIMESTAMP
+
+    @property
+    def name(self) -> str:
+        """Return the name of the client."""
+        return f"{super().name} {self.TYPE.capitalize()}"
+
+    @property
+    def state(self) -> int:
+        """Return the uptime of the client."""
+        return dt_util.utc_from_timestamp(float(self.client.uptime)).isoformat()
+
+    async def options_updated(self) -> None:
+        """Config entry options are updated, remove entity if option is disabled."""
+        if not self.controller.option_allow_uptime_sensors:
+            await self.remove_item({self.client.mac})

homeassistant/components/unifi/strings.json

@@ -54,7 +54,8 @@
       },
       "statistics_sensors": {
         "data": {
-          "allow_bandwidth_sensors": "Bandwidth usage sensors for network clients"
+          "allow_bandwidth_sensors": "Bandwidth usage sensors for network clients",
+          "allow_uptime_sensors": "Uptime sensors for network clients"
         },
         "description": "Configure statistics sensors",
         "title": "UniFi options 3/3"

homeassistant/components/unifi/translations/nl.json

@@ -57,7 +57,8 @@
             },
             "statistics_sensors": {
                 "data": {
-                    "allow_bandwidth_sensors": "Maak bandbreedtegebruiksensoren voor netwerkclients"
+                    "allow_bandwidth_sensors": "Maak bandbreedtegebruiksensoren voor netwerkclients",
+                    "allow_uptime_sensors": "Maak uptime-sensoren voor netwerkclients"
                 },
                 "title": "UniFi-opties 3/3"
             }

tests/components/unifi/test_config_flow.py

@@ -4,6 +4,7 @@
 from homeassistant import data_entry_flow
 from homeassistant.components.unifi.const import (
     CONF_ALLOW_BANDWIDTH_SENSORS,
+    CONF_ALLOW_UPTIME_SENSORS,
     CONF_BLOCK_CLIENT,
     CONF_CONTROLLER,
     CONF_DETECTION_TIME,
@@ -341,7 +342,11 @@ async def test_advanced_option_flow(hass):
     assert result["step_id"] == "statistics_sensors"
 
     result = await hass.config_entries.options.async_configure(
-        result["flow_id"], user_input={CONF_ALLOW_BANDWIDTH_SENSORS: True}
+        result["flow_id"],
+        user_input={
+            CONF_ALLOW_BANDWIDTH_SENSORS: True,
+            CONF_ALLOW_UPTIME_SENSORS: True,
+        },
     )
 
     assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
@@ -355,6 +360,7 @@ async def test_advanced_option_flow(hass):
         CONF_POE_CLIENTS: False,
         CONF_BLOCK_CLIENT: [CLIENTS[0]["mac"]],
         CONF_ALLOW_BANDWIDTH_SENSORS: True,
+        CONF_ALLOW_UPTIME_SENSORS: True,
     }
 
 

tests/components/unifi/test_controller.py

@@ -13,6 +13,7 @@
     CONF_CONTROLLER,
     CONF_SITE_ID,
     DEFAULT_ALLOW_BANDWIDTH_SENSORS,
+    DEFAULT_ALLOW_UPTIME_SENSORS,
     DEFAULT_DETECTION_TIME,
     DEFAULT_TRACK_CLIENTS,
     DEFAULT_TRACK_DEVICES,
@@ -49,6 +50,7 @@
     "sw_port": 1,
     "wired-rx_bytes": 1234000000,
     "wired-tx_bytes": 5678000000,
+    "uptime": 1562600160,
 }
 
 CONTROLLER_DATA = {
@@ -175,6 +177,7 @@ async def test_controller_setup(hass):
     assert controller.site_role == SITES[controller.site_name]["role"]
 
     assert controller.option_allow_bandwidth_sensors == DEFAULT_ALLOW_BANDWIDTH_SENSORS
+    assert controller.option_allow_uptime_sensors == DEFAULT_ALLOW_UPTIME_SENSORS
     assert isinstance(controller.option_block_clients, list)
     assert controller.option_track_clients == DEFAULT_TRACK_CLIENTS
     assert controller.option_track_devices == DEFAULT_TRACK_DEVICES