From d7e52c0843c09d418536f2bd49de741c388f3f81 Mon Sep 17 00:00:00 2001
From: Daniel Chesterton <daniel@chestertondevelopment.com>
Date: Sun, 18 Jul 2021 21:23:01 +0100
Subject: [PATCH] Add STORAGE_POLL_INTERVAL environment variable

---
 .github/workflows/publish.yml |  2 +
 README.md                     |  1 +
 src/amcrest2mqtt.py           | 82 ++++++++++++++++++-----------------
 3 files changed, 45 insertions(+), 40 deletions(-)

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 1415b9d..3cd7640 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -27,6 +27,8 @@ jobs:
                   ref: main
             - name: Set up QEMU
               uses: docker/setup-qemu-action@v1.2.0
+              with:
+                  platforms: linux/386,linux/amd64,linux/arm/v6,linux/arm/v7,linux/arm64/v8,linux/ppc64le,linux/s390x
             - name: Set up Docker Buildx
               uses: docker/setup-buildx-action@v1.5.1
             - name: Login to DockerHub
diff --git a/README.md b/README.md
index 335a4a8..a239d3a 100644
--- a/README.md
+++ b/README.md
@@ -16,6 +16,7 @@ It supports the following environment variables:
 -   `MQTT_PORT` (optional, default = 1883)
 -   `HOME_ASSISTANT` (optional, default = false)
 -   `HOME_ASSISTANT_PREFIX` (optional, default = 'homeassistant')
+-   `STORAGE_POLL_INTERVAL` (optional, default = 3600) - how often to fetch storage data (in seconds)
 
 It exposes events to the following topics:
 
diff --git a/src/amcrest2mqtt.py b/src/amcrest2mqtt.py
index d21ec1d..fc48d73 100644
--- a/src/amcrest2mqtt.py
+++ b/src/amcrest2mqtt.py
@@ -8,7 +8,6 @@
 import signal
 from threading import Timer
 
-storage_sensors_interval = 60  # 1 hour
 is_exiting = False
 mqtt_client = None
 
@@ -17,6 +16,7 @@
 amcrest_port = int(os.getenv("AMCREST_PORT") or 80)
 amcrest_username = os.getenv("AMCREST_USERNAME") or "admin"
 amcrest_password = os.getenv("AMCREST_PASSWORD")
+storage_poll_interval = int(os.getenv("STORAGE_POLL_INTERVAL") or 3600)
 
 mqtt_host = os.getenv("MQTT_HOST") or "localhost"
 mqtt_qos = int(os.getenv("MQTT_QOS") or 0)
@@ -78,9 +78,9 @@ def exit_gracefully(rc, skip_mqtt=False):
     os._exit(rc)
 
 def refresh_storage_sensors():
-    global camera, topics, storage_sensors_interval
+    global camera, topics, storage_poll_interval
 
-    Timer(storage_sensors_interval, refresh_storage_sensors).start()
+    Timer(storage_poll_interval, refresh_storage_sensors).start()
     log("Fetching storage sensors...")
 
     try:
@@ -239,44 +239,45 @@ def signal_handler(sig, frame):
         json=True,
     )
 
-    mqtt_publish(
-        topics["home_assistant"]["storage_used_percent"],
-        base_config
-        | {
-            "state_topic": topics["storage_used_percent"],
-            "unit_of_measurement": "%",
-            "icon": "mdi:micro-sd",
-            "name": f"{device_name} Storage Used %",
-            "unique_id": f"{serial_number}.storage_used_percent",
-        },
-        json=True,
-    )
+    if storage_poll_interval > 0:
+        mqtt_publish(
+            topics["home_assistant"]["storage_used_percent"],
+            base_config
+            | {
+                "state_topic": topics["storage_used_percent"],
+                "unit_of_measurement": "%",
+                "icon": "mdi:micro-sd",
+                "name": f"{device_name} Storage Used %",
+                "unique_id": f"{serial_number}.storage_used_percent",
+            },
+            json=True,
+        )
 
-    mqtt_publish(
-        topics["home_assistant"]["storage_used"],
-        base_config
-        | {
-            "state_topic": topics["storage_used"],
-            "unit_of_measurement": "GB",
-            "icon": "mdi:micro-sd",
-            "name": f"{device_name} Storage Used",
-            "unique_id": f"{serial_number}.storage_used",
-        },
-        json=True,
-    )
+        mqtt_publish(
+            topics["home_assistant"]["storage_used"],
+            base_config
+            | {
+                "state_topic": topics["storage_used"],
+                "unit_of_measurement": "GB",
+                "icon": "mdi:micro-sd",
+                "name": f"{device_name} Storage Used",
+                "unique_id": f"{serial_number}.storage_used",
+            },
+            json=True,
+        )
 
-    mqtt_publish(
-        topics["home_assistant"]["storage_total"],
-        base_config
-        | {
-            "state_topic": topics["storage_total"],
-            "unit_of_measurement": "GB",
-            "icon": "mdi:micro-sd",
-            "name": f"{device_name} Storage Total",
-            "unique_id": f"{serial_number}.storage_total",
-        },
-        json=True,
-    )
+        mqtt_publish(
+            topics["home_assistant"]["storage_total"],
+            base_config
+            | {
+                "state_topic": topics["storage_total"],
+                "unit_of_measurement": "GB",
+                "icon": "mdi:micro-sd",
+                "name": f"{device_name} Storage Total",
+                "unique_id": f"{serial_number}.storage_total",
+            },
+            json=True,
+        )
 
 # Main loop
 mqtt_publish(topics["status"], "online")
@@ -288,7 +289,8 @@ def signal_handler(sig, frame):
     "serial_number": serial_number,
 }, json=True)
 
-refresh_storage_sensors()
+if storage_poll_interval > 0:
+    refresh_storage_sensors()
 
 log("Listening for events...")