Skip to content

Commit

Permalink
Add STORAGE_POLL_INTERVAL environment variable
Browse files Browse the repository at this point in the history
  • Loading branch information
dchesterton committed Jul 18, 2021
1 parent dc6b6c0 commit d7e52c0
Show file tree
Hide file tree
Showing 3 changed files with 45 additions and 40 deletions.
2 changes: 2 additions & 0 deletions .github/workflows/publish.yml
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,8 @@ jobs:
ref: main
- name: Set up QEMU
uses: docker/[email protected]
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/[email protected]
- name: Login to DockerHub
Expand Down
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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:

Expand Down
82 changes: 42 additions & 40 deletions src/amcrest2mqtt.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
import signal
from threading import Timer

storage_sensors_interval = 60 # 1 hour
is_exiting = False
mqtt_client = None

Expand All @@ -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)
Expand Down Expand Up @@ -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:
Expand Down Expand Up @@ -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")
Expand All @@ -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...")

Expand Down

0 comments on commit d7e52c0

Please sign in to comment.