Use new enums in mobile_app (#61929)

This commit is contained in:
epenet 2021-12-16 15:13:13 +01:00 committed by GitHub
parent 5227019d3e
commit 029af94d28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,13 +1,11 @@
"""Sensor platform for mobile_app.""" """Sensor platform for mobile_app."""
from __future__ import annotations from __future__ import annotations
from homeassistant.components.sensor import SensorEntity from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import ( from homeassistant.const import (
CONF_NAME, CONF_NAME,
CONF_UNIQUE_ID, CONF_UNIQUE_ID,
CONF_WEBHOOK_ID, CONF_WEBHOOK_ID,
DEVICE_CLASS_DATE,
DEVICE_CLASS_TIMESTAMP,
STATE_UNKNOWN, STATE_UNKNOWN,
) )
from homeassistant.core import callback from homeassistant.core import callback
@ -95,12 +93,12 @@ class MobileAppSensor(MobileAppEntity, SensorEntity):
if ( if (
self.device_class self.device_class
in ( in (
DEVICE_CLASS_DATE, SensorDeviceClass.DATE,
DEVICE_CLASS_TIMESTAMP, SensorDeviceClass.TIMESTAMP,
) )
and (timestamp := dt_util.parse_datetime(state)) is not None and (timestamp := dt_util.parse_datetime(state)) is not None
): ):
if self.device_class == DEVICE_CLASS_DATE: if self.device_class == SensorDeviceClass.DATE:
return timestamp.date() return timestamp.date()
return timestamp return timestamp