diff --git a/fountain/management/commands/ldap_sync.py b/fountain/management/commands/ldap_sync.py index 8e58798157f95390bed09d556135a7cec5bc3a2b..13595887190f13fbe26ac9b9dc20f113cd9b15bf 100644 --- a/fountain/management/commands/ldap_sync.py +++ b/fountain/management/commands/ldap_sync.py @@ -1,15 +1,17 @@ -from django.core.management.base import NoArgsCommand +# -*- coding: utf-8 -*- +from __future__ import unicode_literals +from django.core.management.base import BaseCommand from fountain.ldap import Ldap from django.contrib.auth import get_user_model -class Command(NoArgsCommand): +class Command(BaseCommand): help = "Updates the attributes from all current Django users from a LDAP server." - def handle_noargs(self, **options): + def handle(self, *args, **options): verbosity = options.get('verbosity') User = get_user_model() l = Ldap() - + values = list(l.LDAP_SYNC_USER_ATTRIBUTES.values()) values.append(User.USERNAME_FIELD) for user_dict in User.objects.all().values(*values).iterator():