improve debug
This commit is contained in:
@@ -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
|
RUN pip install --no-cache-dir paho-mqtt opencv-python-headless numpy requests
|
||||||
|
|
||||||
# Copy the dayglo detector script
|
# Copy the dayglo detector script
|
||||||
COPY dayglo_detector.py /app/dayglo_detector.py
|
COPY dayglo_detector.py test_mqtt.py /app/
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|
||||||
|
|||||||
@@ -8,6 +8,8 @@ import json
|
|||||||
# Configuration
|
# Configuration
|
||||||
MQTT_BROKER = os.environ.get('MQTT_BROKER', '10.59.221.172')
|
MQTT_BROKER = os.environ.get('MQTT_BROKER', '10.59.221.172')
|
||||||
MQTT_PORT = int(os.environ.get('MQTT_PORT', '1883'))
|
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_SNAPSHOT_TOPIC = "/frigate/patiocam/person/snapshot"
|
||||||
MQTT_TOPIC_PUBLISH = "homeassistant/sensor/dayglo_rating/state"
|
MQTT_TOPIC_PUBLISH = "homeassistant/sensor/dayglo_rating/state"
|
||||||
DAYGLO_THRESHOLD_TOPIC = "homeassistant/sensor/dayglo_threshold/state"
|
DAYGLO_THRESHOLD_TOPIC = "homeassistant/sensor/dayglo_threshold/state"
|
||||||
@@ -71,7 +73,9 @@ def calculate_dayglo_rating(image):
|
|||||||
rating = (dayglo_pixels / total_pixels) * 100
|
rating = (dayglo_pixels / total_pixels) * 100
|
||||||
return rating
|
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_connect = on_connect
|
||||||
client.on_message = on_message
|
client.on_message = on_message
|
||||||
client.connect(MQTT_BROKER, MQTT_PORT, 60)
|
client.connect(MQTT_BROKER, MQTT_PORT, 60)
|
||||||
|
|||||||
@@ -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()
|
||||||
|
|
||||||
Reference in New Issue
Block a user