← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-makeuser

Added new script ivle-fetchsubmissions, which pulls submissions from students
svn repos. Currently does all the busy work (takes input, gets data from db,
creates directories), but doesn't actually do any Subversion work.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bin/ivle-makeuser
 
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 getopt
 
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
# Requireds and optionals will be used to display the usage message
 
43
# AND do argument processing
 
44
# The names here must correspond to the fields in the database.
 
45
requireds = ["login", "fullname"]
 
46
optionals = [
 
47
    ('p', 'password', "Cleartext password for this user"),
 
48
    ('n', 'nick', "Display name (defaults to <fullname>)"),
 
49
    ('e', 'email', "Email address"),
 
50
    ('s', 'studentid', "Student ID")
 
51
]
 
52
 
 
53
if len(sys.argv) <= 3:
 
54
    # Nicely format the usage message using the optionals
 
55
    print ("Usage: %s [OPTIONS] %s\n    OPTIONS"
 
56
        % (os.path.basename(sys.argv[0]),
 
57
           ' '.join(['<%s>' % x for x in requireds])))
 
58
    for short, long, desc in optionals:
 
59
        t = "        -%s | --%s" % (short, long)
 
60
        print t + (' ' * max(28 - len(t), 2)) + desc
 
61
    sys.exit(1)
 
62
 
 
63
shorts = ''.join([o[0] + ":" for o in optionals])
 
64
longs = [o[1] + "=" for o in optionals]
 
65
opts, args = getopt.gnu_getopt(sys.argv[1:], shorts, longs)
 
66
opts = dict(opts)
 
67
 
 
68
# Get the dictionary of fields from opts and args
 
69
user = {}
 
70
for i in range(0, len(requireds)):
 
71
    user[requireds[i]] = unicode(args[i])
 
72
for short, long, _ in optionals:
 
73
    try:
 
74
        user[long] = unicode(opts['-' + short])
 
75
    except KeyError:
 
76
        try:
 
77
            user[long] = unicode(opts['--' + long])
 
78
        except KeyError:
 
79
            pass
 
80
login = user['login']
 
81
if 'nick' not in user:
 
82
    user['nick'] = user['fullname']
 
83
 
 
84
store = get_store(Config())
 
85
 
 
86
try:
 
87
    # Make the user's database entry
 
88
    userobj = User(**user)
 
89
    store.add(userobj)
 
90
    enrol_user(store, userobj)
 
91
    store.commit()
 
92
except Exception, message:
 
93
    print "Error: " + str(message)
 
94
    sys.exit(1)
 
95
 
 
96
print "Successfully created user %s (%s)." % (login, user['fullname'])