Fix more basestring py3k issues

basestring has been removed from python 3.

Author: SpamapS

src/idp_test/__init__.py

@@ -6,6 +6,7 @@
 import types
 import argparse
 import sys
+import six
 
 import logging
 import imp
@@ -356,7 +357,7 @@ def list_operations(self):
             item = {"id": key, "name": val["name"]}
             try:
                 _desc = val["descr"]
-                if isinstance(_desc, basestring):
+                if isinstance(_desc, six.string_types):
                     item["descr"] = _desc
                 else:
                     item["descr"] = "\n".join(_desc)
@@ -377,7 +378,7 @@ def list_operations(self):
                 item = {"id": key, "name": val["name"]}
                 try:
                     _desc = val["descr"]
-                    if isinstance(_desc, basestring):
+                    if isinstance(_desc, six.string_types):
                         item["descr"] = _desc
                     else:
                         item["descr"] = "\n".join(_desc)

src/idp_test/interaction.py

@@ -2,6 +2,7 @@
 
 import json
 import logging
+import six
 
 from urlparse import urlparse
 from bs4 import BeautifulSoup
@@ -34,7 +35,8 @@ def pick_interaction(interactions, _base="", content="", req=None):
                     _match += 1
                 else:
                     _c = _bs.title.contents
-                    if isinstance(_c, list) and not isinstance(_c, basestring):
+                    if isinstance(_c, list) and not isinstance(
+                      _c, six.string_types):
                         for _line in _c:
                             if val in _line:
                                 _match += 1
@@ -165,7 +167,7 @@ def pick_form(response, url=None, **kwargs):
                         _default = _ava["value"]
                         try:
                             orig_val = form[prop]
-                            if isinstance(orig_val, basestring):
+                            if isinstance(orig_val, six.string_types):
                                 if orig_val == _default:
                                     _form = form
                             elif _default in orig_val:

src/saml2/entity.py

@@ -4,6 +4,7 @@
 from hashlib import sha1
 from Crypto.PublicKey import RSA
 import requests
+import six
 from saml2.metadata import ENDPOINTS
 from saml2.profile import paos, ecp
 from saml2.soap import parse_soap_enveloped_saml_artifact_resolve
