?HAMA

Edit File Size: 4.32 KB

/proc/self/root/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/integrations/celery.py

Text Base64

from __future__ import absolute_import

import sys

from celery.exceptions import SoftTimeLimitExceeded, Retry  # type: ignore

from sentry_sdk.hub import Hub
from sentry_sdk.utils import capture_internal_exceptions, event_from_exception
from sentry_sdk._compat import reraise
from sentry_sdk.integrations import Integration
from sentry_sdk.integrations.logging import ignore_logger

class CeleryIntegration(Integration):
    identifier = "celery"

@staticmethod
    def setup_once():
        import celery.app.trace as trace  # type: ignore

old_build_tracer = trace.build_tracer

def sentry_build_tracer(name, task, *args, **kwargs):
            # Need to patch both methods because older celery sometimes
            # short-circuits to task.run if it thinks it's safe.
            task.__call__ = _wrap_task_call(task, task.__call__)
            task.run = _wrap_task_call(task, task.run)
            return _wrap_tracer(task, old_build_tracer(name, task, *args, **kwargs))

trace.build_tracer = sentry_build_tracer

_patch_worker_exit()

# This logger logs every status of every task that ran on the worker.
        # Meaning that every task's breadcrumbs are full of stuff like "Task
        # <foo> raised unexpected <bar>".
        ignore_logger("celery.worker.job")

def _wrap_tracer(task, f):
    # Need to wrap tracer for pushing the scope before prerun is sent, and
    # popping it after postrun is sent.
    #
    # This is the reason we don't use signals for hooking in the first place.
    # Also because in Celery 3, signal dispatch returns early if one handler
    # crashes.
    def _inner(*args, **kwargs):
        hub = Hub.current
        if hub.get_integration(CeleryIntegration) is None:
            return f(*args, **kwargs)

with hub.push_scope() as scope:
            scope._name = "celery"
            scope.clear_breadcrumbs()
            scope.add_event_processor(_make_event_processor(task, *args, **kwargs))

return f(*args, **kwargs)

return _inner

def _wrap_task_call(task, f):
    # Need to wrap task call because the exception is caught before we get to
    # see it. Also celery's reported stacktrace is untrustworthy.
    def _inner(*args, **kwargs):
        try:
            return f(*args, **kwargs)
        except Exception:
            exc_info = sys.exc_info()
            with capture_internal_exceptions():
                _capture_exception(task, exc_info)
            reraise(*exc_info)

return _inner

def _make_event_processor(task, uuid, args, kwargs, request=None):
    def event_processor(event, hint):
        with capture_internal_exceptions():
            event["transaction"] = task.name

with capture_internal_exceptions():
            extra = event.setdefault("extra", {})
            extra["celery-job"] = {
                "task_name": task.name,
                "args": args,
                "kwargs": kwargs,
            }

if "exc_info" in hint:
            with capture_internal_exceptions():
                if issubclass(hint["exc_info"][0], SoftTimeLimitExceeded):
                    event["fingerprint"] = [
                        "celery",
                        "SoftTimeLimitExceeded",
                        getattr(task, "name", task),
                    ]

return event

return event_processor

def _capture_exception(task, exc_info):
    hub = Hub.current

if hub.get_integration(CeleryIntegration) is None:
        return
    if isinstance(exc_info[1], Retry):
        return
    if hasattr(task, "throws") and isinstance(exc_info[1], task.throws):
        return

event, hint = event_from_exception(
        exc_info,
        client_options=hub.client.options,
        mechanism={"type": "celery", "handled": False},
    )

hub.capture_event(event, hint=hint)

def _patch_worker_exit():
    # Need to flush queue before worker shutdown because a crashing worker will
    # call os._exit
    from billiard.pool import Worker  # type: ignore

old_workloop = Worker.workloop

def sentry_workloop(*args, **kwargs):
        try:
            return old_workloop(*args, **kwargs)
        finally:
            with capture_internal_exceptions():
                hub = Hub.current
                if hub.get_integration(CeleryIntegration) is not None:
                    hub.flush()

Worker.workloop = sentry_workloop

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
django DIR	-	drwxr-xr-x	2024-03-21 10:24:45	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2024-03-21 10:24:45	Edit Download recursive recursive
aiohttp.py	3.80 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
argv.py	894 B	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
atexit.py	1.46 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
aws_lambda.py	6.60 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
bottle.py	5.40 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
celery.py	4.32 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
dedupe.py	1.06 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
excepthook.py	1.30 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
flask.py	7.10 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
gnu_backtrace.py	2.94 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
logging.py	5.88 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
modules.py	1.21 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
pyramid.py	5.78 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
rq.py	3.20 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
sanic.py	6.90 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
serverless.py	1.21 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
stdlib.py	1.83 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
threading.py	1.56 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
tornado.py	6.24 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
wsgi.py	7.13 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
_wsgi_common.py	3.58 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive
__init__.py	4.14 KB	lrw-r--r--	2019-04-05 23:15:16	Edit Download recursive recursive

Directory Contents