← Back to branch summary

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

« back to all changes in this revision

Viewing changes to makeuser.py

  • Committer: mattgiuca
  • Date: 2008-02-28 07:52:35 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:608
browser/listing: REMOVED the generation of all action links to the right
    panel. (Yes, this is the end of an era).
    Split the "Modified" stamp in the right panel to split over 3 lines so it
    takes up less horizontal space.
    Shrunk the right panel from 300px down to 200px, as requested by Steven
    Bird.
We can now rely solely on the top dropdown for doing actions.

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 modules from the website is tricky since they're in the www
 
36
# directory.
 
37
sys.path.append(os.path.join(os.getcwd(), 'www'))
 
38
import conf
 
39
import common.makeuser, common.db
 
40
 
 
41
# Requireds and optionals will be used to display the usage message
 
42
# AND do argument processing
 
43
# The names here must correspond to the fields in the database.
 
44
requireds = ["login", "fullname", "rolenm"]
 
45
optionals = [
 
46
    ('p', 'password', "Cleartext password for this user"),
 
47
    ('n', 'nick', "Display name (defaults to <fullname>)"),
 
48
    ('e', 'email', "Email address"),
 
49
    ('s', 'studentid', "Student ID")
 
50
]
 
51
 
 
52
if len(sys.argv) <= 3:
 
53
    # Nicely format the usage message using the optionals
 
54
    print ("Usage: python makeuser.py [OPTIONS] %s\n    OPTIONS"
 
55
        % ' '.join(['<%s>' % x for x in requireds]))
 
56
    for short, long, desc in optionals:
 
57
        t = "        -%s | --%s" % (short, long)
 
58
        print t + (' ' * max(28 - len(t), 2)) + desc
 
59
    sys.exit(1)
 
60
 
 
61
if os.getuid() != 0:
 
62
    print "Must run makeuser.py as root."
 
63
    sys.exit(1)
 
64
 
 
65
shorts = ''.join([o[0] + ":" for o in optionals])
 
66
longs = [o[1] + "=" for o in optionals]
 
67
opts, args = getopt.gnu_getopt(sys.argv[1:], shorts, longs)
 
68
opts = dict(opts)
 
69
 
 
70
# Get the dictionary of fields from opts and args
 
71
user = {}
 
72
for i in range(0, len(requireds)):
 
73
    user[requireds[i]] = args[i]
 
74
for short, long, _ in optionals:
 
75
    try:
 
76
        user[long] = opts['-' + short]
 
77
    except KeyError:
 
78
        try:
 
79
            user[long] = opts['--' + long]
 
80
        except KeyError:
 
81
            pass
 
82
login = user['login']
 
83
if 'nick' not in user:
 
84
    user['nick'] = user['fullname']
 
85
 
 
86
try:
 
87
    # Resolve the user's username into a UID
 
88
    # Create the user if it does not exist
 
89
    try:
 
90
        (_,_,uid,_,_,_,_) = pwd.getpwnam(login)
 
91
    except KeyError:
 
92
        if os.system("useradd '%s'" % login) != 0:
 
93
            raise Exception("Failed to add Unix user account")
 
94
        try:
 
95
            (_,_,uid,_,_,_,_) = pwd.getpwnam(login)
 
96
        except KeyError:
 
97
            raise Exception("Failed to add Unix user account")
 
98
    user['unixid'] = uid
 
99
    # Make the user's database entry
 
100
    common.makeuser.make_user_db(**user)
 
101
except Exception, message:
 
102
    print "Error: " + str(message)
 
103
    sys.exit(1)
 
104
 
 
105
print "Successfully created user %s (%s)." % (login, user['fullname'])