Skip to content

Commit

Permalink
Add entity_category and fix bug with storage totals
Browse files Browse the repository at this point in the history
  • Loading branch information
dchesterton committed Dec 24, 2021
1 parent 766aa50 commit cc5e374
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions src/amcrest2mqtt.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,12 +92,16 @@ def refresh_storage_sensors():

try:
storage = camera.storage_all

mqtt_publish(topics["storage_used_percent"], str(storage["used_percent"]))
mqtt_publish(topics["storage_used"], str(storage["used"][0]))
mqtt_publish(topics["storage_total"], str(storage["total"][0]))
mqtt_publish(topics["storage_used"], to_gb(storage["used"]))
mqtt_publish(topics["storage_total"], to_gb(storage["total"]))
except AmcrestError as error:
log(f"Error fetching storage information {error}", level="WARNING")

def to_gb(total):
return str(round(float(total[0]) / 1024 / 1024 / 1024, 2))

def ping_camera():
Timer(30, ping_camera).start()
response = os.system(f"ping -c1 -W100 {amcrest_host} >/dev/null 2>&1")
Expand Down Expand Up @@ -292,6 +296,7 @@ def signal_handler(sig, frame):
"icon": "mdi:micro-sd",
"name": f"{device_name} Storage Used %",
"unique_id": f"{serial_number}.storage_used_percent",
"entity_category": "diagnostic",
},
json=True,
)
Expand All @@ -305,6 +310,7 @@ def signal_handler(sig, frame):
"icon": "mdi:micro-sd",
"name": f"{device_name} Storage Used",
"unique_id": f"{serial_number}.storage_used",
"entity_category": "diagnostic",
},
json=True,
)
Expand All @@ -318,6 +324,7 @@ def signal_handler(sig, frame):
"icon": "mdi:micro-sd",
"name": f"{device_name} Storage Total",
"unique_id": f"{serial_number}.storage_total",
"entity_category": "diagnostic",
},
json=True,
)
Expand Down

0 comments on commit cc5e374

Please sign in to comment.