sentry_span.py

def sentry_span(func):
    if asyncio.iscoroutinefunction(func):
        @async_wraps(func)
        async def wrapped_async_sentry_span(*args, **kwargs):
            __tracebackhide__ = True
            with sentry_sdk.Hub(sentry_sdk.Hub.current):
                with sentry_sdk.start_span(op=func.__qualname__):
                    return await func(*args, **kwargs)

        return wrapped_async_sentry_span

    @functools.wraps(func)
    def wrapped_sync_sentry_span(*args, **kwargs):
        __tracebackhide__ = True
        with sentry_sdk.start_span(op=func.__qualname__):
            return func(*args, **kwargs)

    return wrapped_sync_sentry_span