Skip to content

Commit 7fb69b4

Browse files
author
Ioannis Kakavas
committed
Reformat based on @c00kiemons5ter comments
1 parent 4a94123 commit 7fb69b4

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

src/satosa/base.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -103,13 +103,15 @@ def _auth_req_callback_func(self, context, internal_request):
103103
state = context.state
104104
state[STATE_KEY] = {"requester": internal_request.requester}
105105
# TODO consent module should manage any state it needs by itself
106-
if consent.STATE_KEY in context.state:
107-
context.state[consent.STATE_KEY]["filter"] = internal_request.approved_attributes or []
108-
context.state[consent.STATE_KEY]["requester_name"] = internal_request.requester_name
109-
else:
110-
context.state[consent.STATE_KEY] = {"filter": internal_request.approved_attributes or [],
111-
"requester_name": internal_request.requester_name}
112-
106+
try:
107+
state_dict = context.state[consent.STATE_KEY]
108+
except KeyError:
109+
state_dict = context.state[consent.STATE_KEY] = {}
110+
finally:
111+
state_dict.update({
112+
"filter": internal_request.approved_attributes or [],
113+
"requester_name": internal_request.requester_name,
114+
})
113115
satosa_logging(logger, logging.INFO,
114116
"Requesting provider: {}".format(internal_request.requester), state)
115117

0 commit comments

Comments
 (0)