code.grnet.gr Git - astakos/blob - snf-astakos-app/astakos/im/management/commands/createuser.py

   1 # Copyright 2012 GRNET S.A. All rights reserved.
   2 #
   3 # Redistribution and use in source and binary forms, with or
   4 # without modification, are permitted provided that the following
   5 # conditions are met:
   6 #
   7 #   1. Redistributions of source code must retain the above
   8 #      copyright notice, this list of conditions and the following
   9 #      disclaimer.
  10 #
  11 #   2. Redistributions in binary form must reproduce the above
  12 #      copyright notice, this list of conditions and the following
  13 #      disclaimer in the documentation and/or other materials
  14 #      provided with the distribution.
  15 #
  16 # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  17 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  20 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  21 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  23 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  24 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  26 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  27 # POSSIBILITY OF SUCH DAMAGE.
  28 #
  29 # The views and conclusions contained in the software and
  30 # documentation are those of the authors and should not be
  31 # interpreted as representing official policies, either expressed
  32 # or implied, of GRNET S.A.
  33
  34 import socket
  35
  36 from optparse import make_option
  37 from random import choice
  38 from string import digits, lowercase, uppercase
  39 from uuid import uuid4
  40
  41 from django.core.management.base import BaseCommand, CommandError
  42 from django.core.validators import validate_email
  43 from django.core.exceptions import ValidationError
  44
  45 from astakos.im.models import AstakosUser
  46 from astakos.im.util import reserved_email
  47
  48 class Command(BaseCommand):
  49     args = "<email> <first name> <last name> <affiliation>"
  50     help = "Create a user"
  51
  52     option_list = BaseCommand.option_list + (
  53         make_option('--active',
  54             action='store_true',
  55             dest='active',
  56             default=False,
  57             help="Activate user"),
  58         make_option('--admin',
  59             action='store_true',
  60             dest='admin',
  61             default=False,
  62             help="Give user admin rights"),
  63         make_option('--password',
  64             dest='password',
  65             metavar='PASSWORD',
  66             help="Set user's password")
  67         )
  68
  69     def handle(self, *args, **options):
  70         if len(args) != 4:
  71             raise CommandError("Invalid number of arguments")
  72
  73         args = [a.decode('utf8') for a in args]
  74         email, first, last, affiliation = args
  75
  76         try:
  77             validate_email( email )
  78         except ValidationError:
  79             raise CommandError("Invalid email")
  80
  81         username =  uuid4().hex[:30]
  82         password = options.get('password')
  83         if password is None:
  84             password = AstakosUser.objects.make_random_password()
  85
  86         if reserved_email(email):
  87             raise CommandError("A user with this email already exists")
  88
  89         user = AstakosUser(username=username, first_name=first, last_name=last,
  90                            email=email, affiliation=affiliation,
  91                            provider='local')
  92         user.set_password(password)
  93         user.renew_token()
  94
  95         if options['active']:
  96             user.is_active = True
  97         if options['admin']:
  98             user.is_admin = True
  99
 100         try:
 101             user.save()
 102         except socket.error, e:
 103             raise CommandError(e)
 104         except ValidationError, e:
 105             raise CommandError(e)
 106         else:
 107             msg = "Created user id %d" % (user.id,)
 108             if options['password'] is None:
 109                 msg += " with password '%s'" % (password,)
 110             self.stdout.write(msg + '\n')