@@ -156,7 +157,7 @@ def __init__(self, entity_type, config=None, config_file="",
         self.sec = security_context(self.config)
 
         if virtual_organization:
-            if isinstance(virtual_organization, basestring):
+            if isinstance(virtual_organization, six.string_types):
                 self.vorg = self.config.vorg[virtual_organization]
             elif isinstance(virtual_organization, VirtualOrg):
                 self.vorg = virtual_organization

src/saml2/httpbase.py

@@ -1,4 +1,5 @@
 import calendar
+import six
 from six.moves import http_cookiejar
 import copy
 import re
@@ -260,7 +261,7 @@ def use_http_form_post(message, destination, relay_state,
         :param typ: Whether a Request, Response or Artifact
         :return: dictionary
         """
-        if not isinstance(message, basestring):
+        if not isinstance(message, six.string_types):
             message = "%s" % (message,)
 
         return http_form_post_message(message, destination, relay_state, typ)
@@ -375,7 +376,7 @@ def use_http_get(message, destination, relay_state,
         :param key: Key to use for signing
         :return: dictionary
         """
-        if not isinstance(message, basestring):
+        if not isinstance(message, six.string_types):
             message = "%s" % (message,)
 
         return http_redirect_message(message, destination, relay_state, typ,

src/saml2/ident.py

@@ -1,6 +1,7 @@
 import copy
 import shelve
 import logging
+import six
 
 from hashlib import sha256
 from urllib import quote
@@ -66,7 +67,7 @@ class IdentDB(object):
      Keeps a list of all nameIDs returned per SP
     """
     def __init__(self, db, domain="", name_qualifier=""):
-        if isinstance(db, basestring):
+        if isinstance(db, six.string_types):
             self.db = shelve.open(db)
         else:
             self.db = db
@@ -94,7 +95,7 @@ def store(self, ident, name_id):
         :param ident: user identifier
         :param name_id: NameID instance
         """
-        if isinstance(ident, unicode):
+        if isinstance(ident, six.string_types):
             ident = ident.encode("utf-8")
 
         # One user may have more than one NameID defined

src/saml2/pack.py

@@ -20,6 +20,7 @@
 from saml2.sigver import REQ_ORDER
 from saml2.sigver import RESP_ORDER
 from saml2.sigver import SIGNER_ALGS
+import six
 
 logger = logging.getLogger(__name__)
 
@@ -57,7 +58,7 @@ def http_form_post_message(message, location, relay_state="",
     """
     response = ["<head>", """<title>SAML 2.0 POST</title>""", "</head><body>"]
 
-    if not isinstance(message, basestring):
+    if not isinstance(message, six.string_types):
         message = "%s" % (message,)
 
     if typ == "SAMLRequest" or typ == "SAMLResponse":
@@ -94,7 +95,7 @@ def http_redirect_message(message, location, relay_state="", typ="SAMLRequest",
     :return: A tuple containing header information and a HTML message.
     """
 
-    if not isinstance(message, basestring):
+    if not isinstance(message, six.string_types):
         message = "%s" % (message,)
 
     _order = None
@@ -163,7 +164,7 @@ def make_soap_enveloped_saml_thingy(thingy, header_parts=None):
     body.tag = '{%s}Body' % NAMESPACE
     envelope.append(body)
 
-    if isinstance(thingy, basestring):
+    if isinstance(thingy, six.string_types):
         # remove the first XML version/encoding line
         logger.debug("thingy0: %s" % thingy)
         _part = thingy.split("\n")

src/saml2/s2repoze/plugins/sp.py

@@ -12,6 +12,7 @@
 import shelve
 import traceback
 import saml2
+import six
 from urlparse import parse_qs, urlparse
 from saml2.md import Extensions
 from saml2 import xmldsig as ds
@@ -555,7 +556,7 @@ def identify(self, environ):
     def add_metadata(self, environ, identity):
         """ Add information to the knowledge I have about the user """
         name_id = identity['repoze.who.userid']
-        if isinstance(name_id, basestring):
+        if isinstance(name_id, six.string_types):
             try:
                 # Make sure that userids authenticated by another plugin
                 # don't cause problems here.

src/saml2/sigver.py

@@ -15,6 +15,7 @@
 
 from time import mktime
 from binascii import hexlify
+import six
 
 from Crypto.PublicKey.RSA import importKey
 from Crypto.Signature import PKCS1_v1_5
@@ -728,7 +729,7 @@ class CryptoBackendXmlSec1(CryptoBackend):
 
     def __init__(self, xmlsec_binary, **kwargs):
         CryptoBackend.__init__(self, **kwargs)
-        assert (isinstance(xmlsec_binary, basestring))
+        assert (isinstance(xmlsec_binary, six.string_types))
         self.xmlsec = xmlsec_binary
         if os.environ.get('PYSAML2_KEEP_XMLSEC_TMP', None):
             self._xmlsec_delete_tmpfiles = False
@@ -1353,7 +1354,7 @@ def _check_signature(self, decoded_xml, item, node_name=NODE_NAME,
                 _certs = []
             certs = []
             for cert in _certs:
-                if isinstance(cert, basestring):
+                if isinstance(cert, six.string_types):
                     certs.append(make_temp(pem_format(cert), suffix=".pem",
                                            decode=False,
                                            delete=self._xmlsec_delete_tmpfiles))

src/saml2test/check.py

@@ -1,5 +1,6 @@
 import inspect
 import json
+import six
 
 __author__ = 'rolandh'
 
@@ -91,7 +92,7 @@ def _func(self, conv=None):
         self._status = self.status
         _msg = conv.last_content
 
-        if isinstance(_msg, basestring):
+        if isinstance(_msg, six.string_types):
             self._message = _msg
         else:
             self._message = _msg.to_dict()

src/saml2test/interaction.py

@@ -2,6 +2,7 @@
 
 import json
 import logging
+import six
 
 from urlparse import urlparse
 from bs4 import BeautifulSoup
@@ -125,8 +126,8 @@ def pick_interaction(self, _base="", content="", req=None):
                         _match += 1
                     else:
                         _c = _bs.title.contents
-                        if isinstance(_c, list) and not isinstance(_c,
-                                                                   basestring):
+                        if isinstance(_c, list) and not isinstance(
+                          _c, six.string_types):
                             for _line in _c:
                                 if val in _line:
                                     _match += 1
@@ -186,7 +187,7 @@ def pick_form(self, response, url=None, **kwargs):
                             _default = _ava["value"]
                             try:
                                 orig_val = form[prop]
-                                if isinstance(orig_val, basestring):
+                                if isinstance(orig_val, six.string_types):
                                     if orig_val == _default:
                                         _form = form
                                 elif _default in orig_val: