diff --git a/homeassistant/components/fibaro/binary_sensor.py b/homeassistant/components/fibaro/binary_sensor.py index 96c688bac16..d6382e6f433 100644 --- a/homeassistant/components/fibaro/binary_sensor.py +++ b/homeassistant/components/fibaro/binary_sensor.py @@ -1,10 +1,7 @@ """Support for Fibaro binary sensors.""" from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_DOOR, - DEVICE_CLASS_MOTION, - DEVICE_CLASS_SMOKE, - DEVICE_CLASS_WINDOW, DOMAIN, + BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.const import CONF_DEVICE_CLASS, CONF_ICON @@ -13,11 +10,15 @@ from . import FIBARO_DEVICES, FibaroDevice SENSOR_TYPES = { "com.fibaro.floodSensor": ["Flood", "mdi:water", "flood"], - "com.fibaro.motionSensor": ["Motion", "mdi:run", DEVICE_CLASS_MOTION], - "com.fibaro.doorSensor": ["Door", "mdi:window-open", DEVICE_CLASS_DOOR], - "com.fibaro.windowSensor": ["Window", "mdi:window-open", DEVICE_CLASS_WINDOW], - "com.fibaro.smokeSensor": ["Smoke", "mdi:smoking", DEVICE_CLASS_SMOKE], - "com.fibaro.FGMS001": ["Motion", "mdi:run", DEVICE_CLASS_MOTION], + "com.fibaro.motionSensor": ["Motion", "mdi:run", BinarySensorDeviceClass.MOTION], + "com.fibaro.doorSensor": ["Door", "mdi:window-open", BinarySensorDeviceClass.DOOR], + "com.fibaro.windowSensor": [ + "Window", + "mdi:window-open", + BinarySensorDeviceClass.WINDOW, + ], + "com.fibaro.smokeSensor": ["Smoke", "mdi:smoking", BinarySensorDeviceClass.SMOKE], + "com.fibaro.FGMS001": ["Motion", "mdi:run", BinarySensorDeviceClass.MOTION], "com.fibaro.heatDetector": ["Heat", "mdi:fire", "heat"], } diff --git a/homeassistant/components/fibaro/sensor.py b/homeassistant/components/fibaro/sensor.py index a4b4e744af7..b96d0b23ecc 100644 --- a/homeassistant/components/fibaro/sensor.py +++ b/homeassistant/components/fibaro/sensor.py @@ -1,13 +1,9 @@ """Support for Fibaro sensors.""" from contextlib import suppress -from homeassistant.components.sensor import DOMAIN, SensorEntity +from homeassistant.components.sensor import DOMAIN, SensorDeviceClass, SensorEntity from homeassistant.const import ( CONCENTRATION_PARTS_PER_MILLION, - DEVICE_CLASS_CO2, - DEVICE_CLASS_HUMIDITY, - DEVICE_CLASS_ILLUMINANCE, - DEVICE_CLASS_TEMPERATURE, LIGHT_LUX, PERCENTAGE, TEMP_CELSIUS, @@ -21,7 +17,7 @@ SENSOR_TYPES = { "Temperature", None, None, - DEVICE_CLASS_TEMPERATURE, + SensorDeviceClass.TEMPERATURE, ], "com.fibaro.smokeSensor": [ "Smoke", @@ -34,15 +30,15 @@ SENSOR_TYPES = { CONCENTRATION_PARTS_PER_MILLION, None, None, - DEVICE_CLASS_CO2, + SensorDeviceClass.CO2, ], "com.fibaro.humiditySensor": [ "Humidity", PERCENTAGE, None, - DEVICE_CLASS_HUMIDITY, + SensorDeviceClass.HUMIDITY, ], - "com.fibaro.lightSensor": ["Light", LIGHT_LUX, None, DEVICE_CLASS_ILLUMINANCE], + "com.fibaro.lightSensor": ["Light", LIGHT_LUX, None, SensorDeviceClass.ILLUMINANCE], }