From abc1ecfb60f4b48c685a97ada6a9927ad906856e Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sun, 23 Jun 2024 07:45:10 -0600 Subject: [PATCH] Show correct previous state when updating for end (#12122) * Show correct previous state when updating for end * remove log * Formatting --- frigate/review/maintainer.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/frigate/review/maintainer.py b/frigate/review/maintainer.py index abf5029fb..55b374779 100644 --- a/frigate/review/maintainer.py +++ b/frigate/review/maintainer.py @@ -213,7 +213,11 @@ class ReviewSegmentMaintainer(threading.Thread): ), ) - def end_segment(self, segment: PendingReviewSegment) -> None: + def end_segment( + self, + segment: PendingReviewSegment, + prev_data: dict[str, any], + ) -> None: """End segment.""" final_data = segment.get_data(ended=True) self.requestor.send_data(UPSERT_REVIEW_SEGMENT, final_data) @@ -223,7 +227,7 @@ class ReviewSegmentMaintainer(threading.Thread): json.dumps( { "type": "end", - "before": end_data, + "before": prev_data, "after": end_data, } ), @@ -309,9 +313,9 @@ class ReviewSegmentMaintainer(threading.Thread): if segment.severity == SeverityEnum.alert and frame_time > ( segment.last_update + THRESHOLD_ALERT_ACTIVITY ): - self.end_segment(segment) + self.end_segment(segment, prev_data) elif frame_time > (segment.last_update + THRESHOLD_DETECTION_ACTIVITY): - self.end_segment(segment) + self.end_segment(segment, prev_data) def check_if_new_segment( self,