Synnefo

from django.core.management.base import BaseCommand, CommandError

from django.core.validators import validate_email

from astakos.im.models import AstakosUser

class Command(BaseCommand):

                    dest='password',

                    metavar='PASSWORD',

                    help="Set user's password"),

        make_option('--admin',

                    action='store_true',

                    dest='is_superuser',

        make_option('-g',

                    action='append',

                    dest='groups',

                    help="Add user group (may be used multiple times)"),

        make_option('-p',

                    action='append',

                    dest='permissions',

                    help="Add user permission (may be used multiple times)")

    )

    def handle(self, *args, **options):

        if len(args) != 3:

            raise CommandError("Invalid number of arguments")

        email, first_name, last_name = map(lambda arg: arg.decode('utf8'),

                                           args[:3])

        try:

            validate_email(email)

        except ValidationError:

            raise CommandError("Invalid email")

        try:

            u = AstakosUser(email=email,

                            first_name=first_name,

                u.add_auth_provider('local')

                map(u.add_permission, options['permissions'])

                map(u.add_group, options['groups'])

            except BaseException, e:

                import traceback

                traceback.print_exc()