← Back to branch summary

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

« back to all changes in this revision

Viewing changes to makeuser.py

  • Committer: Matt Giuca
  • Date: 2009-12-01 04:27:58 UTC
  • mfrom: (1164.2.46 sphinx-docs)
  • Revision ID: matt.giuca@gmail.com-20091201042758-wuxd9bdec00c283i
Merged sphinx-docs branch. This adds Sphinx documentation for the entire IVLE system (for system administrators and developers), and removes all of our random old document files (all either irrelevant, or moved into the Sphinx docs nicely). Currently incomplete, but ready to merge.

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 getopt
33
 
import common.makeuser
34
 
 
35
 
# Requireds and optionals will be used to display the usage message
36
 
# AND do argument processing
37
 
# The names here must correspond to the fields in the database.
38
 
requireds = ["login", "fullname", "rolenm"]
39
 
optionals = [
40
 
    ('p', 'password', "Cleartext password for this user"),
41
 
    ('n', 'nick', "Display name (defaults to <fullname>)"),
42
 
    ('e', 'email', "Email address"),
43
 
    ('s', 'studentid', "Student ID")
44
 
]
45
 
 
46
 
if len(sys.argv) <= 3:
47
 
    # Nicely format the usage message using the optionals
48
 
    print ("Usage: python makeuser.py [OPTIONS] %s\n    OPTIONS"
49
 
        % ' '.join(['<%s>' % x for x in requireds]))
50
 
    for short, long, desc in optionals:
51
 
        t = "        -%s | --%s" % (short, long)
52
 
        print t + (' ' * max(28 - len(t), 2)) + desc
53
 
    sys.exit(1)
54
 
 
55
 
if os.getuid() != 0:
56
 
    print "Must run makeuser.py as root."
57
 
    sys.exit(1)
58
 
 
59
 
shorts = ''.join([o[0] + ":" for o in optionals])
60
 
longs = [o[1] + "=" for o in optionals]
61
 
opts, args = getopt.gnu_getopt(sys.argv[1:], shorts, longs)
62
 
opts = dict(opts)
63
 
 
64
 
# Get the dictionary of fields from opts and args
65
 
user = {}
66
 
for i in range(0, len(requireds)):
67
 
    user[requireds[i]] = args[i]
68
 
for short, long, _ in optionals:
69
 
    try:
70
 
        user[long] = opts['-' + short]
71
 
    except KeyError:
72
 
        try:
73
 
            user[long] = opts['--' + long]
74
 
        except KeyError:
75
 
            pass
76
 
login = user['login']
77
 
if 'nick' not in user:
78
 
    user['nick'] = user['fullname']
79
 
 
80
 
try:
81
 
    # Make the user's database entry
82
 
    common.makeuser.make_user_db(**user)
83
 
except Exception, message:
84
 
    print "Error: " + str(message)
85
 
    sys.exit(1)
86
 
 
87
 
print "Successfully created user %s (%s)." % (login, user['fullname'])