← 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-01-29 23:52:19 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:329
Converted Console from an "app" into a "plugin". It can now be plugged in to
any app.
Added "plugins" directory in www. Added "console" plugin. This contains all of
the functionality of what was previously the console app, but modularized so
it can be imported by another app.

apps/console: Removed most of the logic (moved to plugins/console). Replaced
with a simple import of the console plugin. Should behave exactly the same.
apps/tutorial: As proof of concept, imported the console plugin. It now
appears at the bottom of the page (yet to make it have "pop up" behaviour).

Show diffs side-by-side

added added

removed removed

Lines of Context:
makeuser.py
22
22
 
23
23
# Script to create a new user. This can also be done through the
24
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
25
 
30
26
import sys
31
27
import os
32
28
import os.path
33
 
import pwd
34
 
import getopt
35
29
# Import modules from the website is tricky since they're in the www
36
30
# directory.
37
31
sys.path.append(os.path.join(os.getcwd(), 'www'))
38
32
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
 
dict = {}
72
 
for i in range(0, len(requireds)):
73
 
    dict[requireds[i]] = args[i]
74
 
for short, long, _ in optionals:
75
 
    try:
76
 
        dict[long] = opts['-' + short]
77
 
    except KeyError:
78
 
        try:
79
 
            dict[long] = opts['--' + long]
80
 
        except KeyError:
81
 
            pass
82
 
login = dict['login']
83
 
if 'nick' not in dict:
84
 
    dict['nick'] = dict['fullname']
 
33
import common.makeuser
 
34
 
 
35
if len(sys.argv) <= 1:
 
36
    print "Usage: python makeuser.py <username>"
 
37
    sys.exit()
 
38
 
 
39
username = sys.argv[1]
85
40
 
86
41
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
 
    dict['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(**dict)
 
42
    common.makeuser.makeuser(username)
103
43
except Exception, message:
104
44
    print "Error: " + str(message)
105
45
    sys.exit(1)
106
46
 
107
 
print "Successfully created user %s (%s)." % (login, dict['fullname'])
 
47
print "Successfully created user " + username + "."