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

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

from importlib import import_module

import sentry_sdk
from sentry_sdk import capture_event, get_client
from sentry_sdk.integrations import DidNotEnable, Integration, _check_minimum_version
from sentry_sdk.integrations._wsgi_common import request_body_within_bounds
from sentry_sdk.integrations.logging import ignore_logger
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:
    # importing like this is necessary due to name shadowing in ariadne
    # (ariadne.graphql is also a function)
    ariadne_graphql = import_module("ariadne.graphql")
except ImportError:
    raise DidNotEnable("ariadne is not installed")

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from typing import Any, Dict, List, Optional

from ariadne.types import (  # type: ignore
        GraphQLError,
        GraphQLResult,
        GraphQLSchema,
        QueryParser,
    )
    from graphql.language.ast import DocumentNode

from sentry_sdk._types import Event, EventProcessor

class AriadneIntegration(Integration):
    identifier = "ariadne"

@staticmethod
    def setup_once() -> None:
        version = package_version("ariadne")
        _check_minimum_version(AriadneIntegration, version)

ignore_logger("ariadne")

_patch_graphql()

def _patch_graphql() -> None:
    old_parse_query = ariadne_graphql.parse_query
    old_handle_errors = ariadne_graphql.handle_graphql_errors
    old_handle_query_result = ariadne_graphql.handle_query_result

@ensure_integration_enabled(AriadneIntegration, old_parse_query)
    def _sentry_patched_parse_query(
        context_value: "Optional[Any]",
        query_parser: "Optional[QueryParser]",
        data: "Any",
    ) -> "DocumentNode":
        event_processor = _make_request_event_processor(data)
        sentry_sdk.get_isolation_scope().add_event_processor(event_processor)

result = old_parse_query(context_value, query_parser, data)
        return result

@ensure_integration_enabled(AriadneIntegration, old_handle_errors)
    def _sentry_patched_handle_graphql_errors(
        errors: "List[GraphQLError]", *args: "Any", **kwargs: "Any"
    ) -> "GraphQLResult":
        result = old_handle_errors(errors, *args, **kwargs)

event_processor = _make_response_event_processor(result[1])
        sentry_sdk.get_isolation_scope().add_event_processor(event_processor)

client = get_client()
        if client.is_active():
            with capture_internal_exceptions():
                for error in errors:
                    event, hint = event_from_exception(
                        error,
                        client_options=client.options,
                        mechanism={
                            "type": AriadneIntegration.identifier,
                            "handled": False,
                        },
                    )
                    capture_event(event, hint=hint)

return result

@ensure_integration_enabled(AriadneIntegration, old_handle_query_result)
    def _sentry_patched_handle_query_result(
        result: "Any", *args: "Any", **kwargs: "Any"
    ) -> "GraphQLResult":
        query_result = old_handle_query_result(result, *args, **kwargs)

event_processor = _make_response_event_processor(query_result[1])
        sentry_sdk.get_isolation_scope().add_event_processor(event_processor)

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

return query_result

ariadne_graphql.parse_query = _sentry_patched_parse_query  # type: ignore
    ariadne_graphql.handle_graphql_errors = _sentry_patched_handle_graphql_errors  # type: ignore
    ariadne_graphql.handle_query_result = _sentry_patched_handle_query_result  # type: ignore

def _make_request_event_processor(data: "GraphQLSchema") -> "EventProcessor":
    """Add request data and api_target to events."""

def inner(event: "Event", hint: "dict[str, Any]") -> "Event":
        if not isinstance(data, dict):
            return event

with capture_internal_exceptions():
            try:
                content_length = int(
                    (data.get("headers") or {}).get("Content-Length", 0)
                )
            except (TypeError, ValueError):
                return event

if should_send_default_pii() and request_body_within_bounds(
                get_client(), content_length
            ):
                request_info = event.setdefault("request", {})
                request_info["api_target"] = "graphql"
                request_info["data"] = data

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

return event

return inner

def _make_response_event_processor(response: "Dict[str, Any]") -> "EventProcessor":
    """Add response data to the event's response context."""

def inner(event: "Event", hint: "dict[str, Any]") -> "Event":
        with capture_internal_exceptions():
            if should_send_default_pii() and response.get("errors"):
                contexts = event.setdefault("contexts", {})
                contexts["response"] = {
                    "data": response,
                }

return event

return inner

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