feat: add sentry sdk

worker/Makefile

@@ -4,7 +4,7 @@ include ../tools/python.mk
 
 PIP_INSTALL := \
 	--editable ../shared \
-	--editable .[dev]
+	--editable .[dev,sentry]
 PYLINT_ARG := libretime_worker
 MYPY_ARG := libretime_worker || true
 BANDIT_ARG := libretime_worker

worker/libretime_worker/tasks.py

@@ -1,4 +1,5 @@
 import json
+import os
 from email.message import EmailMessage
 from pathlib import Path
 from tempfile import NamedTemporaryFile
@@ -7,17 +8,35 @@ from urllib.parse import urlsplit
 
 import mutagen
 import requests
-from celery import Celery
+from celery import Celery, signals
 from celery.utils.log import get_task_logger
 from mutagen import MutagenError
 from requests import RequestException, Response
 
+from . import PACKAGE, VERSION
 from .config import config
 
 worker = Celery()
 logger = get_task_logger(__name__)
 
 
+@signals.worker_init.connect
+def init_sentry(**_kwargs):
+    if "SENTRY_DSN" in os.environ:
+        logger.info("installing sentry")
+        # pylint: disable=import-outside-toplevel
+        import sentry_sdk
+        from sentry_sdk.integrations.celery import CeleryIntegration
+
+        sentry_sdk.init(
+            traces_sample_rate=1.0,
+            release=f"{PACKAGE}@{VERSION}",
+            integrations=[
+                CeleryIntegration(),
+            ],
+        )
+
+
 @worker.task(name="podcast-download", acks_late=True)
 def podcast_download(
     episode_id: int,

worker/setup.py

@@ -25,6 +25,9 @@ setup(
             "requests-mock>=1.10.0,<1.11",
             "types-requests>=2.25.1,<2.29",
         ],
+        "sentry": [
+            "sentry-sdk>=1.15.0,<1.16",
+        ],
     },
     zip_safe=False,
 )