Skip to content

Commit 1474c4b

Browse files
committed
Refactor create_authn_request message
Signed-off-by: Ivan Kanakarakis <[email protected]>
1 parent 1747988 commit 1474c4b

File tree

1 file changed

+33
-11
lines changed

1 file changed

+33
-11
lines changed

src/saml2/client_base.py

Lines changed: 33 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -438,17 +438,39 @@ def create_authn_request(self, destination, vorg="", scoping=None,
438438
self.sec.cert_handler.update_cert(True, client_crt)
439439
if client_crt is not None:
440440
sign_prepare = True
441-
return self._message(AuthnRequest, destination, message_id,
442-
consent, extensions, sign, sign_prepare,
443-
protocol_binding=binding,
444-
scoping=scoping, nsprefix=nsprefix,
445-
sign_alg=sign_alg, digest_alg=digest_alg,
446-
**args)
447-
return self._message(AuthnRequest, destination, message_id, consent,
448-
extensions, sign, sign_prepare,
449-
protocol_binding=binding,
450-
scoping=scoping, nsprefix=nsprefix,
451-
sign_alg=sign_alg, digest_alg=digest_alg, **args)
441+
msg = self._message(
442+
AuthnRequest,
443+
destination,
444+
message_id,
445+
consent,
446+
extensions,
447+
sign,
448+
sign_prepare,
449+
protocol_binding=binding,
450+
scoping=scoping,
451+
nsprefix=nsprefix,
452+
sign_alg=sign_alg,
453+
digest_alg=digest_alg,
454+
**args,
455+
)
456+
else:
457+
msg = self._message(
458+
AuthnRequest,
459+
destination,
460+
message_id,
461+
consent,
462+
extensions,
463+
sign,
464+
sign_prepare,
465+
protocol_binding=binding,
466+
scoping=scoping,
467+
nsprefix=nsprefix,
468+
sign_alg=sign_alg,
469+
digest_alg=digest_alg,
470+
**args,
471+
)
472+
473+
return msg
452474

453475
def create_attribute_query(self, destination, name_id=None,
454476
attribute=None, message_id=0, consent=None,

0 commit comments

Comments
 (0)