From 526b29ca6709c978f98842ed19ae1d97464936ee Mon Sep 17 00:00:00 2001
From: Marti Markov <1858736+martimarkov@users.noreply.github.com>
Date: Sun, 7 Nov 2021 17:07:43 +0000
Subject: [PATCH] Add measurement to TrafficSensor

Enables the usage of TrafficSensor within the statistics card
---
 custom_components/mikrotik_router/sensor.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/custom_components/mikrotik_router/sensor.py b/custom_components/mikrotik_router/sensor.py
index ba96901..8520d95 100644
--- a/custom_components/mikrotik_router/sensor.py
+++ b/custom_components/mikrotik_router/sensor.py
@@ -478,6 +478,11 @@ class MikrotikControllerTrafficSensor(MikrotikControllerSensor):
     def unique_id(self) -> str:
         """Return a unique id for this entity."""
         return f"{self._inst.lower()}-{self._sensor.lower()}-{self._data['default-name'].lower()}"
+        
+    @property
+    def state_class(self) -> str:
+        """Return the state_class"""
+        return f"measurement"
 
     @property
     def device_info(self) -> Dict[str, Any]: