Skip to content

Commit dec9834

Browse files
committed
Make deflate process when retrieving built SAML messages optional
1 parent 8734358 commit dec9834

3 files changed

Lines changed: 27 additions & 9 deletions

File tree

src/onelogin/saml2/authn_request.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,13 +104,19 @@ def __init__(self, settings, force_authn=False, is_passive=False):
104104

105105
self.__authn_request = request
106106

107-
def get_request(self):
107+
def get_request(self, deflate=True):
108108
"""
109109
Returns unsigned AuthnRequest.
110-
:return: Unsigned AuthnRequest
110+
:param deflate: It makes the deflate process optional
111+
:type: bool
112+
:return: AuthnRequest maybe deflated and base64 encoded
111113
:rtype: str object
112114
"""
113-
return OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__authn_request)
115+
if deflate:
116+
request = OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__authn_request)
117+
else:
118+
request = OneLogin_Saml2_Utils.b64encode(self.__authn_request)
119+
return request
114120

115121
def get_id(self):
116122
"""

src/onelogin/saml2/logout_request.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -97,13 +97,19 @@ def __init__(self, settings, request=None, name_id=None, session_index=None, nq=
9797

9898
self.__logout_request = logout_request
9999

100-
def get_request(self):
100+
def get_request(self, deflate=True):
101101
"""
102102
Returns the Logout Request defated, base64encoded
103-
:return: Deflated base64 encoded Logout Request
103+
:param deflate: It makes the deflate process optional
104+
:type: bool
105+
:return: Logout Request maybe deflated and base64 encoded
104106
:rtype: str object
105107
"""
106-
return OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__logout_request)
108+
if deflate:
109+
request = OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__logout_request)
110+
else:
111+
request = OneLogin_Saml2_Utils.b64encode(self.__logout_request)
112+
return request
107113

108114
@staticmethod
109115
def get_id(request):

src/onelogin/saml2/logout_response.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -147,13 +147,19 @@ def build(self, in_response_to):
147147

148148
self.__logout_response = logout_response
149149

150-
def get_response(self):
150+
def get_response(self, deflate=True):
151151
"""
152152
Returns a Logout Response object.
153-
:return: Logout Response deflated and base64 encoded
153+
:param deflate: It makes the deflate process optional
154+
:type: bool
155+
:return: Logout Response maybe deflated and base64 encoded
154156
:rtype: string
155157
"""
156-
return OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__logout_response)
158+
if deflate:
159+
response = OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__logout_response)
160+
else:
161+
response = OneLogin_Saml2_Utils.b64encode(self.__logout_response)
162+
return response
157163

158164
def get_error(self):
159165
"""

0 commit comments

Comments
 (0)