summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorIvan Kanakarakis <ivan.kanak@gmail.com>2020-03-18 16:07:16 +0200
committerIvan Kanakarakis <ivan.kanak@gmail.com>2020-03-18 16:08:35 +0200
commit1b9723753fae3fc535b0e18e236d2d87f3cc2353 (patch)
treef2c0fc8c20d11315112a9917ae87c05b4841ba87
parentb2335fc07d6feeaa9a05fd6f9c84f5137a48b040 (diff)
downloadpysaml2-1b9723753fae3fc535b0e18e236d2d87f3cc2353.tar.gz
Replace decodestring for decodebytes from base64 module
decodestring has been removed from py39 Signed-off-by: Ivan Kanakarakis <ivan.kanak@gmail.com>
-rw-r--r--src/saml2/saml.py11
1 files changed, 3 insertions, 8 deletions
diff --git a/src/saml2/saml.py b/src/saml2/saml.py
index c4d9f2b2..48edf415 100644
--- a/src/saml2/saml.py
+++ b/src/saml2/saml.py
@@ -87,10 +87,6 @@ XSD = "xs"
NS_SOAP_ENC = "http://schemas.xmlsoap.org/soap/encoding/"
-_b64_decode_fn = getattr(base64, 'decodebytes', base64.decodestring)
-_b64_encode_fn = getattr(base64, 'encodebytes', base64.encodestring)
-
-
class AttributeValueBase(SamlBase):
def __init__(self,
text=None,
@@ -232,10 +228,9 @@ class AttributeValueBase(SamlBase):
'base64Binary': {
'type': _str,
'to_type': _str,
- 'to_text': lambda x:
- _b64_encode_fn(x.encode())
- if base64encode
- else x,
+ 'to_text': (
+ lambda x: base64.encodebytes(x.encode()) if base64encode else x
+ ),
},
'anyType': {
'type': type(value),