Merge pull request #19 from w1tw0lf/update-interval

Allow the user to set the update interval for polling
w1tw0lf · Feb 5, 2025 · 13f0271 · 13f0271
2 parents b0cd596 + aaf7b48
commit 13f0271
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 5 deletions.
diff --git a/custom_components/unifi_mqtt/__init__.py b/custom_components/unifi_mqtt/__init__.py
@@ -32,7 +32,8 @@
     CONF_PORT,
     CONF_VERIFY_SSL,
     CONF_VERSION,
-    UPDATE_INTERVAL,
+    CONF_UPDATE_INTERVAL,
+    DEFAULT_UPDATE_INTERVAL
 )
 
 _LOGGER = logging.getLogger(__name__)
@@ -49,6 +50,7 @@ async def async_setup_entry(hass, entry):
     port = entry.data[CONF_PORT]
     verify_ssl = entry.data[CONF_VERIFY_SSL]
     version = entry.data[CONF_VERSION]
+    update_interval = entry.data.get(CONF_UPDATE_INTERVAL, DEFAULT_UPDATE_INTERVAL)
 
     def init_controller():
         return Controller(
@@ -72,6 +74,11 @@ async def update_unifi_data(now):
         active_devices = []
 
         for device in unifi_devices:
+            # If the device is not a dictionary, log its value and skip processing it.
+            if not isinstance(device, dict):
+                _LOGGER.error("Received device that is not a dictionary: %s", device)
+                continue
+
             if not device.get("adopted"):
                 continue
 
@@ -282,7 +289,7 @@ async def update_unifi_data(now):
 
     global UPDATE_LISTENER
     UPDATE_LISTENER = async_track_time_interval(
-        hass, update_unifi_data, timedelta(seconds=UPDATE_INTERVAL)
+        hass, update_unifi_data, timedelta(seconds=update_interval)
     )
     hass.async_create_task(update_unifi_data(None))
 

diff --git a/custom_components/unifi_mqtt/config_flow.py b/custom_components/unifi_mqtt/config_flow.py
@@ -14,11 +14,13 @@
     CONF_PORT,
     CONF_VERIFY_SSL,
     CONF_VERSION,
+    CONF_UPDATE_INTERVAL,
     VERSION_OPTIONS,
     DEFAULT_SITE_ID,
     DEFAULT_PORT,
     DEFAULT_VERIFY_SSL,
     DEFAULT_VERSION,
+    DEFAULT_UPDATE_INTERVAL
 )
 
 _LOGGER = logging.getLogger(__name__)
@@ -31,10 +33,11 @@
     vol.Optional(CONF_PORT, default=DEFAULT_PORT): cv.port,
     vol.Optional(CONF_VERIFY_SSL, default=DEFAULT_VERIFY_SSL): cv.boolean,
     vol.Optional(CONF_VERSION, default=DEFAULT_VERSION): vol.In(VERSION_OPTIONS),
+    vol.Optional(CONF_UPDATE_INTERVAL, default=DEFAULT_UPDATE_INTERVAL): cv.positive_int,
 })
 
 class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
-    """Handle a config flow for UniFi MQTT integration."""
+    """Handle a config flow for UniFi Device Info integration."""
 
     VERSION = 1
 

diff --git a/custom_components/unifi_mqtt/const.py b/custom_components/unifi_mqtt/const.py
@@ -9,6 +9,7 @@
 CONF_PORT = "port"
 CONF_VERIFY_SSL = "Verify SSL"
 CONF_VERSION = "version"
+CONF_UPDATE_INTERVAL = "Update interval"
 
 VERSION_OPTIONS = [
     "UDMP-unifiOS",  # default
@@ -22,4 +23,4 @@
 DEFAULT_VERIFY_SSL = False
 DEFAULT_VERSION = "UDMP-unifiOS"
 
-UPDATE_INTERVAL = 60  # seconds
+DEFAULT_UPDATE_INTERVAL = 60  # seconds
diff --git a/custom_components/unifi_mqtt/manifest.json b/custom_components/unifi_mqtt/manifest.json
@@ -1,7 +1,7 @@
 {
   "domain": "unifi_mqtt",
   "name": "UniFi Device Info Integration",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "documentation": "https://github.com/w1tw0lf/Unifi-Device-info",
   "dependencies": [
     "mqtt"