X-Git-Url: https://git.mdrn.pl/django-cas-provider.git/blobdiff_plain/3043bb2f2f00a94f8499c00b54b446d2dbd07a98..743e480255ba992e40e4723644f1721b1e2928fe:/cas_provider/views.py

diff --git a/cas_provider/views.py b/cas_provider/views.py
index 3f2607f..715d0ce 100644
--- a/cas_provider/views.py
+++ b/cas_provider/views.py
@@ -336,12 +336,15 @@ def auth_success_response(user, pgt, proxies):
     username = etree.SubElement(auth_success, CAS + 'user')
     username.text = user.username
 
-    if settings.CAS_CUSTOM_ATTRIBUTES_CALLBACK:
-        callback = get_callable(settings.CAS_CUSTOM_ATTRIBUTES_CALLBACK)
-        attrs = callback(user)
-        if len(attrs) > 0:
-            formatter = get_callable(settings.CAS_CUSTOM_ATTRIBUTES_FORMATER)
-            formatter(auth_success, attrs)
+    attrs = {}
+    for receiver, response in signals.cas_collect_custom_attributes.send(sender=auth_success_response, user=user):
+        if response:
+            print "Got custom attributes:", response
+            attrs.update(response)
+
+    if attrs:
+        formatter = get_callable(settings.CAS_CUSTOM_ATTRIBUTES_FORMATER)
+        formatter(auth_success, attrs)
 
     if pgt:
         pgtElement = etree.SubElement(auth_success, CAS + 'proxyGrantingTicket')