Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use enums in unifi (#62029)

pull/62102/head
Robert Hillis 2021-12-16 06:53:01 -05:00 committed by GitHub
parent a3765b2977
commit a49683d09a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
homeassistant/components/unifi/sensor.py
View File

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

9
homeassistant/components/unifi/switch.py
View File

@ -15,11 +15,10 @@ from aiounifi.events import (
) )
from homeassistant.components.switch import DOMAIN, SwitchEntity from homeassistant.components.switch import DOMAIN, SwitchEntity
from homeassistant.const import ENTITY_CATEGORY_CONFIG
from homeassistant.core import callback from homeassistant.core import callback
from homeassistant.helpers.device_registry import DeviceEntryType from homeassistant.helpers.device_registry import DeviceEntryType
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity import DeviceInfo, EntityCategory
from homeassistant.helpers.entity_registry import async_entries_for_config_entry from homeassistant.helpers.entity_registry import async_entries_for_config_entry
from homeassistant.helpers.restore_state import RestoreEntity from homeassistant.helpers.restore_state import RestoreEntity
@ -185,7 +184,7 @@ class UniFiPOEClientSwitch(UniFiClient, SwitchEntity, RestoreEntity):
DOMAIN = DOMAIN DOMAIN = DOMAIN
TYPE = POE_SWITCH TYPE = POE_SWITCH
_attr_entity_category = ENTITY_CATEGORY_CONFIG _attr_entity_category = EntityCategory.CONFIG
def __init__(self, client, controller): def __init__(self, client, controller):
"""Set up POE switch.""" """Set up POE switch."""
@ -274,7 +273,7 @@ class UniFiBlockClientSwitch(UniFiClient, SwitchEntity):
DOMAIN = DOMAIN DOMAIN = DOMAIN
TYPE = BLOCK_SWITCH TYPE = BLOCK_SWITCH
_attr_entity_category = ENTITY_CATEGORY_CONFIG _attr_entity_category = EntityCategory.CONFIG
def __init__(self, client, controller): def __init__(self, client, controller):
"""Set up block switch.""" """Set up block switch."""
@ -325,7 +324,7 @@ class UniFiDPIRestrictionSwitch(UniFiBase, SwitchEntity):
DOMAIN = DOMAIN DOMAIN = DOMAIN
TYPE = DPI_SWITCH TYPE = DPI_SWITCH
_attr_entity_category = ENTITY_CATEGORY_CONFIG _attr_entity_category = EntityCategory.CONFIG
@property @property
def key(self) -> Any: def key(self) -> Any: