From dbdda263999f2e7be46e64e2a861ac67df2ea9cb Mon Sep 17 00:00:00 2001
From: SGCMarkus <markusornik@gmail.com>
Date: Mon, 13 May 2024 14:40:20 +0200
Subject: [PATCH] flaredetector: add quality flag to return lists

---
 main/flaredetector/flaredetector.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py
index 219ee71..bbfe853 100644
--- a/main/flaredetector/flaredetector.py
+++ b/main/flaredetector/flaredetector.py
@@ -50,10 +50,12 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05
             Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index)
             peaks.append({"FlarePeak": Peak,
                           "FlarePeakTime": lightcurve.time[index],
-                          "StandardIndex": index})
+                          "StandardIndex": index,
+                          "Quality": lightcurve.quality[index]})
             fits.append({"FlareFit": fit,
                          "FlareFitTime": lightcurve.time[flareFitDataPoints],
-                         "StandardIndex": flareFitDataPoints})
+                         "StandardIndex": flareFitDataPoints,
+                          "Quality": lightcurve.quality[flareFitDataPoints]})
         except Exception as error:
             pass
             #print("Flare fit exception: ", error)