Skip to content

Commit 85d7458

Browse files
committed
1 parent 2c8d277 commit 85d7458

File tree

3 files changed

+98
-0
lines changed

3 files changed

+98
-0
lines changed

sentry_sdk/client.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
from sentry_sdk.sessions import SessionFlusher
3030
from sentry_sdk.envelope import Envelope
3131
from sentry_sdk.profiler import setup_profiler
32+
from sentry_sdk.scrubber import EventScrubber
3233

3334
from sentry_sdk._types import TYPE_CHECKING
3435

@@ -111,6 +112,9 @@ def _get_options(*args, **kwargs):
111112
if rv["enable_tracing"] is True and rv["traces_sample_rate"] is None:
112113
rv["traces_sample_rate"] = 1.0
113114

115+
if rv["event_scrubber"] is None:
116+
rv["event_scrubber"] = EventScrubber()
117+
114118
return rv
115119

116120

@@ -250,6 +254,11 @@ def _prepare_event(
250254
self.options["project_root"],
251255
)
252256

257+
if event is not None:
258+
event_scrubber = self.options["event_scrubber"]
259+
if event_scrubber and not self.options["send_default_pii"]:
260+
event_scrubber.scrub_event(event)
261+
253262
# Postprocess the event here so that annotated types do
254263
# generally not surface in before_send
255264
if event is not None:

sentry_sdk/consts.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ def __init__(
128128
trace_propagation_targets=[ # noqa: B006
129129
MATCH_ALL
130130
], # type: Optional[Sequence[str]]
131+
event_scrubber=None, # type: Optional[sentry_sdk.scrubber.EventScrubber]
131132
):
132133
# type: (...) -> None
133134
pass

sentry_sdk/scrubber.py

Lines changed: 88 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,88 @@
1+
from sentry_sdk.utils import (
2+
capture_internal_exceptions,
3+
AnnotatedValue,
4+
iter_event_frames,
5+
)
6+
from sentry_sdk._compat import string_types
7+
from sentry_sdk._types import TYPE_CHECKING
8+
9+
if TYPE_CHECKING:
10+
from sentry_sdk._types import Event
11+
from typing import Any
12+
from typing import Dict
13+
from typing import List
14+
from typing import Optional
15+
16+
17+
DEFAULT_DENYLIST = ["password", "custom-header", "custom_var"]
18+
19+
20+
class EventScrubber(object):
21+
def __init__(self, denylist=None):
22+
# type: (Optional[List[str]]) -> None
23+
self.denylist = DEFAULT_DENYLIST if denylist is None else denylist
24+
25+
def scrub_dict(self, d):
26+
# type: (Dict[str, Any]) -> None
27+
if not isinstance(d, dict):
28+
return
29+
30+
for k in d.keys():
31+
if isinstance(k, string_types) and k.lower() in self.denylist:
32+
d[k] = AnnotatedValue.substituted_because_contains_sensitive_data()
33+
34+
def scrub_request(self, event):
35+
# type: (Event) -> None
36+
with capture_internal_exceptions():
37+
if "request" in event:
38+
if "headers" in event["request"]:
39+
self.scrub_dict(event["request"]["headers"])
40+
if "cookies" in event["request"]:
41+
self.scrub_dict(event["request"]["cookies"])
42+
if "data" in event["request"]:
43+
self.scrub_dict(event["request"]["data"])
44+
45+
def scrub_extra(self, event):
46+
# type: (Event) -> None
47+
with capture_internal_exceptions():
48+
if "extra" in event:
49+
self.scrub_dict(event["extra"])
50+
51+
def scrub_user(self, event):
52+
# type: (Event) -> None
53+
with capture_internal_exceptions():
54+
if "user" in event:
55+
self.scrub_dict(event["user"])
56+
57+
def scrub_breadcrumbs(self, event):
58+
# type: (Event) -> None
59+
with capture_internal_exceptions():
60+
if "breadcrumbs" in event:
61+
if "values" in event["breadcrumbs"]:
62+
for value in event["breadcrumbs"]["values"]:
63+
if "data" in value:
64+
self.scrub_dict(value["data"])
65+
66+
def scrub_frames(self, event):
67+
# type: (Event) -> None
68+
with capture_internal_exceptions():
69+
for frame in iter_event_frames(event):
70+
if "vars" in frame:
71+
self.scrub_dict(frame["vars"])
72+
73+
def scrub_spans(self, event):
74+
# type: (Event) -> None
75+
with capture_internal_exceptions():
76+
if "spans" in event:
77+
for span in event["spans"]:
78+
if "data" in span:
79+
self.scrub_dict(span["data"])
80+
81+
def scrub_event(self, event):
82+
# type: (Event) -> None
83+
self.scrub_request(event)
84+
self.scrub_extra(event)
85+
self.scrub_user(event)
86+
self.scrub_breadcrumbs(event)
87+
self.scrub_frames(event)
88+
self.scrub_spans(event)

0 commit comments

Comments
 (0)