From f82f9e9c3ec807d69dc2ba1775cdd39bf9f6e608 Mon Sep 17 00:00:00 2001
From: tomaae <23486452+tomaae@users.noreply.github.com>
Date: Mon, 13 Dec 2021 13:50:53 +0100
Subject: [PATCH] Added Device_class for firmware update binary sensor, fixes
 #137

---
 custom_components/mikrotik_router/binary_sensor.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/custom_components/mikrotik_router/binary_sensor.py b/custom_components/mikrotik_router/binary_sensor.py
index 57afdd5..ba1a363 100644
--- a/custom_components/mikrotik_router/binary_sensor.py
+++ b/custom_components/mikrotik_router/binary_sensor.py
@@ -11,6 +11,7 @@ from homeassistant.components.binary_sensor import (
 from homeassistant.const import (
     CONF_NAME,
     CONF_HOST,
+    ATTR_DEVICE_CLASS,
     ATTR_ATTRIBUTION,
 )
 from homeassistant.core import callback
@@ -37,6 +38,7 @@ ATTR_CTGR = "entity_category"
 
 SENSOR_TYPES = {
     "system_fwupdate": {
+        ATTR_DEVICE_CLASS: BinarySensorDeviceClass.UPDATE,
         ATTR_LABEL: "Firmware update",
         ATTR_GROUP: "System",
         ATTR_PATH: "fw-update",