Compose/frigate
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix indentation issues in main function for proper script execution

Browse Source 
- Corrected inconsistent indentation in the main function to avoid runtime `IndentationError`.
- Ensured the MQTT client properly connects and publishes when processing images from the command line.
This commit is contained in:
Marcus Brown
2024-10-20 19:08:00 +11:00
parent fe209a9c2a
commit b2db064021
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dayglo_detector/dayglo_detector.py
+6 -6
View File
@@ -89,8 +89,8 @@ async def process_snapshot(client, payload):
# Attempt to read the saved image with OpenCV # Attempt to read the saved image with OpenCV
image = cv2.imread(temp_image_path) image = cv2.imread(temp_image_path)
if image is not None: if image is not None:
rating = calculate_dayglo_rating(image) rating = calculate_dayglo_rating(image)
print("Dayglo Rating calculated:", rating) print("Dayglo Rating calculated:", rating)
await publish_rating(client, rating) await publish_rating(client, rating)
else: else:
@@ -119,19 +119,19 @@ async def main():
# Handle command line argument for image file # Handle command line argument for image file
if len(sys.argv) > 1: if len(sys.argv) > 1:
image_file = sys.argv[1] image_file = sys.argv[1]
if os.path.exists(image_file): if os.path.exists(image_file):
client = DaygloDetectorMQTTClient("dayglo_detector") client = DaygloDetectorMQTTClient("dayglo_detector")
client.set_auth_credentials(MQTT_USERNAME, MQTT_PASSWORD) client.set_auth_credentials(MQTT_USERNAME, MQTT_PASSWORD)
await client.connect(MQTT_BROKER, MQTT_PORT, keepalive=60) await client.connect(MQTT_BROKER, MQTT_PORT, keepalive=60)
print(f"Processing image from file: {image_file}") print(f"Processing image from file: {image_file}")
image = cv2.imread(image_file) image = cv2.imread(image_file)
if image is not None: if image is not None:
rating = calculate_dayglo_rating(image) rating = calculate_dayglo_rating(image)
print("Dayglo Rating calculated from file:", rating) print("Dayglo Rating calculated from file:", rating)
await publish_rating(client, rating) await publish_rating(client, rating)
else: else:
print("Invalid image file provided.") print("Invalid image file provided.")
else: else:
print(f"File not found: {image_file}") print(f"File not found: {image_file}")
# Set up MQTT client for normal operation # Set up MQTT client for normal operation