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