File Manager

X7ROOT File Manager

Current Path: /opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations

opt / hc_python / lib / python3.12 / site-packages / sentry_sdk / integrations /

?? ..
?? __init__.py(12.44 KB)
?? __pycache__
?? _asgi_common.py(3.11 KB)
?? _wsgi_common.py(7.38 KB)
?? aiohttp.py(12.7 KB)
?? anthropic.py(14.17 KB)
?? argv.py(911 B)
?? ariadne.py(5.7 KB)
?? arq.py(7.7 KB)
?? asgi.py(12.5 KB)
?? asyncio.py(4.5 KB)
?? asyncpg.py(6.37 KB)
?? atexit.py(1.61 KB)
?? aws_lambda.py(17.65 KB)
?? beam.py(5.06 KB)
?? boto3.py(4.31 KB)
?? bottle.py(6.46 KB)
?? celery
?? chalice.py(4.59 KB)
?? clickhouse_driver.py(5.96 KB)
?? cloud_resource_context.py(7.6 KB)
?? cohere.py(9.18 KB)
?? dedupe.py(1.93 KB)
?? django
?? dramatiq.py(7.28 KB)
?? excepthook.py(2.35 KB)
?? executing.py(1.95 KB)
?? falcon.py(9.28 KB)
?? fastapi.py(4.48 KB)
?? flask.py(8.54 KB)
?? gcp.py(8.26 KB)
?? gnu_backtrace.py(2.75 KB)
?? google_genai
?? gql.py(4.69 KB)
?? graphene.py(4.92 KB)
?? grpc
?? httpx.py(5.94 KB)
?? huey.py(5.32 KB)
?? huggingface_hub.py(14.6 KB)
?? langchain.py(39.02 KB)
?? langgraph.py(11.56 KB)
?? launchdarkly.py(1.89 KB)
?? litellm.py(9.99 KB)
?? litestar.py(11.55 KB)
?? logging.py(13.57 KB)
?? loguru.py(6.38 KB)
?? mcp.py(19.58 KB)
?? modules.py(820 B)
?? openai.py(24.54 KB)
?? openai_agents
?? openfeature.py(1.1 KB)
?? opentelemetry
?? otlp.py(3.06 KB)
?? pure_eval.py(4.5 KB)
?? pydantic_ai
?? pymongo.py(6.23 KB)
?? pyramid.py(7.19 KB)
?? quart.py(7.24 KB)
?? ray.py(5.2 KB)
?? redis
?? rq.py(5.18 KB)
?? rust_tracing.py(8.87 KB)
?? sanic.py(12.66 KB)
?? serverless.py(1.76 KB)
?? socket.py(3.09 KB)
?? spark
?? sqlalchemy.py(4.24 KB)
?? starlette.py(25.62 KB)
?? starlite.py(10.31 KB)
?? statsig.py(1.2 KB)
?? stdlib.py(8.76 KB)
?? strawberry.py(13.82 KB)
?? sys_exit.py(2.43 KB)
?? threading.py(6.94 KB)
?? tornado.py(7.04 KB)
?? trytond.py(1.61 KB)
?? typer.py(1.77 KB)
?? unleash.py(1.03 KB)
?? unraisablehook.py(1.71 KB)
?? wsgi.py(10.56 KB)

Editing: beam.py

import sys
import types
from functools import wraps

import sentry_sdk
from sentry_sdk.integrations import Integration
from sentry_sdk.integrations.logging import ignore_logger
from sentry_sdk.utils import (
    capture_internal_exceptions,
    ensure_integration_enabled,
    event_from_exception,
    reraise,
)

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from typing import Any
    from typing import Iterator
    from typing import TypeVar
    from typing import Callable

from sentry_sdk._types import ExcInfo

T = TypeVar("T")
    F = TypeVar("F", bound=Callable[..., Any])

WRAPPED_FUNC = "_wrapped_{}_"
INSPECT_FUNC = "_inspect_{}"  # Required format per apache_beam/transforms/core.py
USED_FUNC = "_sentry_used_"

class BeamIntegration(Integration):
    identifier = "beam"

@staticmethod
    def setup_once():
        # type: () -> None
        from apache_beam.transforms.core import DoFn, ParDo  # type: ignore

ignore_logger("root")
        ignore_logger("bundle_processor.create")

function_patches = ["process", "start_bundle", "finish_bundle", "setup"]
        for func_name in function_patches:
            setattr(
                DoFn,
                INSPECT_FUNC.format(func_name),
                _wrap_inspect_call(DoFn, func_name),
            )

old_init = ParDo.__init__

def sentry_init_pardo(self, fn, *args, **kwargs):
            # type: (ParDo, Any, *Any, **Any) -> Any
            # Do not monkey patch init twice
            if not getattr(self, "_sentry_is_patched", False):
                for func_name in function_patches:
                    if not hasattr(fn, func_name):
                        continue
                    wrapped_func = WRAPPED_FUNC.format(func_name)

# Check to see if inspect is set and process is not
                    # to avoid monkey patching process twice.
                    # Check to see if function is part of object for
                    # backwards compatibility.
                    process_func = getattr(fn, func_name)
                    inspect_func = getattr(fn, INSPECT_FUNC.format(func_name))
                    if not getattr(inspect_func, USED_FUNC, False) and not getattr(
                        process_func, USED_FUNC, False
                    ):
                        setattr(fn, wrapped_func, process_func)
                        setattr(fn, func_name, _wrap_task_call(process_func))

self._sentry_is_patched = True
            old_init(self, fn, *args, **kwargs)

ParDo.__init__ = sentry_init_pardo

def _wrap_inspect_call(cls, func_name):
    # type: (Any, Any) -> Any

if not hasattr(cls, func_name):
        return None

def _inspect(self):
        # type: (Any) -> Any
        """
        Inspect function overrides the way Beam gets argspec.
        """
        wrapped_func = WRAPPED_FUNC.format(func_name)
        if hasattr(self, wrapped_func):
            process_func = getattr(self, wrapped_func)
        else:
            process_func = getattr(self, func_name)
            setattr(self, func_name, _wrap_task_call(process_func))
            setattr(self, wrapped_func, process_func)

# getfullargspec is deprecated in more recent beam versions and get_function_args_defaults
        # (which uses Signatures internally) should be used instead.
        try:
            from apache_beam.transforms.core import get_function_args_defaults

return get_function_args_defaults(process_func)
        except ImportError:
            from apache_beam.typehints.decorators import getfullargspec  # type: ignore

return getfullargspec(process_func)

setattr(_inspect, USED_FUNC, True)
    return _inspect

def _wrap_task_call(func):
    # type: (F) -> F
    """
    Wrap task call with a try catch to get exceptions.
    """

@wraps(func)
    def _inner(*args, **kwargs):
        # type: (*Any, **Any) -> Any
        try:
            gen = func(*args, **kwargs)
        except Exception:
            raise_exception()

if not isinstance(gen, types.GeneratorType):
            return gen
        return _wrap_generator_call(gen)

setattr(_inner, USED_FUNC, True)
    return _inner  # type: ignore

@ensure_integration_enabled(BeamIntegration)
def _capture_exception(exc_info):
    # type: (ExcInfo) -> None
    """
    Send Beam exception to Sentry.
    """
    client = sentry_sdk.get_client()

event, hint = event_from_exception(
        exc_info,
        client_options=client.options,
        mechanism={"type": "beam", "handled": False},
    )
    sentry_sdk.capture_event(event, hint=hint)

def raise_exception():
    # type: () -> None
    """
    Raise an exception.
    """
    exc_info = sys.exc_info()
    with capture_internal_exceptions():
        _capture_exception(exc_info)
    reraise(*exc_info)

def _wrap_generator_call(gen):
    # type: (Iterator[T]) -> Iterator[T]
    """
    Wrap the generator to handle any failures.
    """
    while True:
        try:
            yield next(gen)
        except StopIteration:
            break
        except Exception:
            raise_exception()

X7ROOT File Manager

Editing: beam.py

Upload File

Create Folder