From f111131a9b2917d7f108d6ac0cf6c9c0545629b7 Mon Sep 17 00:00:00 2001
From: tomaae <23486452+tomaae@users.noreply.github.com>
Date: Wed, 11 Mar 2020 23:25:51 +0100
Subject: [PATCH] changed traffic sensors to kbps #10

---
 custom_components/mikrotik_router/mikrotik_controller.py | 6 ++++++
 custom_components/mikrotik_router/sensor.py              | 4 ++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/custom_components/mikrotik_router/mikrotik_controller.py b/custom_components/mikrotik_router/mikrotik_controller.py
index 23acd93..c4384c1 100644
--- a/custom_components/mikrotik_router/mikrotik_controller.py
+++ b/custom_components/mikrotik_router/mikrotik_controller.py
@@ -220,6 +220,12 @@ class MikrotikControllerData():
             ]
         )
 
+        for uid in self.data['interface']:
+            self.data['interface'][uid]['rx-bits-per-second'] = round(
+                self.data['interface'][uid]['rx-bits-per-second'] / 1000)
+            self.data['interface'][uid]['tx-bits-per-second'] = round(
+                self.data['interface'][uid]['tx-bits-per-second'] / 1000)
+
     # ---------------------------
     #   get_interface_client
     # ---------------------------
diff --git a/custom_components/mikrotik_router/sensor.py b/custom_components/mikrotik_router/sensor.py
index 68011ab..a83fff6 100644
--- a/custom_components/mikrotik_router/sensor.py
+++ b/custom_components/mikrotik_router/sensor.py
@@ -58,7 +58,7 @@ SENSOR_TYPES = {
         ATTR_DEVICE_CLASS: None,
         ATTR_ICON: "mdi:upload-network-outline",
         ATTR_LABEL: 'TX',
-        ATTR_UNIT: "bps",
+        ATTR_UNIT: "kbps",
         ATTR_PATH: "interface",
         ATTR_ATTR: "tx-bits-per-second",
     },
@@ -66,7 +66,7 @@ SENSOR_TYPES = {
         ATTR_DEVICE_CLASS: None,
         ATTR_ICON: "mdi:download-network-outline",
         ATTR_LABEL: 'RX',
-        ATTR_UNIT: "bps",
+        ATTR_UNIT: "kbps",
         ATTR_PATH: "interface",
         ATTR_ATTR: "rx-bits-per-second",
     },