Statistics
| Branch: | Tag: | Revision:

root / snf-astakos-app / astakos / im / management / commands / modifyuser.py @ 3597f9c9

History | View | Annotate | Download (4 kB)

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('--level',
51
            dest='level',
52
            metavar='NUM',
53
            help="Update user's level"),
54
        make_option('--password',
55
            dest='password',
56
            metavar='PASSWORD',
57
            help="Set user's password"),
58
        make_option('--renew-token',
59
            action='store_true',
60
            dest='renew_token',
61
            default=False,
62
            help="Renew the user's token"),
63
        make_option('--set-admin',
64
            action='store_true',
65
            dest='admin',
66
            default=False,
67
            help="Give user admin rights"),
68
        make_option('--set-noadmin',
69
            action='store_true',
70
            dest='noadmin',
71
            default=False,
72
            help="Revoke user's admin rights"),
73
        make_option('--set-active',
74
            action='store_true',
75
            dest='active',
76
            default=False,
77
            help="Change user's state to inactive"),
78
        make_option('--set-inactive',
79
            action='store_true',
80
            dest='inactive',
81
            default=False,
82
            help="Change user's state to inactive"),
83
        )
84
    
85
    def handle(self, *args, **options):
86
        if len(args) != 1:
87
            raise CommandError("Please provide a user ID or email")
88
        
89
        user = get_user(args[0])
90
        if not user:
91
            raise CommandError("Unknown user")
92
        
93
        if options.get('admin'):
94
            user.is_superuser = True
95
        elif options.get('noadmin'):
96
            user.is_superuser = False
97
        
98
        if options.get('active'):
99
            user.is_active = True
100
        elif options.get('inactive'):
101
            user.is_active = False
102
        
103
        invitations = options.get('invitations')
104
        if invitations is not None:
105
            user.invitations = int(invitations)
106
        
107
        level = options.get('level')
108
        if level is not None:
109
            user.level = int(level)
110
        
111
        password = options.get('password')
112
        if password is not None:
113
            user.set_password(password)
114
        
115
        if options['renew_token']:
116
            user.renew_token()
117
        
118
        user.save()