45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
from functools import wraps
|
|
|
|
import sentry_sdk
|
|
from sentry_sdk import get_current_span
|
|
from sentry_sdk._types import TYPE_CHECKING
|
|
from sentry_sdk.consts import OP
|
|
from sentry_sdk.utils import logger, qualname_from_function
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Any
|
|
|
|
|
|
def start_child_span_decorator(func):
|
|
# type: (Any) -> Any
|
|
"""
|
|
Decorator to add child spans for functions.
|
|
|
|
This is the Python 2 compatible version of the decorator.
|
|
Duplicated code from ``sentry_sdk.tracing_utils_python3.start_child_span_decorator``.
|
|
|
|
See also ``sentry_sdk.tracing.trace()``.
|
|
"""
|
|
|
|
@wraps(func)
|
|
def func_with_tracing(*args, **kwargs):
|
|
# type: (*Any, **Any) -> Any
|
|
|
|
span = get_current_span(sentry_sdk.Hub.current)
|
|
|
|
if span is None:
|
|
logger.warning(
|
|
"Can not create a child span for %s. "
|
|
"Please start a Sentry transaction before calling this function.",
|
|
qualname_from_function(func),
|
|
)
|
|
return func(*args, **kwargs)
|
|
|
|
with span.start_child(
|
|
op=OP.FUNCTION,
|
|
description=qualname_from_function(func),
|
|
):
|
|
return func(*args, **kwargs)
|
|
|
|
return func_with_tracing
|