Skip to content

Commit 122625d

Browse files
committed
stop using helper for security strategies
1 parent c48f3b8 commit 122625d

File tree

1 file changed

+26
-16
lines changed

1 file changed

+26
-16
lines changed

src/sentry/grouping/strategies/security.py

+26-16
Original file line numberDiff line numberDiff line change
@@ -8,45 +8,55 @@
88
produces_variants,
99
strategy,
1010
)
11-
from sentry.interfaces.security import Csp, ExpectCT, ExpectStaple, Hpkp, SecurityReport
11+
from sentry.interfaces.security import Csp, ExpectCT, ExpectStaple, Hpkp
1212

1313

14-
def _security_v1(
15-
reported_id: str, obj: SecurityReport, context: GroupingContext, **meta: Any
14+
@strategy(ids=["expect-ct:v1"], interface=ExpectCT, score=1000)
15+
@produces_variants(["default"])
16+
def expect_ct_v1(
17+
interface: ExpectCT, event: Event, context: GroupingContext, **meta: Any
1618
) -> ReturnedVariants:
1719
return {
1820
context["variant"]: BaseGroupingComponent(
19-
id=reported_id,
21+
id="expect-ct",
2022
values=[
21-
BaseGroupingComponent(id="salt", values=[reported_id]),
22-
BaseGroupingComponent(id="hostname", values=[obj.hostname]),
23+
BaseGroupingComponent(id="salt", values=["expect-ct"]),
24+
BaseGroupingComponent(id="hostname", values=[interface.hostname]),
2325
],
2426
)
2527
}
2628

2729

28-
@strategy(ids=["expect-ct:v1"], interface=ExpectCT, score=1000)
29-
@produces_variants(["default"])
30-
def expect_ct_v1(
31-
interface: ExpectCT, event: Event, context: GroupingContext, **meta: Any
32-
) -> ReturnedVariants:
33-
return _security_v1("expect-ct", interface, context=context, **meta)
34-
35-
3630
@strategy(ids=["expect-staple:v1"], interface=ExpectStaple, score=1001)
3731
@produces_variants(["default"])
3832
def expect_staple_v1(
3933
interface: ExpectStaple, event: Event, context: GroupingContext, **meta: Any
4034
) -> ReturnedVariants:
41-
return _security_v1("expect-staple", interface, context=context, **meta)
35+
return {
36+
context["variant"]: BaseGroupingComponent(
37+
id="expect-staple",
38+
values=[
39+
BaseGroupingComponent(id="salt", values=["expect-staple"]),
40+
BaseGroupingComponent(id="hostname", values=[interface.hostname]),
41+
],
42+
)
43+
}
4244

4345

4446
@strategy(ids=["hpkp:v1"], interface=Hpkp, score=1002)
4547
@produces_variants(["default"])
4648
def hpkp_v1(
4749
interface: Hpkp, event: Event, context: GroupingContext, **meta: Any
4850
) -> ReturnedVariants:
49-
return _security_v1("hpkp", interface, context=context, **meta)
51+
return {
52+
context["variant"]: BaseGroupingComponent(
53+
id="hpkp",
54+
values=[
55+
BaseGroupingComponent(id="salt", values=["hpkp"]),
56+
BaseGroupingComponent(id="hostname", values=[interface.hostname]),
57+
],
58+
)
59+
}
5060

5161

5262
@strategy(ids=["csp:v1"], interface=Csp, score=1003)

0 commit comments

Comments
 (0)