Use new enums in mqtt (#61936)

* Use new enums in mqtt

* Fix typo
This commit is contained in:
epenet 2021-12-16 15:09:31 +01:00 committed by GitHub
parent ff654a9753
commit 079b7f217f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 9 deletions

View file

@ -13,6 +13,7 @@ from homeassistant.components.sensor import (
DEVICE_CLASSES_SCHEMA,
ENTITY_ID_FORMAT,
STATE_CLASSES_SCHEMA,
SensorDeviceClass,
SensorEntity,
)
from homeassistant.const import (
@ -21,8 +22,6 @@ from homeassistant.const import (
CONF_NAME,
CONF_UNIT_OF_MEASUREMENT,
CONF_VALUE_TEMPLATE,
DEVICE_CLASS_DATE,
DEVICE_CLASS_TIMESTAMP,
)
from homeassistant.core import HomeAssistant, callback
import homeassistant.helpers.config_validation as cv
@ -200,14 +199,14 @@ class MqttSensor(MqttEntity, SensorEntity):
)
if payload is not None and self.device_class in (
DEVICE_CLASS_DATE,
DEVICE_CLASS_TIMESTAMP,
SensorDeviceClass.DATE,
SensorDeviceClass.TIMESTAMP,
):
if (payload := dt_util.parse_datetime(payload)) is None:
_LOGGER.warning(
"Invalid state message '%s' from '%s'", msg.payload, msg.topic
)
elif self.device_class == DEVICE_CLASS_DATE:
elif self.device_class == SensorDeviceClass.DATE:
payload = payload.date()
self._state = payload