Skip to content

Commit 1bf3d0d

Browse files
committed
fix itemgetters
1 parent 93418fd commit 1bf3d0d

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

flag_engine/segments/evaluator.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@
22
import re
33
import typing
44
from contextlib import suppress
5-
from functools import partial, wraps
6-
from operator import itemgetter
5+
from functools import partial, reduce, wraps
76

87
import semver
98

@@ -224,7 +223,7 @@ def inner(
224223

225224

226225
CONTEXT_VALUE_GETTERS_BY_PROPERTY = {
227-
"$.identity.identifier": itemgetter("identity", "identifier"),
228-
"$.identity.key": itemgetter("identity", "key"),
229-
"$.environment.name": itemgetter("environment", "name"),
226+
"$.identity.identifier": partial(reduce, dict.__getitem__, ["identity", "identifier"]),
227+
"$.identity.key": partial(reduce, dict.__getitem__, ["identity", "key"]),
228+
"$.environment.name": partial(reduce, dict.__getitem__, ["environment", "name"]),
230229
}

0 commit comments

Comments
 (0)