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

Use enums in velux (#61991)

Browse Source
This commit is contained in:
Robert Hillis 2021-12-16 03:11:22 -05:00 committed by GitHub
parent 173582d4fc
commit 1568ee67c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
homeassistant/components/velux/cover.py
View File

@ -5,12 +5,6 @@ from pyvlx.opening_device import Awning, Blind, GarageDoor, Gate, RollerShutter,
from homeassistant.components.cover import ( from homeassistant.components.cover import (
ATTR_POSITION, ATTR_POSITION,
ATTR_TILT_POSITION, ATTR_TILT_POSITION,
DEVICE_CLASS_AWNING,
DEVICE_CLASS_BLIND,
DEVICE_CLASS_GARAGE,
DEVICE_CLASS_GATE,
DEVICE_CLASS_SHUTTER,
DEVICE_CLASS_WINDOW,
SUPPORT_CLOSE, SUPPORT_CLOSE,
SUPPORT_CLOSE_TILT, SUPPORT_CLOSE_TILT,
SUPPORT_OPEN, SUPPORT_OPEN,
@ -19,6 +13,7 @@ from homeassistant.components.cover import (
SUPPORT_SET_TILT_POSITION, SUPPORT_SET_TILT_POSITION,
SUPPORT_STOP, SUPPORT_STOP,
SUPPORT_STOP_TILT, SUPPORT_STOP_TILT,
CoverDeviceClass,
CoverEntity, CoverEntity,
) )
@ -68,18 +63,18 @@ class VeluxCover(VeluxEntity, CoverEntity):
def device_class(self): def device_class(self):
"""Define this cover as either awning, blind, garage, gate, shutter or window.""" """Define this cover as either awning, blind, garage, gate, shutter or window."""
if isinstance(self.node, Awning): if isinstance(self.node, Awning):
return DEVICE_CLASS_AWNING return CoverDeviceClass.AWNING
if isinstance(self.node, Blind): if isinstance(self.node, Blind):
return DEVICE_CLASS_BLIND return CoverDeviceClass.BLIND
if isinstance(self.node, GarageDoor): if isinstance(self.node, GarageDoor):
return DEVICE_CLASS_GARAGE return CoverDeviceClass.GARAGE
if isinstance(self.node, Gate): if isinstance(self.node, Gate):
return DEVICE_CLASS_GATE return CoverDeviceClass.GATE
if isinstance(self.node, RollerShutter): if isinstance(self.node, RollerShutter):
return DEVICE_CLASS_SHUTTER return CoverDeviceClass.SHUTTER
if isinstance(self.node, Window): if isinstance(self.node, Window):
return DEVICE_CLASS_WINDOW return CoverDeviceClass.WINDOW
return DEVICE_CLASS_WINDOW return CoverDeviceClass.WINDOW
@property @property
def is_closed(self): def is_closed(self):