Skip to content

Commit

Permalink
Add ping functionality to stop app hanging
Browse files Browse the repository at this point in the history
  • Loading branch information
dchesterton committed Aug 7, 2021
1 parent 0f8858c commit 8e96fc5
Showing 1 changed file with 25 additions and 9 deletions.
34 changes: 25 additions & 9 deletions src/amcrest2mqtt.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,8 @@ def on_mqtt_disconnect(client, userdata, rc):
def exit_gracefully(rc, skip_mqtt=False):
global topics, mqtt_client

log("Exiting app...")

if mqtt_client is not None and mqtt_client.is_connected() and skip_mqtt == False:
mqtt_publish(topics["status"], "offline", exit_on_error=False)
mqtt_client.loop_stop(force=True)
Expand All @@ -96,6 +98,14 @@ def refresh_storage_sensors():
except AmcrestError as error:
log(f"Error fetching storage information {error}", level="WARNING")

def ping_camera():
Timer(30, ping_camera).start()
response = os.system(f"ping -c1 -W100 {amcrest_host} >/dev/null 2>&1")

if response != 0:
log("Ping unsuccessful", level="ERROR")
exit_gracefully(1)

def signal_handler(sig, frame):
# exit immediately upon receiving a second SIGINT
global is_exiting
Expand Down Expand Up @@ -134,14 +144,18 @@ def signal_handler(sig, frame):
# Fetch camera details
log("Fetching camera details...")

device_type = camera.device_type.replace("type=", "").strip()
is_ad110 = device_type == "AD110"
is_ad410 = device_type == "AD410"
is_doorbell = is_ad110 or is_ad410
serial_number = camera.serial_number.strip()
sw_version = camera.software_information[0].replace("version=", "").strip()
device_name = camera.machine_name.replace("name=", "").strip()
device_slug = slugify(device_name, separator="_")
try:
device_type = camera.device_type.replace("type=", "").strip()
is_ad110 = device_type == "AD110"
is_ad410 = device_type == "AD410"
is_doorbell = is_ad110 or is_ad410
serial_number = camera.serial_number.strip()
sw_version = camera.software_information[0].replace("version=", "").strip()
device_name = camera.machine_name.replace("name=", "").strip()
device_slug = slugify(device_name, separator="_")
except AmcrestError as error:
log(f"Error fetching camera details", level="ERROR")
exit_gracefully(1)

log(f"Device type: {device_type}")
log(f"Serial number: {serial_number}")
Expand Down Expand Up @@ -316,10 +330,12 @@ def signal_handler(sig, frame):
if storage_poll_interval > 0:
refresh_storage_sensors()

ping_camera()

log("Listening for events...")

try:
for code, payload in camera.event_actions("All", retries=5):
for code, payload in camera.event_actions("All", retries=5, timeout_cmd=(10.00, 3600)):
if (is_ad110 and code == "ProfileAlarmTransmit") or (code == "VideoMotion" and not is_ad110):
motion_payload = "on" if payload["action"] == "Start" else "off"
mqtt_publish(topics["motion"], motion_payload)
Expand Down

0 comments on commit 8e96fc5

Please sign in to comment.