blakeblackshear.frigate/frigate/mqtt.py

154 lines
5.8 KiB
Python
Raw Normal View History

2020-11-04 04:26:39 +01:00
import logging
2020-11-04 13:28:07 +01:00
import threading
2020-11-01 15:06:15 +01:00
2020-11-04 13:31:25 +01:00
import paho.mqtt.client as mqtt
from frigate.config import FrigateConfig
2020-11-03 15:15:58 +01:00
2020-11-04 04:26:39 +01:00
logger = logging.getLogger(__name__)
2021-02-17 14:23:32 +01:00
2021-01-16 03:52:59 +01:00
def create_mqtt_client(config: FrigateConfig, camera_metrics):
mqtt_config = config.mqtt
def on_clips_command(client, userdata, message):
payload = message.payload.decode()
logger.debug(f"on_clips_toggle: {message.topic} {payload}")
2021-02-17 14:23:32 +01:00
camera_name = message.topic.split("/")[-3]
2020-12-24 05:16:36 +01:00
clips_settings = config.cameras[camera_name].clips
2021-02-17 14:23:32 +01:00
if payload == "ON":
2020-12-24 05:16:36 +01:00
if not clips_settings.enabled:
logger.info(f"Turning on clips for {camera_name} via mqtt")
clips_settings.enabled = True
2021-02-17 14:23:32 +01:00
elif payload == "OFF":
2020-12-24 05:16:36 +01:00
if clips_settings.enabled:
logger.info(f"Turning off clips for {camera_name} via mqtt")
clips_settings.enabled = False
2020-12-24 05:16:36 +01:00
else:
logger.warning(f"Received unsupported value at {message.topic}: {payload}")
2021-01-19 14:41:17 +01:00
state_topic = f"{message.topic[:-4]}/state"
client.publish(state_topic, payload, retain=True)
2020-12-24 05:16:36 +01:00
def on_snapshots_command(client, userdata, message):
payload = message.payload.decode()
logger.debug(f"on_snapshots_toggle: {message.topic} {payload}")
2021-02-17 14:23:32 +01:00
camera_name = message.topic.split("/")[-3]
2020-12-24 05:16:36 +01:00
snapshots_settings = config.cameras[camera_name].snapshots
2021-02-17 14:23:32 +01:00
if payload == "ON":
2020-12-24 05:16:36 +01:00
if not snapshots_settings.enabled:
logger.info(f"Turning on snapshots for {camera_name} via mqtt")
snapshots_settings.enabled = True
2021-02-17 14:23:32 +01:00
elif payload == "OFF":
2020-12-24 05:16:36 +01:00
if snapshots_settings.enabled:
logger.info(f"Turning off snapshots for {camera_name} via mqtt")
snapshots_settings.enabled = False
else:
logger.warning(f"Received unsupported value at {message.topic}: {payload}")
2021-01-19 14:41:17 +01:00
state_topic = f"{message.topic[:-4]}/state"
client.publish(state_topic, payload, retain=True)
2021-02-17 14:23:32 +01:00
2021-01-16 03:52:59 +01:00
def on_detect_command(client, userdata, message):
payload = message.payload.decode()
logger.debug(f"on_detect_toggle: {message.topic} {payload}")
2021-02-17 14:23:32 +01:00
camera_name = message.topic.split("/")[-3]
2021-01-16 03:52:59 +01:00
detect_settings = config.cameras[camera_name].detect
2021-02-17 14:23:32 +01:00
if payload == "ON":
2021-01-16 03:52:59 +01:00
if not camera_metrics[camera_name]["detection_enabled"].value:
logger.info(f"Turning on detection for {camera_name} via mqtt")
camera_metrics[camera_name]["detection_enabled"].value = True
detect_settings.enabled = True
2021-02-17 14:23:32 +01:00
elif payload == "OFF":
2021-01-16 03:52:59 +01:00
if camera_metrics[camera_name]["detection_enabled"].value:
logger.info(f"Turning off detection for {camera_name} via mqtt")
camera_metrics[camera_name]["detection_enabled"].value = False
detect_settings.enabled = False
2021-01-16 03:52:59 +01:00
else:
logger.warning(f"Received unsupported value at {message.topic}: {payload}")
2021-01-19 14:41:17 +01:00
state_topic = f"{message.topic[:-4]}/state"
client.publish(state_topic, payload, retain=True)
2020-11-01 15:06:15 +01:00
def on_connect(client, userdata, flags, rc):
2020-11-04 13:28:07 +01:00
threading.current_thread().name = "mqtt"
2020-11-01 15:06:15 +01:00
if rc != 0:
if rc == 3:
2020-11-04 04:26:39 +01:00
logger.error("MQTT Server unavailable")
2020-11-01 15:06:15 +01:00
elif rc == 4:
2020-11-04 04:26:39 +01:00
logger.error("MQTT Bad username or password")
2020-11-01 15:06:15 +01:00
elif rc == 5:
2020-11-04 04:26:39 +01:00
logger.error("MQTT Not authorized")
2020-11-01 15:06:15 +01:00
else:
2021-02-17 14:23:32 +01:00
logger.error(
"Unable to connect to MQTT: Connection refused. Error code: "
+ str(rc)
)
2020-11-04 13:28:07 +01:00
logger.info("MQTT connected")
client.subscribe(f"{mqtt_config.topic_prefix}/#")
2021-02-17 14:23:32 +01:00
client.publish(mqtt_config.topic_prefix + "/available", "online", retain=True)
2021-02-17 14:23:32 +01:00
client = mqtt.Client(client_id=mqtt_config.client_id)
2020-11-01 15:06:15 +01:00
client.on_connect = on_connect
2021-02-17 14:23:32 +01:00
client.will_set(
mqtt_config.topic_prefix + "/available", payload="offline", qos=1, retain=True
)
# register callbacks
for name in config.cameras.keys():
2021-02-17 14:23:32 +01:00
client.message_callback_add(
f"{mqtt_config.topic_prefix}/{name}/clips/set", on_clips_command
)
client.message_callback_add(
f"{mqtt_config.topic_prefix}/{name}/snapshots/set", on_snapshots_command
)
client.message_callback_add(
f"{mqtt_config.topic_prefix}/{name}/detect/set", on_detect_command
)
if not mqtt_config.tls_ca_certs is None:
if not mqtt_config.tls_client_cert is None and not mqtt_config.tls_client_key is None:
client.tls_set(mqtt_config.tls_ca_certs, mqtt_config.tls_client_cert, mqtt_config.tls_client_key)
else:
client.tls_set(mqtt_config.tls_ca_certs)
if not mqtt_config.tls_insecure is None:
client.tls_insecure_set(mqtt_config.tls_insecure)
if not mqtt_config.user is None:
client.username_pw_set(mqtt_config.user, password=mqtt_config.password)
2020-12-05 16:47:43 +01:00
try:
client.connect(mqtt_config.host, mqtt_config.port, 60)
2020-12-05 16:47:43 +01:00
except Exception as e:
logger.error(f"Unable to connect to MQTT server: {e}")
raise
2020-11-01 15:06:15 +01:00
client.loop_start()
2021-01-16 17:22:24 +01:00
for name in config.cameras.keys():
2021-02-17 14:23:32 +01:00
client.publish(
f"{mqtt_config.topic_prefix}/{name}/clips/state",
"ON" if config.cameras[name].clips.enabled else "OFF",
retain=True,
)
client.publish(
f"{mqtt_config.topic_prefix}/{name}/snapshots/state",
"ON" if config.cameras[name].snapshots.enabled else "OFF",
retain=True,
)
client.publish(
f"{mqtt_config.topic_prefix}/{name}/detect/state",
"ON" if config.cameras[name].detect.enabled else "OFF",
retain=True,
)
2020-11-04 13:31:25 +01:00
return client