Use new enums in mobile_app (#61929)

pull/62130/head
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
1 changed files with 4 additions and 6 deletions

View File

@ -1,13 +1,11 @@
"""Sensor platform for mobile_app."""
from __future__ import annotations
from homeassistant.components.sensor import SensorEntity
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import (
CONF_NAME,
CONF_UNIQUE_ID,
CONF_WEBHOOK_ID,
DEVICE_CLASS_DATE,
DEVICE_CLASS_TIMESTAMP,
STATE_UNKNOWN,
)
from homeassistant.core import callback
@ -95,12 +93,12 @@ class MobileAppSensor(MobileAppEntity, SensorEntity):
if (
self.device_class
in (
DEVICE_CLASS_DATE,
DEVICE_CLASS_TIMESTAMP,
SensorDeviceClass.DATE,
SensorDeviceClass.TIMESTAMP,
)
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