code.grnet.gr Git - astakos/blob - snf-astakos-app/astakos/im/management/commands/modifyuser.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 from optparse import make_option
  35
  36 from django.core.management.base import BaseCommand, CommandError
  37
  38 from ._common import get_user
  39
  40
  41 class Command(BaseCommand):
  42     args = "<user_id or email>"
  43     help = "Modify a user's attributes"
  44
  45     option_list = BaseCommand.option_list + (
  46         make_option('--invitations',
  47             dest='invitations',
  48             metavar='NUM',
  49             help="Update user's invitations"),
  50         make_option('--password',
  51             dest='password',
  52             metavar='PASSWORD',
  53             help="Set user's password"),
  54         make_option('--renew-token',
  55             action='store_true',
  56             dest='renew_token',
  57             default=False,
  58             help="Renew the user's token"),
  59         make_option('--set-admin',
  60             action='store_true',
  61             dest='admin',
  62             default=False,
  63             help="Give user admin rights"),
  64         make_option('--set-noadmin',
  65             action='store_true',
  66             dest='noadmin',
  67             default=False,
  68             help="Revoke user's admin rights"),
  69         make_option('--set-active',
  70             action='store_true',
  71             dest='active',
  72             default=False,
  73             help="Change user's state to inactive"),
  74         make_option('--set-inactive',
  75             action='store_true',
  76             dest='inactive',
  77             default=False,
  78             help="Change user's state to inactive"),
  79         )
  80
  81     def handle(self, *args, **options):
  82         if len(args) != 1:
  83             raise CommandError("Please provide a user_id or email")
  84
  85         user = get_user(args[0])
  86         if not user:
  87             raise CommandError("Unknown user")
  88
  89         if options.get('admin'):
  90             user.is_superuser = True
  91         elif options.get('noadmin'):
  92             user.is_superuser = False
  93
  94         if options.get('active'):
  95             user.is_active = True
  96         elif options.get('inactive'):
  97             user.is_active = False
  98
  99         invitations = options.get('invitations')
 100         if invitations is not None:
 101             user.invitations = int(invitations)
 102
 103         password = options.get('password')
 104         if password is not None:
 105             user.set_password(password)
 106
 107         if options['renew_token']:
 108             user.renew_token()
 109
 110         user.save()