From 3457dcddfecaaf47a4e74f60e66cdbdc410ad7b3 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Fri, 10 Nov 2023 17:12:20 -0700
Subject: [PATCH] Don't run forever in autotracking (#8579)

---
 frigate/ptz/autotrack.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/frigate/ptz/autotrack.py b/frigate/ptz/autotrack.py
index 71a28504e..a7352a8f5 100644
--- a/frigate/ptz/autotrack.py
+++ b/frigate/ptz/autotrack.py
@@ -150,7 +150,9 @@ class PtzAutoTrackerThread(threading.Thread):
     ) -> None:
         threading.Thread.__init__(self)
         self.name = "ptz_autotracker"
-        self.ptz_autotracker = PtzAutoTracker(config, onvif, ptz_metrics, dispatcher)
+        self.ptz_autotracker = PtzAutoTracker(
+            config, onvif, ptz_metrics, dispatcher, stop_event
+        )
         self.stop_event = stop_event
         self.config = config
 
@@ -178,11 +180,13 @@ class PtzAutoTracker:
         onvif: OnvifController,
         ptz_metrics: PTZMetricsTypes,
         dispatcher: Dispatcher,
+        stop_event: MpEvent,
     ) -> None:
         self.config = config
         self.onvif = onvif
         self.ptz_metrics = ptz_metrics
         self.dispatcher = dispatcher
+        self.stop_event = stop_event
         self.tracked_object: dict[str, object] = {}
         self.tracked_object_history: dict[str, object] = {}
         self.tracked_object_metrics: dict[str, object] = {}
@@ -581,7 +585,7 @@ class PtzAutoTracker:
         camera_config.frame_shape[1]
         camera_config.frame_shape[0]
 
-        while True:
+        while not self.stop_event.is_set():
             move_data = self.move_queues[camera].get()
 
             with self.move_queue_locks[camera]: