mirror of https://github.com/zulip/zulip.git
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
from typing import Any
|
|
|
|
from argparse import ArgumentParser
|
|
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
|
|
|
|
|
|
# Quick tool to test whether you're correctly authenticating to LDAP
|
|
def query_ldap(**options):
|
|
# type: (**str) -> None
|
|
email = options['email']
|
|
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 add_arguments(self, parser):
|
|
# type: (ArgumentParser) -> None
|
|
parser.add_argument('email', metavar='<email>', type=str,
|
|
help="email of user to query")
|
|
|
|
def handle(self, *args, **options):
|
|
# type: (*Any, **str) -> None
|
|
query_ldap(**options)
|