Tool to test LDAP functionality

(imported from commit c4b4d8b25ba3afd333faf09bc232da2f92a57b7e)

zerver/management/commands/query_ldap.py

@@ -0,0 +1,29 @@
+from __future__ import absolute_import
+
+import sys
+
+from django.contrib.auth import authenticate, login, get_backends
+from django.core.management.base import BaseCommand
+from django.conf import settings
+
+from django_auth_ldap.backend import LDAPBackend, _LDAPUser
+
+
+# Run this on a cronjob to pick up on name changes.
+def query_ldap(*args):
+    if len(args) != 1:
+        print "Usage: query_ldap <email address>"
+        sys.exit(1)
+    email = args[0]
+    for backend in get_backends():
+        if isinstance(backend, LDAPBackend):
+            ldap_attrs = _LDAPUser(backend, backend.django_to_ldap_username(email)).attrs
+            if ldap_attrs is None:
+                print "No such user found"
+            else:
+                for django_field, ldap_field in settings.AUTH_LDAP_USER_ATTR_MAP.items():
+                    print "%s: %s" % (django_field, ldap_attrs[ldap_field])
+
+class Command(BaseCommand):
+    def handle(self, *args, **options):
+        query_ldap(*args)