← Back to branch summary

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

« back to all changes in this revision

Viewing changes to makeuser.py

Added module ivle.config, which takes care of some work interfacing with
    configobj, including searching for the file and opening the object.
ivle.conf.conf now uses this instead of having its own search.

Show diffs side-by-side

added added

removed removed

Lines of Context:
makeuser.py
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 os.path
33
 
import pwd
34
 
import getopt
35
 
import random
36
 
# Import modules from the website is tricky since they're in the www
37
 
# directory.
38
 
sys.path.append(os.path.join(os.getcwd(), 'www'))
39
 
import conf
40
 
import common.makeuser, common.db
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", "rolenm"]
46
 
optionals = [
47
 
    ('p', 'password', "Cleartext password for this user"),
48
 
    ('u', 'unixid', "Numeric unix user id"),
49
 
    ('n', 'nick', "Display name (defaults to <fullname>)"),
50
 
    ('e', 'email', "Email address"),
51
 
    ('s', 'studentid', "Student ID")
52
 
]
53
 
 
54
 
if len(sys.argv) <= 3:
55
 
    # Nicely format the usage message using the optionals
56
 
    print ("Usage: python makeuser.py [OPTIONS] %s\n    OPTIONS"
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
 
if os.getuid() != 0:
64
 
    print "Must run makeuser.py as root."
65
 
    sys.exit(1)
66
 
 
67
 
shorts = ''.join([o[0] + ":" for o in optionals])
68
 
longs = [o[1] + "=" for o in optionals]
69
 
opts, args = getopt.gnu_getopt(sys.argv[1:], shorts, longs)
70
 
opts = dict(opts)
71
 
 
72
 
# Get the dictionary of fields from opts and args
73
 
user = {}
74
 
for i in range(0, len(requireds)):
75
 
    user[requireds[i]] = args[i]
76
 
for short, long, _ in optionals:
77
 
    try:
78
 
        user[long] = opts['-' + short]
79
 
    except KeyError:
80
 
        try:
81
 
            user[long] = opts['--' + long]
82
 
        except KeyError:
83
 
            pass
84
 
login = user['login']
85
 
if 'nick' not in user:
86
 
    user['nick'] = user['fullname']
87
 
 
88
 
# Resolve the user's username into a UID
89
 
if 'unixid' not in user:
90
 
    try:
91
 
        (_,_,uid,_,_,_,_) = pwd.getpwnam(login)
92
 
        user['unixid'] = uid
93
 
    except KeyError:
94
 
        user['unixid'] = random.randrange(5000,10000)
95
 
 
96
 
try:
97
 
    # Make the user's database entry
98
 
    common.makeuser.make_user_db(**user)
99
 
except Exception, message:
100
 
    print "Error: " + str(message)
101
 
    sys.exit(1)
102
 
 
103
 
print "Successfully created user %s (%s)." % (login, user['fullname'])