Refactor automatic flagging

iop4lib/db/photopolresult.py

@@ -296,6 +296,18 @@ def save(self, *args, **kwargs):
         """ Overriden to enforce clean() before saving. See PhotoPolResult.__docstring__ for more info."""
         self.clean()
         super().save(*args, **kwargs)
+
+    # Auto-flagging
+
+    def auto_flag(self):
+
+        if self.p is not None and not (0 <= self.p <= 1):
+            self.set_flag(PhotoPolResult.FLAGS.BAD_POLARIMETRY)
+
+        if self.mag is not None and self.mag_err is not None and self.mag_err > 0.3:
+            self.set_flag(PhotoPolResult.FLAGS.BAD_PHOTOMETRY)
+
+        self.save()        
 
     # Host galaxy correction
 

iop4lib/iop4.py

@@ -82,9 +82,7 @@ def process_epochs(epochname_list: Iterable[str], args):
     logger.info("Auto-flagging points.")
 
     for result in PhotoPolResult.objects.filter(epoch__in=epoch_L).all():
-        if result.p is not None and not (0 <= result.p <= 1):
-            result.set_flag(PhotoPolResult.FLAGS.BAD_POLARIMETRY)
-            result.save()
+        result.auto_flag()
 
     logger.info("Applying corrections.")
 
@@ -162,9 +160,7 @@ def process_astrosource(args):
         logger.info("Auto-flagging points.")
 
         for result in PhotoPolResult.objects.filter(epoch__in=epoch_L).all():
-            if result.p is not None and not (0 <= result.p <= 1):
-                result.set_flag(PhotoPolResult.FLAGS.BAD_POLARIMETRY)
-                result.save()
+            result.auto_flag()
 
         logger.info("Applying corrections.")