ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

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

from contextlib import contextmanager

import sentry_sdk
from sentry_sdk.consts import OP
from sentry_sdk.integrations import _check_minimum_version, DidNotEnable, Integration
from sentry_sdk.scope import should_send_default_pii
from sentry_sdk.utils import (
    capture_internal_exceptions,
    ensure_integration_enabled,
    event_from_exception,
    package_version,
)

try:
    from graphene.types import schema as graphene_schema  # type: ignore
except ImportError:
    raise DidNotEnable("graphene is not installed")

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from collections.abc import Generator
    from typing import Any, Dict, Union
    from graphene.language.source import Source  # type: ignore
    from graphql.execution import ExecutionResult
    from graphql.type import GraphQLSchema
    from sentry_sdk._types import Event

class GrapheneIntegration(Integration):
    identifier = "graphene"

@staticmethod
    def setup_once():
        # type: () -> None
        version = package_version("graphene")
        _check_minimum_version(GrapheneIntegration, version)

_patch_graphql()

def _patch_graphql():
    # type: () -> None
    old_graphql_sync = graphene_schema.graphql_sync
    old_graphql_async = graphene_schema.graphql

@ensure_integration_enabled(GrapheneIntegration, old_graphql_sync)
    def _sentry_patched_graphql_sync(schema, source, *args, **kwargs):
        # type: (GraphQLSchema, Union[str, Source], Any, Any) -> ExecutionResult
        scope = sentry_sdk.get_isolation_scope()
        scope.add_event_processor(_event_processor)

with graphql_span(schema, source, kwargs):
            result = old_graphql_sync(schema, source, *args, **kwargs)

with capture_internal_exceptions():
            client = sentry_sdk.get_client()
            for error in result.errors or []:
                event, hint = event_from_exception(
                    error,
                    client_options=client.options,
                    mechanism={
                        "type": GrapheneIntegration.identifier,
                        "handled": False,
                    },
                )
                sentry_sdk.capture_event(event, hint=hint)

return result

async def _sentry_patched_graphql_async(schema, source, *args, **kwargs):
        # type: (GraphQLSchema, Union[str, Source], Any, Any) -> ExecutionResult
        integration = sentry_sdk.get_client().get_integration(GrapheneIntegration)
        if integration is None:
            return await old_graphql_async(schema, source, *args, **kwargs)

scope = sentry_sdk.get_isolation_scope()
        scope.add_event_processor(_event_processor)

with graphql_span(schema, source, kwargs):
            result = await old_graphql_async(schema, source, *args, **kwargs)

return result

graphene_schema.graphql_sync = _sentry_patched_graphql_sync
    graphene_schema.graphql = _sentry_patched_graphql_async

def _event_processor(event, hint):
    # type: (Event, Dict[str, Any]) -> Event
    if should_send_default_pii():
        request_info = event.setdefault("request", {})
        request_info["api_target"] = "graphql"

elif event.get("request", {}).get("data"):
        del event["request"]["data"]

return event

@contextmanager
def graphql_span(schema, source, kwargs):
    # type: (GraphQLSchema, Union[str, Source], Dict[str, Any]) -> Generator[None, None, None]
    operation_name = kwargs.get("operation_name")

operation_type = "query"
    op = OP.GRAPHQL_QUERY
    if source.strip().startswith("mutation"):
        operation_type = "mutation"
        op = OP.GRAPHQL_MUTATION
    elif source.strip().startswith("subscription"):
        operation_type = "subscription"
        op = OP.GRAPHQL_SUBSCRIPTION

sentry_sdk.add_breadcrumb(
        crumb={
            "data": {
                "operation_name": operation_name,
                "operation_type": operation_type,
            },
            "category": "graphql.operation",
        },
    )

scope = sentry_sdk.get_current_scope()
    if scope.span:
        _graphql_span = scope.span.start_child(op=op, name=operation_name)
    else:
        _graphql_span = sentry_sdk.start_span(op=op, name=operation_name)

_graphql_span.set_data("graphql.document", source)
    _graphql_span.set_data("graphql.operation.name", operation_name)
    _graphql_span.set_data("graphql.operation.type", operation_type)

try:
        yield
    finally:
        _graphql_span.finish()

[-] aws_lambda.py [edit]
[+] __pycache__
[-] chalice.py [edit]
[-] anthropic.py [edit]
[-] unleash.py [edit]
[-] atexit.py [edit]
[-] rq.py [edit]
[-] gnu_backtrace.py [edit]
[-] langgraph.py [edit]
[+] spark
[-] aiohttp.py [edit]
[-] socket.py [edit]
[-] sys_exit.py [edit]
[-] boto3.py [edit]
[-] sqlalchemy.py [edit]
[-] asyncpg.py [edit]
[-] openfeature.py [edit]
[-] modules.py [edit]
[-] executing.py [edit]
[-] wsgi.py [edit]
[-] dedupe.py [edit]
[-] __init__.py [edit]
[-] huey.py [edit]
[-] fastapi.py [edit]
[-] stdlib.py [edit]
[-] cloud_resource_context.py [edit]
[-] starlette.py [edit]
[-] flask.py [edit]
[-] bottle.py [edit]
[-] starlite.py [edit]
[-] rust_tracing.py [edit]
[-] otlp.py [edit]
[-] statsig.py [edit]
[-] excepthook.py [edit]
[-] ariadne.py [edit]
[+] pydantic_ai
[+] ..
[-] sanic.py [edit]
[+] google_genai
[+] django
[+] openai_agents
[-] dramatiq.py [edit]
[-] langchain.py [edit]
[-] graphene.py [edit]
[-] loguru.py [edit]
[-] clickhouse_driver.py [edit]
[+] celery
[-] tornado.py [edit]
[-] strawberry.py [edit]
[-] arq.py [edit]
[-] litellm.py [edit]
[-] openai.py [edit]
[-] beam.py [edit]
[-] launchdarkly.py [edit]
[+] grpc
[-] threading.py [edit]
[+] opentelemetry
[-] _wsgi_common.py [edit]
[-] _asgi_common.py [edit]
[-] argv.py [edit]
[-] litestar.py [edit]
[-] httpx.py [edit]
[-] pymongo.py [edit]
[-] mcp.py [edit]
[-] cohere.py [edit]
[-] pyramid.py [edit]
[-] unraisablehook.py [edit]
[-] quart.py [edit]
[-] falcon.py [edit]
[-] pure_eval.py [edit]
[+] redis
[-] logging.py [edit]
[-] asgi.py [edit]
[-] gql.py [edit]
[-] gcp.py [edit]
[-] ray.py [edit]
[-] trytond.py [edit]
[-] huggingface_hub.py [edit]
[-] asyncio.py [edit]
[-] typer.py [edit]
[-] serverless.py [edit]