76 lines
1.8 KiB
Python
76 lines
1.8 KiB
Python
import sys
|
|
from functools import wraps
|
|
|
|
import sentry_sdk
|
|
from sentry_sdk.utils import event_from_exception, reraise
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Any
|
|
from typing import Callable
|
|
from typing import TypeVar
|
|
from typing import Union
|
|
from typing import Optional
|
|
from typing import overload
|
|
|
|
F = TypeVar("F", bound=Callable[..., Any])
|
|
|
|
else:
|
|
|
|
def overload(x):
|
|
# type: (F) -> F
|
|
return x
|
|
|
|
|
|
@overload
|
|
def serverless_function(f, flush=True):
|
|
# type: (F, bool) -> F
|
|
pass
|
|
|
|
|
|
@overload
|
|
def serverless_function(f=None, flush=True): # noqa: F811
|
|
# type: (None, bool) -> Callable[[F], F]
|
|
pass
|
|
|
|
|
|
def serverless_function(f=None, flush=True): # noqa
|
|
# type: (Optional[F], bool) -> Union[F, Callable[[F], F]]
|
|
def wrapper(f):
|
|
# type: (F) -> F
|
|
@wraps(f)
|
|
def inner(*args, **kwargs):
|
|
# type: (*Any, **Any) -> Any
|
|
with sentry_sdk.isolation_scope() as scope:
|
|
scope.clear_breadcrumbs()
|
|
|
|
try:
|
|
return f(*args, **kwargs)
|
|
except Exception:
|
|
_capture_and_reraise()
|
|
finally:
|
|
if flush:
|
|
sentry_sdk.flush()
|
|
|
|
return inner # type: ignore
|
|
|
|
if f is None:
|
|
return wrapper
|
|
else:
|
|
return wrapper(f)
|
|
|
|
|
|
def _capture_and_reraise():
|
|
# type: () -> None
|
|
exc_info = sys.exc_info()
|
|
client = sentry_sdk.get_client()
|
|
if client.is_active():
|
|
event, hint = event_from_exception(
|
|
exc_info,
|
|
client_options=client.options,
|
|
mechanism={"type": "serverless", "handled": False},
|
|
)
|
|
sentry_sdk.capture_event(event, hint=hint)
|
|
|
|
reraise(*exc_info)
|