fix: flatten React Router structure — no nested Routes, each page is a top-level route
This commit is contained in:
@@ -0,0 +1,9 @@
|
||||
from sentry_sdk.crons.api import capture_checkin
|
||||
from sentry_sdk.crons.consts import MonitorStatus
|
||||
from sentry_sdk.crons.decorator import monitor
|
||||
|
||||
__all__ = [
|
||||
"capture_checkin",
|
||||
"MonitorStatus",
|
||||
"monitor",
|
||||
]
|
||||
@@ -0,0 +1,60 @@
|
||||
import uuid
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
import sentry_sdk
|
||||
from sentry_sdk.utils import logger
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from typing import Optional
|
||||
|
||||
from sentry_sdk._types import Event, MonitorConfig
|
||||
|
||||
|
||||
def _create_check_in_event(
|
||||
monitor_slug: "Optional[str]" = None,
|
||||
check_in_id: "Optional[str]" = None,
|
||||
status: "Optional[str]" = None,
|
||||
duration_s: "Optional[float]" = None,
|
||||
monitor_config: "Optional[MonitorConfig]" = None,
|
||||
) -> "Event":
|
||||
options = sentry_sdk.get_client().options
|
||||
check_in_id = check_in_id or uuid.uuid4().hex
|
||||
|
||||
check_in: "Event" = {
|
||||
"type": "check_in",
|
||||
"monitor_slug": monitor_slug,
|
||||
"check_in_id": check_in_id,
|
||||
"status": status,
|
||||
"duration": duration_s,
|
||||
"environment": options.get("environment", None),
|
||||
"release": options.get("release", None),
|
||||
}
|
||||
|
||||
if monitor_config:
|
||||
check_in["monitor_config"] = monitor_config
|
||||
|
||||
return check_in
|
||||
|
||||
|
||||
def capture_checkin(
|
||||
monitor_slug: "Optional[str]" = None,
|
||||
check_in_id: "Optional[str]" = None,
|
||||
status: "Optional[str]" = None,
|
||||
duration: "Optional[float]" = None,
|
||||
monitor_config: "Optional[MonitorConfig]" = None,
|
||||
) -> str:
|
||||
check_in_event = _create_check_in_event(
|
||||
monitor_slug=monitor_slug,
|
||||
check_in_id=check_in_id,
|
||||
status=status,
|
||||
duration_s=duration,
|
||||
monitor_config=monitor_config,
|
||||
)
|
||||
|
||||
sentry_sdk.capture_event(check_in_event)
|
||||
|
||||
logger.debug(
|
||||
f"[Crons] Captured check-in ({check_in_event.get('check_in_id')}): {check_in_event.get('monitor_slug')} -> {check_in_event.get('status')}"
|
||||
)
|
||||
|
||||
return check_in_event["check_in_id"]
|
||||
@@ -0,0 +1,4 @@
|
||||
class MonitorStatus:
|
||||
IN_PROGRESS = "in_progress"
|
||||
OK = "ok"
|
||||
ERROR = "error"
|
||||
@@ -0,0 +1,137 @@
|
||||
from functools import wraps
|
||||
from inspect import iscoroutinefunction
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from sentry_sdk.crons import capture_checkin
|
||||
from sentry_sdk.crons.consts import MonitorStatus
|
||||
from sentry_sdk.utils import now
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from collections.abc import Awaitable, Callable
|
||||
from types import TracebackType
|
||||
from typing import (
|
||||
Any,
|
||||
Optional,
|
||||
ParamSpec,
|
||||
Type,
|
||||
TypeVar,
|
||||
Union,
|
||||
cast,
|
||||
overload,
|
||||
)
|
||||
|
||||
from sentry_sdk._types import MonitorConfig
|
||||
|
||||
P = ParamSpec("P")
|
||||
R = TypeVar("R")
|
||||
|
||||
|
||||
class monitor: # noqa: N801
|
||||
"""
|
||||
Decorator/context manager to capture checkin events for a monitor.
|
||||
|
||||
Usage (as decorator):
|
||||
```
|
||||
import sentry_sdk
|
||||
|
||||
app = Celery()
|
||||
|
||||
@app.task
|
||||
@sentry_sdk.monitor(monitor_slug='my-fancy-slug')
|
||||
def test(arg):
|
||||
print(arg)
|
||||
```
|
||||
|
||||
This does not have to be used with Celery, but if you do use it with celery,
|
||||
put the `@sentry_sdk.monitor` decorator below Celery's `@app.task` decorator.
|
||||
|
||||
Usage (as context manager):
|
||||
```
|
||||
import sentry_sdk
|
||||
|
||||
def test(arg):
|
||||
with sentry_sdk.monitor(monitor_slug='my-fancy-slug'):
|
||||
print(arg)
|
||||
```
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
monitor_slug: "Optional[str]" = None,
|
||||
monitor_config: "Optional[MonitorConfig]" = None,
|
||||
) -> None:
|
||||
self.monitor_slug = monitor_slug
|
||||
self.monitor_config = monitor_config
|
||||
|
||||
def __enter__(self) -> None:
|
||||
self.start_timestamp = now()
|
||||
self.check_in_id = capture_checkin(
|
||||
monitor_slug=self.monitor_slug,
|
||||
status=MonitorStatus.IN_PROGRESS,
|
||||
monitor_config=self.monitor_config,
|
||||
)
|
||||
|
||||
def __exit__(
|
||||
self,
|
||||
exc_type: "Optional[Type[BaseException]]",
|
||||
exc_value: "Optional[BaseException]",
|
||||
traceback: "Optional[TracebackType]",
|
||||
) -> None:
|
||||
duration_s = now() - self.start_timestamp
|
||||
|
||||
if exc_type is None and exc_value is None and traceback is None:
|
||||
status = MonitorStatus.OK
|
||||
else:
|
||||
status = MonitorStatus.ERROR
|
||||
|
||||
capture_checkin(
|
||||
monitor_slug=self.monitor_slug,
|
||||
check_in_id=self.check_in_id,
|
||||
status=status,
|
||||
duration=duration_s,
|
||||
monitor_config=self.monitor_config,
|
||||
)
|
||||
|
||||
if TYPE_CHECKING:
|
||||
|
||||
@overload
|
||||
def __call__(
|
||||
self, fn: "Callable[P, Awaitable[Any]]"
|
||||
) -> "Callable[P, Awaitable[Any]]":
|
||||
# Unfortunately, mypy does not give us any reliable way to type check the
|
||||
# return value of an Awaitable (i.e. async function) for this overload,
|
||||
# since calling iscouroutinefunction narrows the type to Callable[P, Awaitable[Any]].
|
||||
...
|
||||
|
||||
@overload
|
||||
def __call__(self, fn: "Callable[P, R]") -> "Callable[P, R]": ...
|
||||
|
||||
def __call__(
|
||||
self,
|
||||
fn: "Union[Callable[P, R], Callable[P, Awaitable[Any]]]",
|
||||
) -> "Union[Callable[P, R], Callable[P, Awaitable[Any]]]":
|
||||
if iscoroutinefunction(fn):
|
||||
return self._async_wrapper(fn)
|
||||
|
||||
else:
|
||||
if TYPE_CHECKING:
|
||||
fn = cast("Callable[P, R]", fn)
|
||||
return self._sync_wrapper(fn)
|
||||
|
||||
def _async_wrapper(
|
||||
self, fn: "Callable[P, Awaitable[Any]]"
|
||||
) -> "Callable[P, Awaitable[Any]]":
|
||||
@wraps(fn)
|
||||
async def inner(*args: "P.args", **kwargs: "P.kwargs") -> "R":
|
||||
with self:
|
||||
return await fn(*args, **kwargs)
|
||||
|
||||
return inner
|
||||
|
||||
def _sync_wrapper(self, fn: "Callable[P, R]") -> "Callable[P, R]":
|
||||
@wraps(fn)
|
||||
def inner(*args: "P.args", **kwargs: "P.kwargs") -> "R":
|
||||
with self:
|
||||
return fn(*args, **kwargs)
|
||||
|
||||
return inner
|
||||
Reference in New Issue
Block a user