improve debug

This commit is contained in:
2024-10-20 15:55:45 +11:00
parent 3fd01f4551
commit f5dee30509
3 changed files with 28 additions and 2 deletions
+1 -1
View File
@@ -10,7 +10,7 @@ RUN apt-get update && apt-get install -y \
RUN pip install --no-cache-dir paho-mqtt opencv-python-headless numpy requests
# Copy the dayglo detector script
COPY dayglo_detector.py /app/dayglo_detector.py
COPY dayglo_detector.py test_mqtt.py /app/
WORKDIR /app
+5 -1
View File
@@ -8,6 +8,8 @@ import json
# Configuration
MQTT_BROKER = os.environ.get('MQTT_BROKER', '10.59.221.172')
MQTT_PORT = int(os.environ.get('MQTT_PORT', '1883'))
MQTT_USERNAME = os.getenv('MQTT_USERNAME', 'your_username')
MQTT_PASSWORD = os.getenv('MQTT_PASSWORD', 'your_password')
MQTT_SNAPSHOT_TOPIC = "/frigate/patiocam/person/snapshot"
MQTT_TOPIC_PUBLISH = "homeassistant/sensor/dayglo_rating/state"
DAYGLO_THRESHOLD_TOPIC = "homeassistant/sensor/dayglo_threshold/state"
@@ -71,7 +73,9 @@ def calculate_dayglo_rating(image):
rating = (dayglo_pixels / total_pixels) * 100
return rating
client = mqtt.Client()
#client = mqtt.Client()
client = mqtt.Client(protocol=mqtt.MQTTv311)
client.username_pw_set(MQTT_USERNAME, MQTT_PASSWORD)
client.on_connect = on_connect
client.on_message = on_message
client.connect(MQTT_BROKER, MQTT_PORT, 60)
+22
View File
@@ -0,0 +1,22 @@
import os
import paho.mqtt.client as mqtt
# Configuration
MQTT_BROKER = os.environ.get('MQTT_BROKER', '10.59.221.172')
MQTT_PORT = int(os.environ.get('MQTT_PORT', '1883'))
MQTT_USERNAME = os.getenv('MQTT_USERNAME', 'your_username')
MQTT_PASSWORD = os.getenv('MQTT_PASSWORD', 'your_password')
client = mqtt.Client(protocol=mqtt.MQTTv311) # Ensure you are using the correct client version
client.username_pw_set(MQTT_USERNAME, MQTT_PASSWORD) # Set the username and password
def on_connect(client, userdata, flags, rc):
if rc == 0:
print("Connected successfully.")
else:
print(f"Connected with result code {rc}")
client.on_connect = on_connect
client.connect(MQTT_BROKER, MQTT_PORT, 60)
client.loop_forever()