Skip to content

Commit 6acaf87

Browse files
committed
Simplify Base::create_discovery_service_request method
Allow values of None in the collected information. Filter out those fields later. Signed-off-by: Ivan Kanakarakis <[email protected]>
1 parent ab58c44 commit 6acaf87

File tree

1 file changed

+14
-21
lines changed

1 file changed

+14
-21
lines changed

src/saml2/client_base.py

Lines changed: 14 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -913,28 +913,21 @@ def create_discovery_service_request(url, entity_id, **kwargs):
913913
:return: A URL
914914
"""
915915

916-
args = {"entityID": entity_id}
917-
for key in ["policy", "returnIDParam"]:
918-
try:
919-
args[key] = kwargs[key]
920-
except KeyError:
921-
pass
922-
923-
try:
924-
args["return"] = kwargs["return_url"]
925-
except KeyError:
926-
try:
927-
args["return"] = kwargs["return"]
928-
except KeyError:
929-
pass
930-
931-
if "isPassive" in kwargs:
932-
if kwargs["isPassive"]:
933-
args["isPassive"] = "true"
934-
else:
935-
args["isPassive"] = "false"
916+
args = {
917+
"entityID": entity_id,
918+
"policy": kwargs.get("policy"),
919+
"returnIDParam": kwargs.get("returnIDParam"),
920+
"return": kwargs.get("return_url") or kwargs.get("return"),
921+
"isPassive": (
922+
None
923+
if "isPassive" not in kwargs.keys()
924+
else "true"
925+
if kwargs.get("isPassive")
926+
else "false"
927+
),
928+
}
936929

937-
params = urlencode(args)
930+
params = urlencode({k: v for k, v in args.items() if v})
938931
return "%s?%s" % (url, params)
939932

940933
@staticmethod

0 commit comments

Comments
 (0)