Use new SwitchDeviceClass in aten_pe (#61281)
Co-authored-by: epenet <epenet@users.noreply.github.com>pull/61231/head
parent
fd7328ce23
commit
c893ad80e4
|
@ -6,8 +6,8 @@ from atenpdu import AtenPE, AtenPEError
|
||||||
import voluptuous as vol
|
import voluptuous as vol
|
||||||
|
|
||||||
from homeassistant.components.switch import (
|
from homeassistant.components.switch import (
|
||||||
DEVICE_CLASS_OUTLET,
|
|
||||||
PLATFORM_SCHEMA,
|
PLATFORM_SCHEMA,
|
||||||
|
SwitchDeviceClass,
|
||||||
SwitchEntity,
|
SwitchEntity,
|
||||||
)
|
)
|
||||||
from homeassistant.const import CONF_HOST, CONF_PORT, CONF_USERNAME
|
from homeassistant.const import CONF_HOST, CONF_PORT, CONF_USERNAME
|
||||||
|
@ -67,7 +67,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
|
||||||
class AtenSwitch(SwitchEntity):
|
class AtenSwitch(SwitchEntity):
|
||||||
"""Represents an ATEN PE switch."""
|
"""Represents an ATEN PE switch."""
|
||||||
|
|
||||||
_attr_device_class = DEVICE_CLASS_OUTLET
|
_attr_device_class = SwitchDeviceClass.OUTLET
|
||||||
|
|
||||||
def __init__(self, device, mac, outlet, name):
|
def __init__(self, device, mac, outlet, name):
|
||||||
"""Initialize an ATEN PE switch."""
|
"""Initialize an ATEN PE switch."""
|
||||||
|
|
Loading…
Reference in New Issue