← 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-19 00:54:28 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:500
db: get_user and get_users now return User objects instead of dictionaries.
    This is the major part of replacing dicts with User objects, as it
    propagates upwards.

Propagated User objects up through the following modules:
    listusers.py, dispatch.login, authenticate, userservice, forumutil
All of these now treat users as an object rather than a dict.

To save on the size of the changes so far, login still individually copies
fields over to the session (so the session does not yet store a user object;
that is the second part of this refactor).

WOO!! Revision 500 :)

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 jail
 
100
    common.makeuser.make_jail(login, uid)
 
101
    # Make the user's database entry
 
102
    common.makeuser.make_user_db(**user)
 
103
except Exception, message:
 
104
    print "Error: " + str(message)
 
105
    sys.exit(1)
 
106
 
 
107
print "Successfully created user %s (%s)." % (login, user['fullname'])