Ticket #1178: custom_identity_provider.diff

turbogears/identity/__init__.py

@@ -10 +10 @@
 log = logging.getLogger("turbogears.identity")
 
 import turbogears
-from turbogears.util import request_available
+from turbogears.util import request_available, load_class
 from turbogears.identity.exceptions import *
 
 def create_default_provider():
@@ -19 +19 @@
                                               provider_plugin )
 
     log.debug( "Loading provider from plugin: %s", provider_plugin )
-
+    
+    provider_class = None
     for entrypoint in plugins:
-        plugin= entrypoint.load()
-        return plugin()
+        provider_class = entrypoint.load()
+        break
+    
+    if not provider_class:
+        provider_class = load_class(provider_plugin)
 
-    raise IdentityConfigurationException( "IdentityProvider plugin missing: %s" %
-                                          provider_plugin )
+    if not provider_class:
+        raise IdentityConfigurationException( "IdentityProvider plugin missing: %s" %
+                                              provider_plugin )
+    else:
+        return provider_class()
 
 
 class IdentityWrapper(object):