jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-12 15:57:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use new SwitchDeviceClass enum in home-plus-control (#61760)

Browse Source 
Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-12-14 09:56:52 +01:00 committed by GitHub
parent de45144750
commit 8fc69b7242
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/home_plus_control/switch.py
View File

@ -1,11 +1,7 @@
"""Legrand Home+ Control Switch Entity Module that uses the HomeAssistant DataUpdateCoordinator.""" """Legrand Home+ Control Switch Entity Module that uses the HomeAssistant DataUpdateCoordinator."""
from functools import partial from functools import partial
from homeassistant.components.switch import ( from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
DEVICE_CLASS_OUTLET,
DEVICE_CLASS_SWITCH,
SwitchEntity,
)
from homeassistant.core import callback from homeassistant.core import callback
from homeassistant.helpers import dispatcher from homeassistant.helpers import dispatcher
from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity import DeviceInfo
@ -97,8 +93,8 @@ class HomeControlSwitchEntity(CoordinatorEntity, SwitchEntity):
def device_class(self): def device_class(self):
"""Return the class of this device, from component DEVICE_CLASSES.""" """Return the class of this device, from component DEVICE_CLASSES."""
if self.module.device == "plug": if self.module.device == "plug":
return DEVICE_CLASS_OUTLET return SwitchDeviceClass.OUTLET
return DEVICE_CLASS_SWITCH return SwitchDeviceClass.SWITCH
@property @property
def available(self) -> bool: def available(self) -> bool: