Gecko [ renovalkoronred.com.tr ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
django	[ DIR ]	drwxr-xr-x
grpc	[ DIR ]	drwxr-xr-x
opentelemetry	[ DIR ]	drwxr-xr-x
redis	[ DIR ]	drwxr-xr-x
spark	[ DIR ]	drwxr-xr-x
__init__.py	6.67 KB	-rw-r--r--
_wsgi_common.py	4.43 KB	-rw-r--r--
aiohttp.py	11.28 KB	-rw-r--r--
argv.py	963 B	-rw-r--r--
arq.py	6.58 KB	-rw-r--r--
asgi.py	11.54 KB	-rw-r--r--
asyncio.py	2.98 KB	-rw-r--r--
atexit.py	1.8 KB	-rw-r--r--
aws_lambda.py	15.44 KB	-rw-r--r--
beam.py	5.56 KB	-rw-r--r--
boto3.py	4.44 KB	-rw-r--r--
bottle.py	6.32 KB	-rw-r--r--
celery.py	18.65 KB	-rw-r--r--
chalice.py	4.66 KB	-rw-r--r--
cloud_resource_context.py	6.6 KB	-rw-r--r--
dedupe.py	1.16 KB	-rw-r--r--
excepthook.py	2.21 KB	-rw-r--r--
executing.py	1.99 KB	-rw-r--r--
falcon.py	7.8 KB	-rw-r--r--
fastapi.py	4.39 KB	-rw-r--r--
flask.py	7.72 KB	-rw-r--r--
gcp.py	8.02 KB	-rw-r--r--
gnu_backtrace.py	2.86 KB	-rw-r--r--
httpx.py	4.89 KB	-rw-r--r--
huey.py	4.59 KB	-rw-r--r--
logging.py	8.97 KB	-rw-r--r--
loguru.py	2.98 KB	-rw-r--r--
modules.py	2.06 KB	-rw-r--r--
pure_eval.py	4.45 KB	-rw-r--r--
pymongo.py	5.87 KB	-rw-r--r--
pyramid.py	7.27 KB	-rw-r--r--
quart.py	7.2 KB	-rw-r--r--
rq.py	5.28 KB	-rw-r--r--
sanic.py	11.06 KB	-rw-r--r--
serverless.py	1.93 KB	-rw-r--r--
socket.py	2.88 KB	-rw-r--r--
sqlalchemy.py	4.14 KB	-rw-r--r--
starlette.py	22.67 KB	-rw-r--r--
starlite.py	9.85 KB	-rw-r--r--
stdlib.py	8.06 KB	-rw-r--r--
threading.py	2.87 KB	-rw-r--r--
tornado.py	7.17 KB	-rw-r--r--
trytond.py	1.7 KB	-rw-r--r--
wsgi.py	9.36 KB	-rw-r--r--

Code Editor : beam.py

from __future__ import absolute_import

import sys
import types
from sentry_sdk._functools import wraps

from sentry_sdk.hub import Hub
from sentry_sdk._compat import reraise
from sentry_sdk.utils import capture_internal_exceptions, event_from_exception
from sentry_sdk.integrations import Integration
from sentry_sdk.integrations.logging import ignore_logger
from sentry_sdk._types import TYPE_CHECKING

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

from sentry_sdk.client import Client
    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.
    Pass the client on to raise_exception so it can get rebinded.
    """
    client = Hub.current.client

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

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

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

def _capture_exception(exc_info, hub):
    # type: (ExcInfo, Hub) -> None
    """
    Send Beam exception to Sentry.
    """
    integration = hub.get_integration(BeamIntegration)
    if integration is None:
        return

client = hub.client
    if client is None:
        return

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

def raise_exception(client):
    # type: (Optional[Client]) -> None
    """
    Raise an exception. If the client is not in the hub, rebind it.
    """
    hub = Hub.current
    if hub.client is None:
        hub.bind_client(client)
    exc_info = sys.exc_info()
    with capture_internal_exceptions():
        _capture_exception(exc_info, hub)
    reraise(*exc_info)

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

${this.title}

Code Editor : beam.py