Make deflate process when retrieving built SAML messages optional

Author: pitbulk

src/onelogin/saml2/authn_request.py

@@ -8,12 +8,10 @@
 AuthNRequest class of OneLogin's Python Toolkit.
 
 """
-
 from base64 import b64encode
-from zlib import compress
 
-from onelogin.saml2.utils import OneLogin_Saml2_Utils
 from onelogin.saml2.constants import OneLogin_Saml2_Constants
+from onelogin.saml2.utils import OneLogin_Saml2_Utils
 
 
 class OneLogin_Saml2_Authn_Request(object):
@@ -121,14 +119,19 @@ def __init__(self, settings, force_authn=False, is_passive=False):
 
         self.__authn_request = request
 
-    def get_request(self):
+    def get_request(self, deflate=True):
         """
         Returns unsigned AuthnRequest.
-        :return: Unsigned AuthnRequest
+        :param deflate: It makes the deflate process optional
+        :type: bool
+        :return: AuthnRequest maybe deflated and base64 encoded
         :rtype: str object
         """
-        deflated_request = compress(self.__authn_request)[2:-4]
-        return b64encode(deflated_request)
+        if deflate:
+            request = OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__authn_request)
+        else:
+            request = b64encode(self.__authn_request)
+        return request
 
     def get_id(self):
         """

src/onelogin/saml2/logout_request.py

@@ -10,7 +10,7 @@
 """
 
 from zlib import decompress
-from base64 import b64decode
+from base64 import b64encode, b64decode
 from lxml import etree
 from defusedxml.lxml import fromstring
 from urllib import quote_plus
@@ -117,13 +117,19 @@ def __init__(self, settings, request=None, name_id=None, session_index=None, nq=
 
         self.__logout_request = logout_request
 
-    def get_request(self):
+    def get_request(self, deflate=True):
         """
         Returns the Logout Request defated, base64encoded
-        :return: Deflated base64 encoded Logout Request
+        :param deflate: It makes the deflate process optional
+        :type: bool
+        :return: Logout Request maybe deflated and base64 encoded
         :rtype: str object
         """
-        return OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__logout_request)
+        if deflate:
+            request = OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__logout_request)
+        else:
+            request = b64encode(self.__logout_request)
+        return request
 
     @staticmethod
     def get_id(request):

src/onelogin/saml2/logout_response.py

@@ -9,7 +9,7 @@
 
 """
 
-from base64 import b64decode
+from base64 import b64encode, b64decode
 from defusedxml.lxml import fromstring
 
 from urllib import quote_plus
@@ -188,13 +188,19 @@ def build(self, in_response_to):
 
         self.__logout_response = logout_response
 
-    def get_response(self):
+    def get_response(self, deflate=True):
         """
-        Returns a Logout Response object.
-        :return: Logout Response deflated and base64 encoded
+        Returns the Logout Response defated, base64encoded
+        :param deflate: It makes the deflate process optional
+        :type: bool
+        :return: Logout Response maybe deflated and base64 encoded
         :rtype: string
         """
-        return OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__logout_response)
+        if deflate:
+            response = OneLogin_Saml2_Utils.deflate_and_base64_encode(self.__logout_response)
+        else:
+            response = b64encode(self.__logout_response)
+        return response
 
     def get_error(self):
         """