evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add entity category to UniFi sensors and switches (#58484)

Browse source
This commit is contained in:
Robert Svensson 2021-10-26 20:23:20 +02:00 • committed by GitHub
parent dd1154ad08
commit 47f6313e5b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 33 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
homeassistant/components/unifi/sensor.py
View file

@ -7,7 +7,7 @@ Support for uptime sensors of network clients.
from datetime import datetime, timedelta
from homeassistant.components.sensor import DEVICE_CLASS_TIMESTAMP, DOMAIN, SensorEntity
from homeassistant.const import DATA_MEGABYTES
from homeassistant.const import DATA_MEGABYTES, ENTITY_CATEGORY_DIAGNOSTIC
from homeassistant.core import callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
import homeassistant.util.dt as dt_util
@ -86,6 +86,7 @@ class UniFiBandwidthSensor(UniFiClient, SensorEntity):
DOMAIN = DOMAIN
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
_attr_native_unit_of_measurement = DATA_MEGABYTES
@property
@ -132,6 +133,7 @@ class UniFiUpTimeSensor(UniFiClient, SensorEntity):
TYPE = UPTIME_SENSOR
_attr_device_class = DEVICE_CLASS_TIMESTAMP
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
def __init__(self, client, controller):
"""Set up tracked client."""