Fix improper indentation in process_snapshot function

- Corrected inconsistent indentation to avoid runtime .
- Ensured that all indentation levels match Python's expected format.
This commit is contained in:
2024-10-20 19:10:44 +11:00
parent b2db064021
commit 296775bc2e
+12 -12
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:
@@ -118,21 +118,21 @@ async def main():
print("Starting Dayglo Detector...") print("Starting Dayglo Detector...")
# 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:
print("Invalid image file provided.")
else: else:
print("Invalid image file provided.") print(f"File not found: {image_file}")
else:
print(f"File not found: {image_file}")
# Set up MQTT client for normal operation # Set up MQTT client for normal operation
client = DaygloDetectorMQTTClient("dayglo_detector") client = DaygloDetectorMQTTClient("dayglo_detector")