From f7f50563dddb6f4b6193774dc6375f7c48d48a98 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 9 Dec 2021 09:14:16 +0100 Subject: [PATCH] Use new DeviceClass enums in bloomsky (#61316) Co-authored-by: epenet --- homeassistant/components/bloomsky/binary_sensor.py | 4 ++-- homeassistant/components/bloomsky/sensor.py | 9 ++++++--- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/bloomsky/binary_sensor.py b/homeassistant/components/bloomsky/binary_sensor.py index 2ef4586f537..2b8cdd57c95 100644 --- a/homeassistant/components/bloomsky/binary_sensor.py +++ b/homeassistant/components/bloomsky/binary_sensor.py @@ -2,8 +2,8 @@ import voluptuous as vol from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_MOISTURE, PLATFORM_SCHEMA, + BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.const import CONF_MONITORED_CONDITIONS @@ -11,7 +11,7 @@ import homeassistant.helpers.config_validation as cv from . import DOMAIN -SENSOR_TYPES = {"Rain": DEVICE_CLASS_MOISTURE, "Night": None} +SENSOR_TYPES = {"Rain": BinarySensorDeviceClass.MOISTURE, "Night": None} PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( { diff --git a/homeassistant/components/bloomsky/sensor.py b/homeassistant/components/bloomsky/sensor.py index 9a14145d1d8..f0bc31a337a 100644 --- a/homeassistant/components/bloomsky/sensor.py +++ b/homeassistant/components/bloomsky/sensor.py @@ -1,11 +1,14 @@ """Support the sensor of a BloomSky weather station.""" import voluptuous as vol -from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity +from homeassistant.components.sensor import ( + PLATFORM_SCHEMA, + SensorDeviceClass, + SensorEntity, +) from homeassistant.const import ( AREA_SQUARE_METERS, CONF_MONITORED_CONDITIONS, - DEVICE_CLASS_TEMPERATURE, ELECTRIC_POTENTIAL_MILLIVOLT, PERCENTAGE, PRESSURE_INHG, @@ -47,7 +50,7 @@ SENSOR_UNITS_METRIC = { # Device class SENSOR_DEVICE_CLASS = { - "Temperature": DEVICE_CLASS_TEMPERATURE, + "Temperature": SensorDeviceClass.TEMPERATURE, } # Which sensors to format numerically