From 87425d4ab1b7be00c9405bc63fe912230d838f7c Mon Sep 17 00:00:00 2001 From: Marius <33354141+Mariusthvdb@users.noreply.github.com> Date: Sun, 30 Aug 2020 11:08:57 +0200 Subject: [PATCH] Add device_class safety to synology_dsm storage binary_sensors (#39310) * add device_class: safety to storage binary_sensors * Update binary_sensor.py * Update homeassistant/components/synology_dsm/binary_sensor.py Co-authored-by: Martin Hjelmare * Update binary_sensor.py * Import device_class Safety from homeassistant.components.binary_sensor * Update binary_sensor.py Co-authored-by: Martin Hjelmare --- .../components/synology_dsm/binary_sensor.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/synology_dsm/binary_sensor.py b/homeassistant/components/synology_dsm/binary_sensor.py index a75f57db678..c95b4298f5d 100644 --- a/homeassistant/components/synology_dsm/binary_sensor.py +++ b/homeassistant/components/synology_dsm/binary_sensor.py @@ -1,7 +1,10 @@ """Support for Synology DSM binary sensors.""" from typing import Dict -from homeassistant.components.binary_sensor import BinarySensorEntity +from homeassistant.components.binary_sensor import ( + DEVICE_CLASS_SAFETY, + BinarySensorEntity, +) from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_DISKS from homeassistant.helpers.typing import HomeAssistantType @@ -14,6 +17,8 @@ from .const import ( SYNO_API, ) +DEFAULT_DEVICE_CLASS = DEVICE_CLASS_SAFETY + async def async_setup_entry( hass: HomeAssistantType, entry: ConfigEntry, async_add_entities @@ -71,3 +76,8 @@ class SynoDSMStorageBinarySensor(SynologyDSMDeviceEntity, BinarySensorEntity): if attr is None: return None return attr + + @property + def device_class(self): + """Return the device class of this binary sensor.""" + return DEFAULT_DEVICE_CLASS