Skip to content

Instantly share code, notes, and snippets.

@JosephRedfern

JosephRedfern/add_user.py

Created October 8, 2018 17:01
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save JosephRedfern/65c0534bda8e3606ec298462896d484d to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save JosephRedfern/65c0534bda8e3606ec298462896d484d to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. JosephRedfern created this gist Oct 8, 2018.
    36 changes: 36 additions & 0 deletions add_user.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,36 @@
    from django.core.management.base import BaseCommand, CommandError
    from django.contrib.auth.models import User
    from django.conf import settings
    import ldap

    class Command(BaseCommand):
    help = 'Add a specific user from LDAP'

    def add_arguments(self, parser):
    parser.add_argument('user_id', nargs='+', type=str)

    def handle(self, *args, **options):
    l = ldap.initialize(settings.AUTH_LDAP_SERVER_URI)

    total = 0
    total_created = 0

    for user_id in options['user_id']:
    print("Looking up {}".format(user_id))
    results = l.search_s(settings.USERS_DN, ldap.SCOPE_SUBTREE, "(uid={})".format(user_id))
    for e, r in results:
    username = r['uid'][0].decode('utf-8')
    first_name = r['givenName'][0].decode('utf-8')
    last_name = r['sn'][0].decode('utf-8')
    email = r['mail'][0].decode('utf-8')
    user, created = User.objects.get_or_create(username=username, email=email, first_name=first_name, last_name=last_name)
    total += 1
    if created:
    user.set_unusable_password()
    user.save()
    total_created += 1
    print("Added {}".format(user_id))
    else:
    print("{} already exists".format(user_id))

    print("Found {}/{} users, added {}".format(total, len(options['user_id']), total_created))