chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

enable turning clips on and off via mqtt

Browse Source
This commit is contained in:
Blake Blackshear
2020-12-23 08:54:08 -06:00
parent 80a5a7b129
commit e5399ae07a
4 changed files with 45 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
frigate/mqtt.py
View File

@@ -3,12 +3,31 @@ import threading
import paho.mqtt.client as mqtt
from frigate.config import MqttConfig
from frigate.config import FrigateConfig
logger = logging.getLogger(__name__)
def create_mqtt_client(config: MqttConfig):
client = mqtt.Client(client_id=config.client_id)
def create_mqtt_client(config: FrigateConfig):
mqtt_config = config.mqtt
def on_clips_command(client, userdata, message):
payload = message.payload.decode()
logger.debug(f"on_clips_toggle: {message.topic} {payload}")
camera_name = message.topic.split('/')[-3]
command = message.topic.split('/')[-1]
if payload == 'ON':
config.cameras[camera_name].clips._enabled = True
elif payload == 'OFF':
config.cameras[camera_name].clips._enabled = False
else:
logger.warning(f"Received unsupported value at {message.topic}: {payload}")
if command == "set":
state_topic = f"{message.topic[:-4]}/state"
client.publish(state_topic, payload, retain=True)
def on_connect(client, userdata, flags, rc):
threading.current_thread().name = "mqtt"
if rc != 0:
@@ -22,15 +41,28 @@ def create_mqtt_client(config: MqttConfig):
logger.error("Unable to connect to MQTT: Connection refused. Error code: " + str(rc))
logger.info("MQTT connected")
client.publish(config.topic_prefix+'/available', 'online', retain=True)
client.publish(mqtt_config.topic_prefix+'/available', 'online', retain=True)
client = mqtt.Client(client_id=mqtt_config.client_id)
client.on_connect = on_connect
client.will_set(config.topic_prefix+'/available', payload='offline', qos=1, retain=True)
if not config.user is None:
client.username_pw_set(config.user, password=config.password)
client.will_set(mqtt_config.topic_prefix+'/available', payload='offline', qos=1, retain=True)
# register callbacks
for name in config.cameras.keys():
clips_topic = f"{mqtt_config.topic_prefix}/{name}/clips/#"
client.message_callback_add(clips_topic, on_clips_command)
if not mqtt_config.user is None:
client.username_pw_set(mqtt_config.user, password=mqtt_config.password)
try:
client.connect(config.host, config.port, 60)
client.connect(mqtt_config.host, mqtt_config.port, 60)
except Exception as e:
logger.error(f"Unable to connect to MQTT server: {e}")
raise
client.loop_start()
clips_topic = f"{mqtt_config.topic_prefix}/+/clips/#"
client.subscribe(clips_topic)
return client