Add AMCREST_POLL_STORAGE option to allow user to skip polling storage information.

README.md

@@ -9,6 +9,7 @@ It supports the following environment variables:
 -   `AMCREST_PORT` (optional, default = 80)
 -   `AMCREST_USERNAME` (optional, default = admin)
 -   `AMCREST_PASSWORD` (required)
+-   `AMCREST_PROBE_STORAGE` (optional, default = true)
 -   `MQTT_USERNAME` (required)
 -   `MQTT_PASSWORD` (optional, default = empty password)
 -   `MQTT_HOST` (optional, default = 'localhost')

src/amcrest2mqtt.py

@@ -17,6 +17,7 @@ amcrest_host = os.getenv("AMCREST_HOST")
 amcrest_port = int(os.getenv("AMCREST_PORT") or 80)
 amcrest_username = os.getenv("AMCREST_USERNAME") or "admin"
 amcrest_password = os.getenv("AMCREST_PASSWORD")
+amcrest_probe_storage = os.getenv("AMCREST_PROBE_STORAGE") == "true"
 
 mqtt_host = os.getenv("MQTT_HOST") or "localhost"
 mqtt_qos = int(os.getenv("MQTT_QOS") or 0)
@@ -239,44 +240,45 @@ if home_assistant:
         json=True,
     )
 
-    mqtt_publish(
+    if amcrest_probe_storage:
-        topics["home_assistant"]["storage_used_percent"],
+        mqtt_publish(
-        base_config
+            topics["home_assistant"]["storage_used_percent"],
-        | {
+            base_config
-            "state_topic": topics["storage_used_percent"],
+            | {
-            "unit_of_measurement": "%",
+                "state_topic": topics["storage_used_percent"],
-            "icon": "mdi:micro-sd",
+                "unit_of_measurement": "%",
-            "name": f"{device_name} Storage Used %",
+                "icon": "mdi:micro-sd",
-            "unique_id": f"{serial_number}.storage_used_percent",
+                "name": f"{device_name} Storage Used %",
-        },
+                "unique_id": f"{serial_number}.storage_used_percent",
-        json=True,
+            },
-    )
+            json=True,
+        )
 
-    mqtt_publish(
+        mqtt_publish(
-        topics["home_assistant"]["storage_used"],
+            topics["home_assistant"]["storage_used"],
-        base_config
+            base_config
-        | {
+            | {
-            "state_topic": topics["storage_used"],
+                "state_topic": topics["storage_used"],
-            "unit_of_measurement": "GB",
+                "unit_of_measurement": "GB",
-            "icon": "mdi:micro-sd",
+                "icon": "mdi:micro-sd",
-            "name": f"{device_name} Storage Used",
+                "name": f"{device_name} Storage Used",
-            "unique_id": f"{serial_number}.storage_used",
+                "unique_id": f"{serial_number}.storage_used",
-        },
+            },
-        json=True,
+            json=True,
-    )
+        )
 
-    mqtt_publish(
+        mqtt_publish(
-        topics["home_assistant"]["storage_total"],
+            topics["home_assistant"]["storage_total"],
-        base_config
+            base_config
-        | {
+            | {
-            "state_topic": topics["storage_total"],
+                "state_topic": topics["storage_total"],
-            "unit_of_measurement": "GB",
+                "unit_of_measurement": "GB",
-            "icon": "mdi:micro-sd",
+                "icon": "mdi:micro-sd",
-            "name": f"{device_name} Storage Total",
+                "name": f"{device_name} Storage Total",
-            "unique_id": f"{serial_number}.storage_total",
+                "unique_id": f"{serial_number}.storage_total",
-        },
+            },
-        json=True,
+            json=True,
-    )
+        )
 
 # Main loop
 mqtt_publish(topics["status"], "online")
@@ -288,7 +290,8 @@ mqtt_publish(topics["config"], {
     "serial_number": serial_number,
 }, json=True)
 
-refresh_storage_sensors()
+if amcrest_probe_storage:
+    refresh_storage_sensors()
 
 log("Listening for events...")