~azzar1/unity/add-show-desktop-key

1079 by William Grant
Merge setup-refactor branch. This completely breaks existing installations;
1
#!/usr/bin/env python
2
# IVLE - Informatics Virtual Learning Environment
3
# Copyright (C) 2007-2008 The University of Melbourne
4
#
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
9
#
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
# GNU General Public License for more details.
14
#
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
19
# Program: MakeUser
20
# Author:  Matt Giuca
21
# Date:    9/1/2008
22
23
# Script to create a new user. This can also be done through the
24
# administration interface.
25
# This script wraps common.makeuser.
26
# It also creates a unix account which common.makeuser does not.
27
# (This script may not be appropriate for production on a multi-node
28
# environment).
29
30
import sys
31
import os
1293 by William Grant
Allow admins to set the admin flag with ivle-adduser (issue #151).
32
import optparse
1079 by William Grant
Merge setup-refactor branch. This completely breaks existing installations;
33
1099.1.140 by Matt Giuca
ivle-makeuser: Moved root-user check earlier to avoid a nonsensical exception
34
if os.getuid() != 0:
35
    print "Must run %s as root." % os.path.basename(sys.argv[0])
36
    sys.exit(1)
37
1201 by William Grant
ivle.database.get_store() now takes a configuration object.
38
from ivle.config import Config
1080.1.70 by William Grant
bin/ivle-makeuser: Don't use ivle.makeuser.make_user_db; do it ourselves.
39
from ivle.database import get_store, User
40
from ivle.pulldown_subj import enrol_user
1079 by William Grant
Merge setup-refactor branch. This completely breaks existing installations;
41
1293 by William Grant
Allow admins to set the admin flag with ivle-adduser (issue #151).
42
usage = """usage: %prog [OPTIONS] <LOGIN> <FULLNAME>
43
Creates an IVLE user."""
44
45
parser = optparse.OptionParser(usage)
46
parser.add_option("-p", "--password",
47
    action="store", dest="password",
48
    help="Cleartext password"
49
    )
50
parser.add_option("-n", "--nick",
51
    action="store", dest="nick",
52
    help="Display name (defaults to <FULLNAME>)"
53
    )
54
parser.add_option("-e", "--email",
55
    action="store", dest="email",
56
    help="Email address"
57
    )
58
parser.add_option("-s", "--studentid",
59
    action="store", dest="studentid",
60
    help="Student ID"
61
    )
62
parser.add_option("--admin",
63
    action="store_true", dest="admin",
64
    help="Give the user full administrative privileges",
65
    default=False
66
    )
67
68
(options, args) = parser.parse_args()
69
70
if len(args) != 2:
71
    parser.error("incorrect number of arguments")
72
73
user = {'login': unicode(args[0]), 'fullname': unicode(args[1])}
74
75
for attr in ('password', 'nick', 'email', 'studentid', 'admin'):
76
    val = getattr(options, attr)
77
    if val is not None:
78
        if isinstance(val, str):
79
            val = unicode(val)
80
        user[attr] = val
81
1079 by William Grant
Merge setup-refactor branch. This completely breaks existing installations;
82
if 'nick' not in user:
83
    user['nick'] = user['fullname']
84
1279 by William Grant
Drop ivle.conf usage from ivle.pulldown_subj.
85
config = Config()
86
store = get_store(config)
1080.1.70 by William Grant
bin/ivle-makeuser: Don't use ivle.makeuser.make_user_db; do it ourselves.
87
1293 by William Grant
Allow admins to set the admin flag with ivle-adduser (issue #151).
88
if User.get_by_login(store, user['login']) is not None:
89
    print "user '%s' already exists" % user['login']
1079 by William Grant
Merge setup-refactor branch. This completely breaks existing installations;
90
    sys.exit(1)
91
1293 by William Grant
Allow admins to set the admin flag with ivle-adduser (issue #151).
92
# Make the user's database entry
93
userobj = User(**user)
94
store.add(userobj)
95
enrol_user(config, store, userobj)
96
store.commit()
97
98
print "successfully created user '%s' (%s)." % (userobj.login, userobj.fullname)