ref: Use scope.should_send_default_pii in FastAPI integration (getsentry#2846)

szokeasaurusrex · web-flow · commit 819fa0d40005 · 2024-03-19T14:33:34.000+01:00
* Created async and sync decorators * Added use of each sentry decorator * Fix circular import * Revert changes to starlette.py * Rename method * Use actual generics, move async implementation to utils * Refactor parameters * Undo changes to _types.py * Use client instead of Hub * Add doc string * Move type comments * Fix mypy * Fix circular import * Added unit tests for decorators * Revert gql changes * Revert "Revert gql changes" This reverts commit 85c1a1f. * ref: Shortcut for `should_send_default_pii` * revert gql changes * ref: Use `scope.should_send_default_pii` in FastAPI integration
diff --git a/sentry_sdk/integrations/fastapi.py b/sentry_sdk/integrations/fastapi.py
@@ -5,7 +5,7 @@
 import sentry_sdk
 from sentry_sdk._types import TYPE_CHECKING
 from sentry_sdk.integrations import DidNotEnable
-from sentry_sdk.scope import Scope
+from sentry_sdk.scope import Scope, should_send_default_pii
 from sentry_sdk.tracing import SOURCE_FOR_STYLE, TRANSACTION_SOURCE_ROUTE
 from sentry_sdk.utils import (
     transaction_from_function,
@@ -118,10 +118,7 @@ def event_processor(event, hint):
                     # Extract information from request
                     request_info = event.get("request", {})
                     if info:
-                        if (
-                            "cookies" in info
-                            and sentry_sdk.get_client().should_send_default_pii()
-                        ):
+                        if "cookies" in info and should_send_default_pii():
                             request_info["cookies"] = info["cookies"]
                         if "data" in info:
                             request_info["data"] = info["data"]
