From 149d80fb473851010f0397f52d385f0f81edbdbe Mon Sep 17 00:00:00 2001
From: aradhyacp <cparadhya@proton.me>
Date: Tue, 30 Dec 2025 17:35:44 +0530
Subject: [PATCH] Enhance Sentry integration in workers for improved error
 tracking and performance monitoring

---
 aperisolve/workers.py | 32 +++++++++++++++++++++++++++++++-
 1 file changed, 31 insertions(+), 1 deletion(-)

diff --git a/aperisolve/workers.py b/aperisolve/workers.py
index c5be8c7..d577fa2 100644
--- a/aperisolve/workers.py
+++ b/aperisolve/workers.py
@@ -1,9 +1,31 @@
 """Asynchronous worker for analyzing image submissions."""
 
+import os
 import threading
 from pathlib import Path
 from typing import Any
 
+import sentry_sdk
+from sentry_sdk.integrations.rq import RqIntegration
+from sentry_sdk.integrations.threading import ThreadingIntegration
+from sentry_sdk.integrations.sqlalchemy import SqlalchemyIntegration
+
+# Initialize Sentry for workers (separate process from Flask app)
+SENTRY_DSN = os.environ.get("SENTRY_DSN")
+if SENTRY_DSN:
+    sentry_sdk.init(
+        dsn=SENTRY_DSN,
+        integrations=[
+            RqIntegration(),
+            ThreadingIntegration(propagate_hub=True),  # Capture exceptions in threads
+            SqlalchemyIntegration(),
+        ],
+        traces_sample_rate=float(os.environ.get("SENTRY_TRACES_SAMPLE_RATE", "0.1")),
+        environment=os.environ.get("ENVIRONMENT", "development"),
+        release=os.environ.get("SENTRY_RELEASE", "1.0.0"),
+        send_default_pii=False,
+    )
+
 from .analyzers.binwalk import analyze_binwalk
 from .analyzers.decomposer import analyze_decomposer
 from .analyzers.exiftool import analyze_exiftool
@@ -51,6 +73,10 @@ def analyze_image(submission_hash: str) -> None:
         image = Image.query.get_or_404(submission.image_hash)  # type: ignore
 
         if not submission or not image:  # No submission found
+            sentry_sdk.capture_message(
+                f"Submission or image not found: {submission_hash}",
+                level="warning"
+            )
             return
 
         submission.status = "running"  # type: ignore
@@ -69,6 +95,7 @@ def analyze_image(submission_hash: str) -> None:
                     analyzer_func(*args)
                 except Exception as e:
                     print(f"Error in {analyzer_func.__name__}: {e}")
+                    sentry_sdk.capture_exception(e)
 
             analyzers = [
                 (analyze_binwalk, img_path, result_path),
@@ -100,7 +127,10 @@ def analyze_image(submission_hash: str) -> None:
                 thread.join()
 
             submission.status = "completed"
-        except Exception:
+        except Exception as e:
+            sentry_sdk.capture_exception(e)
             submission.status = "error"
         finally:
             db.session.commit()  # pylint: disable=no-member
+            # Flush Sentry events - wait up to 5 seconds for events to be sent
+            sentry_sdk.flush(timeout=5)