Replace local Barrier CC constants with library enums (#53109)

This commit is contained in:
kpine 2021-07-16 23:54:11 -07:00 committed by GitHub
parent e6e1118dd4
commit 24cc5c8a0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 22 deletions

View File

@ -5,6 +5,7 @@ import logging
from typing import Any from typing import Any
from zwave_js_server.client import Client as ZwaveClient from zwave_js_server.client import Client as ZwaveClient
from zwave_js_server.const import BarrierState
from zwave_js_server.model.value import Value as ZwaveValue from zwave_js_server.model.value import Value as ZwaveValue
from homeassistant.components.cover import ( from homeassistant.components.cover import (
@ -29,15 +30,6 @@ from .entity import ZWaveBaseEntity
LOGGER = logging.getLogger(__name__) LOGGER = logging.getLogger(__name__)
BARRIER_TARGET_CLOSE = 0
BARRIER_TARGET_OPEN = 255
BARRIER_STATE_CLOSED = 0
BARRIER_STATE_CLOSING = 252
BARRIER_STATE_STOPPED = 253
BARRIER_STATE_OPENING = 254
BARRIER_STATE_OPEN = 255
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
@ -172,14 +164,14 @@ class ZwaveMotorizedBarrier(ZWaveBaseEntity, CoverEntity):
"""Return if the cover is opening or not.""" """Return if the cover is opening or not."""
if self.info.primary_value.value is None: if self.info.primary_value.value is None:
return None return None
return bool(self.info.primary_value.value == BARRIER_STATE_OPENING) return bool(self.info.primary_value.value == BarrierState.OPENING)
@property @property
def is_closing(self) -> bool | None: def is_closing(self) -> bool | None:
"""Return if the cover is closing or not.""" """Return if the cover is closing or not."""
if self.info.primary_value.value is None: if self.info.primary_value.value is None:
return None return None
return bool(self.info.primary_value.value == BARRIER_STATE_CLOSING) return bool(self.info.primary_value.value == BarrierState.CLOSING)
@property @property
def is_closed(self) -> bool | None: def is_closed(self) -> bool | None:
@ -190,15 +182,15 @@ class ZwaveMotorizedBarrier(ZWaveBaseEntity, CoverEntity):
# issuing an open cover command. Return None in this case which # issuing an open cover command. Return None in this case which
# produces an unknown state and allows it to be resolved with an open # produces an unknown state and allows it to be resolved with an open
# command. # command.
if self.info.primary_value.value == BARRIER_STATE_STOPPED: if self.info.primary_value.value == BarrierState.STOPPED:
return None return None
return bool(self.info.primary_value.value == BARRIER_STATE_CLOSED) return bool(self.info.primary_value.value == BarrierState.CLOSED)
async def async_open_cover(self, **kwargs: Any) -> None: async def async_open_cover(self, **kwargs: Any) -> None:
"""Open the garage door.""" """Open the garage door."""
await self.info.node.async_set_value(self._target_state, BARRIER_TARGET_OPEN) await self.info.node.async_set_value(self._target_state, BarrierState.OPEN)
async def async_close_cover(self, **kwargs: Any) -> None: async def async_close_cover(self, **kwargs: Any) -> None:
"""Close the garage door.""" """Close the garage door."""
await self.info.node.async_set_value(self._target_state, BARRIER_TARGET_CLOSE) await self.info.node.async_set_value(self._target_state, BarrierState.CLOSED)

View File

@ -5,6 +5,7 @@ import logging
from typing import Any from typing import Any
from zwave_js_server.client import Client as ZwaveClient from zwave_js_server.client import Client as ZwaveClient
from zwave_js_server.const import BarrierEventSignalingSubsystemState
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN, SwitchEntity from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN, SwitchEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
@ -19,10 +20,6 @@ from .entity import ZWaveBaseEntity
LOGGER = logging.getLogger(__name__) LOGGER = logging.getLogger(__name__)
BARRIER_EVENT_SIGNALING_OFF = 0
BARRIER_EVENT_SIGNALING_ON = 255
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: ConfigEntry,
@ -108,7 +105,7 @@ class ZWaveBarrierEventSignalingSwitch(ZWaveBaseEntity, SwitchEntity):
async def async_turn_on(self, **kwargs: Any) -> None: async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn the switch on.""" """Turn the switch on."""
await self.info.node.async_set_value( await self.info.node.async_set_value(
self.info.primary_value, BARRIER_EVENT_SIGNALING_ON self.info.primary_value, BarrierEventSignalingSubsystemState.ON
) )
# this value is not refreshed, so assume success # this value is not refreshed, so assume success
self._state = True self._state = True
@ -117,7 +114,7 @@ class ZWaveBarrierEventSignalingSwitch(ZWaveBaseEntity, SwitchEntity):
async def async_turn_off(self, **kwargs: Any) -> None: async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn the switch off.""" """Turn the switch off."""
await self.info.node.async_set_value( await self.info.node.async_set_value(
self.info.primary_value, BARRIER_EVENT_SIGNALING_OFF self.info.primary_value, BarrierEventSignalingSubsystemState.OFF
) )
# this value is not refreshed, so assume success # this value is not refreshed, so assume success
self._state = False self._state = False
@ -127,4 +124,6 @@ class ZWaveBarrierEventSignalingSwitch(ZWaveBaseEntity, SwitchEntity):
def _update_state(self) -> None: def _update_state(self) -> None:
self._state = None self._state = None
if self.info.primary_value.value is not None: if self.info.primary_value.value is not None:
self._state = self.info.primary_value.value == BARRIER_EVENT_SIGNALING_ON self._state = (
self.info.primary_value.value == BarrierEventSignalingSubsystemState.ON
)