diff options
| author | Ivan Kanakarakis <ivan.kanak@gmail.com> | 2020-06-11 02:11:00 +0300 |
|---|---|---|
| committer | Ivan Kanakarakis <ivan.kanak@gmail.com> | 2020-06-11 02:11:00 +0300 |
| commit | 1474c4bb03439cbf382757e21b7926aa3c52017c (patch) | |
| tree | 96829fa20e597464e13d07494a6097b865c8c2e1 | |
| parent | 174798886fe7037c056e1db70fce48ee8f6dd86b (diff) | |
| download | pysaml2-1474c4bb03439cbf382757e21b7926aa3c52017c.tar.gz | |
Refactor create_authn_request message
Signed-off-by: Ivan Kanakarakis <ivan.kanak@gmail.com>
| -rw-r--r-- | src/saml2/client_base.py | 44 |
1 files changed, 33 insertions, 11 deletions
diff --git a/src/saml2/client_base.py b/src/saml2/client_base.py index 0b35fd8d..3362d7e5 100644 --- a/src/saml2/client_base.py +++ b/src/saml2/client_base.py @@ -438,17 +438,39 @@ class Base(Entity): self.sec.cert_handler.update_cert(True, client_crt) if client_crt is not None: sign_prepare = True - return self._message(AuthnRequest, destination, message_id, - consent, extensions, sign, sign_prepare, - protocol_binding=binding, - scoping=scoping, nsprefix=nsprefix, - sign_alg=sign_alg, digest_alg=digest_alg, - **args) - return self._message(AuthnRequest, destination, message_id, consent, - extensions, sign, sign_prepare, - protocol_binding=binding, - scoping=scoping, nsprefix=nsprefix, - sign_alg=sign_alg, digest_alg=digest_alg, **args) + msg = self._message( + AuthnRequest, + destination, + message_id, + consent, + extensions, + sign, + sign_prepare, + protocol_binding=binding, + scoping=scoping, + nsprefix=nsprefix, + sign_alg=sign_alg, + digest_alg=digest_alg, + **args, + ) + else: + msg = self._message( + AuthnRequest, + destination, + message_id, + consent, + extensions, + sign, + sign_prepare, + protocol_binding=binding, + scoping=scoping, + nsprefix=nsprefix, + sign_alg=sign_alg, + digest_alg=digest_alg, + **args, + ) + + return msg def create_attribute_query(self, destination, name_id=None, attribute=None, message_id=0, consent=None, |