Use six.string_types instead of basestring

In python3 strings are different, so basestring isn't available
anymore.  While examining these uses, all of them still work fine with
six.string_types and should not need any new special handling to deal
with bytes.

Author: SpamapS

setup.py

@@ -31,7 +31,8 @@ def run_tests(self):
     'pytz',
     'pyOpenSSL',
     'python-dateutil',
-    'argparse'
+    'argparse',
+    'six'
 ]
 
 tests_require = [

src/saml2/__init__.py

@@ -19,6 +19,7 @@
 
 
 import logging
+import six
 from saml2.validate import valid_instance
 
 try:
@@ -727,7 +728,7 @@ def set_text(self, val, base64encode=False):
                 setattr(self, "text", "false")
         elif isinstance(val, int):
             setattr(self, "text", "%d" % val)
-        elif isinstance(val, basestring):
+        elif isinstance(val, six.string_types):
             setattr(self, "text", val)
         elif val is None:
             pass
@@ -809,7 +810,7 @@ def __eq__(self, other):
                 continue
             svals = self.__dict__[key]
             ovals = other.__dict__[key]
-            if isinstance(svals, basestring):
+            if isinstance(svals, six.string_types):
                 if svals != ovals:
                     return False
             elif isinstance(svals, list):

src/saml2/assertion.py

@@ -5,6 +5,7 @@
 
 import re
 from saml2.saml import NAME_FORMAT_URI
+import six
 import xmlenc
 
 from saml2 import saml
@@ -31,7 +32,7 @@ def _filter_values(vals, vlist=None, must=False):
     if not vlist:  # No value specified equals any value
         return vals
 
-    if isinstance(vlist, basestring):
+    if isinstance(vlist, six.string_types):
         vlist = [vlist]
 
     res = []
@@ -237,7 +238,7 @@ def filter_attribute_value_assertions(ava, attribute_restrictions=None):
         else:
             if _rests is None:
                 continue
-            if isinstance(vals, basestring):
+            if isinstance(vals, six.string_types):
                 vals = [vals]
             rvals = []
             for restr in _rests:

src/saml2/authn.py

@@ -2,6 +2,7 @@
 from urllib import urlencode
 from urlparse import parse_qs
 from urlparse import urlsplit
+import six
 import time
 import ldap
 from saml2 import SAMLError
@@ -77,7 +78,7 @@ def create_return_url(base, query, **kwargs):
 
     for key, values in parse_qs(query).items():
         if key in kwargs:
-            if isinstance(kwargs[key], basestring):
+            if isinstance(kwargs[key], six.string_types):
                 kwargs[key] = [kwargs[key]]
             kwargs[key].extend(values)
         else:
@@ -86,7 +87,7 @@ def create_return_url(base, query, **kwargs):
     if part.query:
         for key, values in parse_qs(part.query).items():
             if key in kwargs:
-                if isinstance(kwargs[key], basestring):
+                if isinstance(kwargs[key], six.string_types):
                     kwargs[key] = [kwargs[key]]
                 kwargs[key].extend(values)
             else:
@@ -160,7 +161,7 @@ def verify(self, request, **kwargs):
         """
 
         #logger.debug("verify(%s)" % request)
-        if isinstance(request, basestring):
+        if isinstance(request, six.string_types):
             _dict = parse_qs(request)
         elif isinstance(request, dict):
             _dict = request

src/saml2/client.py

@@ -1,6 +1,7 @@
 # !/usr/bin/env python
 # -*- coding: utf-8 -*-
 #
+import six
 
 """Contains classes and functions that a SAML2.0 Service Provider (SP) may use
 to conclude its tasks.
@@ -149,7 +150,7 @@ def global_logout(self, name_id, reason="", expire=None, sign=None):
             conversation.
         """
 
-        if isinstance(name_id, basestring):
+        if isinstance(name_id, six.string_types):
             name_id = decode(name_id)
 
         logger.info("logout request for: %s" % name_id)
@@ -363,7 +364,7 @@ def do_assertion_id_request(self, assertion_ids, entity_id,
             raise NoServiceDefined("%s: %s" % (entity_id,
                                                "assertion_id_request_service"))
 
-        if isinstance(assertion_ids, basestring):
+        if isinstance(assertion_ids, six.string_types):
             assertion_ids = [assertion_ids]
 
         _id_refs = [AssertionIDRef(_id) for _id in assertion_ids]

src/saml2/client_base.py

@@ -8,6 +8,7 @@
 import threading
 from urllib import urlencode
 from urlparse import urlparse
+import six
 
 from saml2.entity import Entity
 
@@ -382,7 +383,7 @@ def create_attribute_query(self, destination, name_id=None,
                         pass
             else:
                 raise AttributeError("Missing required parameter")
-        elif isinstance(name_id, basestring):
+        elif isinstance(name_id, six.string_types):
             name_id = saml.NameID(text=name_id)
             for key in ["sp_name_qualifier", "name_qualifier", "format"]:
                 try:
@@ -453,7 +454,7 @@ def create_authz_decision_query_using_assertion(self, destination,
         """
 
         if action:
-            if isinstance(action, basestring):
+            if isinstance(action, six.string_types):
                 _action = [saml.Action(text=action)]
             else:
                 _action = [saml.Action(text=a) for a in action]
@@ -473,7 +474,7 @@ def create_assertion_id_request(assertion_id_refs, **kwargs):
         :return: One ID ref
         """
 
-        if isinstance(assertion_id_refs, basestring):
+        if isinstance(assertion_id_refs, six.string_types):
             return 0, assertion_id_refs
         else:
             return 0, assertion_id_refs[0]

src/saml2/httputil.py

@@ -3,6 +3,7 @@
 import logging
 import time
 import cgi
+import six
 
 from urllib import quote
 from urlparse import parse_qs
@@ -61,7 +62,7 @@ def _response(self, message="", **argv):
             mte = self.mako_lookup.get_template(self.mako_template)
             return [mte.render(**argv)]
         else:
-            if isinstance(message, basestring):
+            if isinstance(message, six.string_types):
                 return [message]
             else:
                 return message

src/saml2/mdie.py

@@ -1,4 +1,6 @@
 #!/usr/bin/env python
+import six
+
 from saml2 import element_to_extension_element
 from saml2 import extension_elements_to_elements
 from saml2 import SamlBase
@@ -22,7 +24,7 @@ def _eval(val, onts, mdb_safe):
     :param onts: Schemas to be used in the conversion
     :return: The basic dictionary
     """
-    if isinstance(val, basestring):
+    if isinstance(val, six.string_types):
         val = val.strip()
         if not val:
             return None
@@ -140,7 +142,7 @@ def from_dict(val, onts, mdb_safe=False):
                     key = key.replace("__", ".")
                 res[key] = from_dict(v, onts)
             return res
-    elif isinstance(val, basestring):
+    elif isinstance(val, six.string_types):
         return val
     elif isinstance(val, list):
         return [from_dict(v, onts) for v in val]

src/saml2/metadata.py

@@ -20,6 +20,7 @@
 from saml2 import class_name
 
 from saml2 import xmldsig as ds
+import six
 
 from saml2.sigver import pre_signature_part
 
@@ -137,7 +138,7 @@ def do_organization_info(ava):
     for dkey, (ckey, klass) in ORG_ATTR_TRANSL.items():
         if ckey not in ava:
             continue
-        if isinstance(ava[ckey], basestring):
+        if isinstance(ava[ckey], six.string_types):
             setattr(org, dkey, [_localized_name(ava[ckey], klass)])
         elif isinstance(ava[ckey], list):
             setattr(org, dkey,
@@ -163,7 +164,7 @@ def do_contact_person_info(lava):
                 data = []
                 if isinstance(classpec, list):
                     # What if value is not a list ?
-                    if isinstance(value, basestring):
+                    if isinstance(value, six.string_types):
                         data = [classpec[0](text=value)]
                     else:
                         for val in value:
@@ -253,7 +254,7 @@ def do_uiinfo(_uiinfo):
 
         aclass = uii.child_class(attr)
         inst = getattr(uii, attr)
-        if isinstance(val, basestring):
+        if isinstance(val, six.string_types):
             ainst = aclass(text=val)
             inst.append(ainst)
         elif isinstance(val, dict):
@@ -263,7 +264,7 @@ def do_uiinfo(_uiinfo):
             inst.append(ainst)
         else:
             for value in val:
-                if isinstance(value, basestring):
+                if isinstance(value, six.string_types):
                     ainst = aclass(text=value)
                     inst.append(ainst)
                 elif isinstance(value, dict):
@@ -299,11 +300,11 @@ def do_uiinfo(_uiinfo):
         _attr = "keywords"
         val = _uiinfo[_attr]
         inst = getattr(uii, _attr)
-        # list of basestrings, dictionary or list of dictionaries
+        # list of six.string_types, dictionary or list of dictionaries
         if isinstance(val, list):
             for value in val:
                 keyw = mdui.Keywords()
-                if isinstance(value, basestring):
+                if isinstance(value, six.string_types):
                     keyw.text = value
                 elif isinstance(value, dict):
                     keyw.text = " ".join(value["text"])
@@ -398,7 +399,7 @@ def do_extensions(mname, item):
 def _do_nameid_format(cls, conf, typ):
     namef = conf.getattr("name_id_format", typ)
     if namef:
-        if isinstance(namef, basestring):
+        if isinstance(namef, six.string_types):
             ids = [md.NameIDFormat(namef)]
         else:
             ids = [md.NameIDFormat(text=form) for form in namef]
@@ -413,7 +414,7 @@ def do_endpoints(conf, endpoints):
             servs = []
             i = 1
             for args in conf[endpoint]:
-                if isinstance(args, basestring):  # Assume it's the location
+                if isinstance(args, six.string_types):  # Assume it's the location
                     args = {"location": args,
                             "binding": DEFAULT_BINDING[endpoint]}
                 elif isinstance(args, tuple) or isinstance(args, list):

src/saml2/mongo_store.py

@@ -21,6 +21,7 @@
 from saml2.extension import ui
 from saml2 import xmldsig
 from saml2 import xmlenc
+import six
 
 
 ONTS = {
@@ -324,14 +325,14 @@ def protect(dic):
     res = {}
     for key, val in dic.items():
         key = key.replace(".", "__")
-        if isinstance(val, basestring):
+        if isinstance(val, six.string_types):
             pass
         elif isinstance(val, dict):
             val = protect(val)
         elif isinstance(val, list):
             li = []
             for va in val:
-                if isinstance(va, basestring):
+                if isinstance(va, six.string_types):
                     pass
                 elif isinstance(va, dict):
                     va = protect(va)
@@ -349,14 +350,14 @@ def unprotect(dic):
             pass
         else:
             key = key.replace("__", ".")
-        if isinstance(val, basestring):
+        if isinstance(val, six.string_types):
             pass
         elif isinstance(val, dict):
             val = unprotect(val)
         elif isinstance(val, list):
             li = []
             for va in val:
-                if isinstance(va, basestring):
+                if isinstance(va, six.string_types):
                     pass
                 elif isinstance(val, dict):
                     va = unprotect(va)