← Back to branch summary

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

« back to all changes in this revision

Viewing changes to makeuser.py

  • Committer: William Grant
  • Date: 2009-02-23 23:47:02 UTC
  • mfrom: (1099.1.211 new-dispatch)
  • Revision ID: grantw@unimelb.edu.au-20090223234702-db4b1llly46ignwo
Merge from lp:~ivle-dev/ivle/new-dispatch.

Pretty much everything changes. Reread the setup docs. Backup your databases.
Every file is now in a different installed location, the configuration system
is rewritten, the dispatch system is rewritten, URLs are different, the
database is different, worksheets and exercises are no longer on the
filesystem, we use a templating engine, jail service protocols are rewritten,
we don't repeat ourselves, we have authorization rewritten, phpBB is gone,
and probably lots of other things that I cannot remember.

This is certainly the biggest commit I have ever made, and hopefully
the largest I ever will.

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'])