From 53e95c9d57c447cb87e34daa1e07813f8d79a5d7 Mon Sep 17 00:00:00 2001
From: tomaae <23486452+tomaae@users.noreply.github.com>
Date: Sat, 12 Dec 2020 12:30:36 +0100
Subject: [PATCH] Sensor nat selection #68

---
 .../mikrotik_router/mikrotik_controller.py           | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/custom_components/mikrotik_router/mikrotik_controller.py b/custom_components/mikrotik_router/mikrotik_controller.py
index 8750082..0226f9f 100644
--- a/custom_components/mikrotik_router/mikrotik_controller.py
+++ b/custom_components/mikrotik_router/mikrotik_controller.py
@@ -39,6 +39,8 @@ from .const import (
     DEFAULT_SENSOR_CLIENT_TRAFFIC,
     CONF_SENSOR_SIMPLE_QUEUES,
     DEFAULT_SENSOR_SIMPLE_QUEUES,
+    CONF_SENSOR_NAT,
+    DEFAULT_SENSOR_NAT,
     CONF_SENSOR_SCRIPTS,
     DEFAULT_SENSOR_SCRIPTS,
 )
@@ -181,6 +183,14 @@ class MikrotikControllerData:
             CONF_SENSOR_SIMPLE_QUEUES, DEFAULT_SENSOR_SIMPLE_QUEUES
         )
 
+    # ---------------------------
+    #   option_sensor_nat
+    # ---------------------------
+    @property
+    def option_sensor_nat(self):
+        """Config entry option to not track ARP."""
+        return self.config_entry.options.get(CONF_SENSOR_NAT, DEFAULT_SENSOR_NAT)
+
     # ---------------------------
     #   option_sensor_scripts
     # ---------------------------
@@ -465,7 +475,7 @@ class MikrotikControllerData:
         if self.api.connected():
             await self.hass.async_add_executor_job(self.process_interface_client)
 
-        if self.api.connected():
+        if self.api.connected() and self.option_sensor_nat:
             await self.hass.async_add_executor_job(self.get_nat)
 
         if self.api.connected():