← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-adduser

  • Committer: William Grant
  • Date: 2012-06-28 01:52:02 UTC
  • Revision ID: me@williamgrant.id.au-20120628015202-f6ru7o367gt6nvgz
Hah

Show diffs side-by-side

added added

removed removed

Lines of Context:
bin/ivle-adduser
 
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
 
32
import optparse
 
33
 
 
34
if os.getuid() != 0:
 
35
    print "Must run %s as root." % os.path.basename(sys.argv[0])
 
36
    sys.exit(1)
 
37
 
 
38
from ivle.config import Config
 
39
from ivle.database import get_store, User
 
40
from ivle.pulldown_subj import enrol_user
 
41
 
 
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
 
 
82
if 'nick' not in user:
 
83
    user['nick'] = user['fullname']
 
84
 
 
85
config = Config()
 
86
store = get_store(config)
 
87
 
 
88
if User.get_by_login(store, user['login']) is not None:
 
89
    print "user '%s' already exists" % user['login']
 
90
    sys.exit(1)
 
91
 
 
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)