22

✏️ Editing: views.py

from sentry_sdk.hub import Hub
from sentry_sdk._types import MYPY
from sentry_sdk import _functools

if MYPY:
    from typing import Any

try:
    from asyncio import iscoroutinefunction
except ImportError:
    iscoroutinefunction = None  # type: ignore

try:
    from sentry_sdk.integrations.django.asgi import wrap_async_view
except (ImportError, SyntaxError):
    wrap_async_view = None  # type: ignore

def patch_views():
    # type: () -> None

from django.core.handlers.base import BaseHandler
    from sentry_sdk.integrations.django import DjangoIntegration

old_make_view_atomic = BaseHandler.make_view_atomic

@_functools.wraps(old_make_view_atomic)
    def sentry_patched_make_view_atomic(self, *args, **kwargs):
        # type: (Any, *Any, **Any) -> Any
        callback = old_make_view_atomic(self, *args, **kwargs)

# XXX: The wrapper function is created for every request. Find more
        # efficient way to wrap views (or build a cache?)

hub = Hub.current
        integration = hub.get_integration(DjangoIntegration)

if integration is not None and integration.middleware_spans:

if (
                iscoroutinefunction is not None
                and wrap_async_view is not None
                and iscoroutinefunction(callback)
            ):
                sentry_wrapped_callback = wrap_async_view(hub, callback)
            else:
                sentry_wrapped_callback = _wrap_sync_view(hub, callback)

else:
            sentry_wrapped_callback = callback

return sentry_wrapped_callback

BaseHandler.make_view_atomic = sentry_patched_make_view_atomic

def _wrap_sync_view(hub, callback):
    # type: (Hub, Any) -> Any
    @_functools.wraps(callback)
    def sentry_wrapped_callback(request, *args, **kwargs):
        # type: (Any, *Any, **Any) -> Any
        with hub.start_span(
            op="django.view", description=request.resolver_match.view_name
        ):
            return callback(request, *args, **kwargs)

return sentry_wrapped_callback

MR SKB