Extend the service URL field.

[django-cas-provider.git] / cas_provider / attribute_formatters.py

diff --git a/cas_provider/attribute_formatters.py b/cas_provider/attribute_formatters.py
index 3f3ab3c..8574962 100644 (file)
--- a/cas_provider/attribute_formatters.py
+++ b/cas_provider/attribute_formatters.py
@@ -1,16 +1,25 @@
+from __future__ import unicode_literals
+

 from lxml import etree
 from lxml import etree

+import collections

 
 CAS_URI = 'http://www.yale.edu/tp/cas'
 NSMAP = {'cas': CAS_URI}
 CAS = '{%s}' % CAS_URI
 
 
 
 CAS_URI = 'http://www.yale.edu/tp/cas'
 NSMAP = {'cas': CAS_URI}
 CAS = '{%s}' % CAS_URI
 
 

+try:
+    basestring
+except NameError:
+    basestring = (str, bytes)
+
+

 def jasig(auth_success, attrs):
     attributes = etree.SubElement(auth_success, CAS + 'attributes')
     style = etree.SubElement(attributes, CAS + 'attraStyle')
 def jasig(auth_success, attrs):
     attributes = etree.SubElement(auth_success, CAS + 'attributes')
     style = etree.SubElement(attributes, CAS + 'attraStyle')

-    style.text = u'Jasig'
-    for name, value in attrs.items():
-        if isinstance(value, list):
+    style.text = 'Jasig'
+    for name, value in sorted(attrs.items()):
+        if isinstance(value, collections.Iterable) and not isinstance(value, basestring):

             for e in value:
                 element = etree.SubElement(attributes, CAS + name)
                 element.text = e
             for e in value:
                 element = etree.SubElement(attributes, CAS + name)
                 element.text = e


@@ -21,9 +30,9 @@ def jasig(auth_success, attrs):
 
 def ruby_cas(auth_success, attrs):
     style = etree.SubElement(auth_success, CAS + 'attraStyle')
 
 def ruby_cas(auth_success, attrs):
     style = etree.SubElement(auth_success, CAS + 'attraStyle')

-    style.text = u'RubyCAS'
-    for name, value in attrs.items():
-        if isinstance(value, list):
+    style.text = 'RubyCAS'
+    for name, value in sorted(attrs.items()):
+        if isinstance(value, collections.Iterable) and not isinstance(value, basestring):

             for e in value:
                 element = etree.SubElement(auth_success, CAS + name)
                 element.text = e
             for e in value:
                 element = etree.SubElement(auth_success, CAS + name)
                 element.text = e


@@ -31,10 +40,11 @@ def ruby_cas(auth_success, attrs):
             element = etree.SubElement(auth_success, CAS + name)
             element.text = value
 
             element = etree.SubElement(auth_success, CAS + name)
             element.text = value
 

+

 def name_value(auth_success, attrs):
 def name_value(auth_success, attrs):

-    etree.SubElement(auth_success, CAS + 'attribute', name=u'attraStyle', value=u'Name-Value')
-    for name, value in attrs.items():
-        if isinstance(value, list):
+    etree.SubElement(auth_success, CAS + 'attribute', name='attraStyle', value='Name-Value')
+    for name, value in sorted(attrs.items()):
+        if isinstance(value, collections.Iterable) and not isinstance(value, basestring):

             for e in value:
                 etree.SubElement(auth_success, CAS + 'attribute', name=name, value=e)
         else:
             for e in value:
                 etree.SubElement(auth_success, CAS + 'attribute', name=name, value=e)
         else: