← Back to branch summary

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

« back to all changes in this revision

Viewing changes to makeuser.py

  • Committer: dilshan_a
  • Date: 2008-01-23 04:54:10 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:270
Initial check in for test framework, and example problems

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
29
# Import modules from the website is tricky since they're in the www
35
30
# directory.
36
31
sys.path.append(os.path.join(os.getcwd(), 'www'))
37
32
import conf
38
33
import common.makeuser
39
34
 
40
 
if len(sys.argv) <= 5:
41
 
    print "Usage: python makeuser.py <username> <password> <nick> " \
42
 
        "<fullname> <rolenm> [<studentid>]"
43
 
    sys.exit()
44
 
 
45
 
if os.getuid() != 0:
46
 
    print "Must run makeuser.py as root."
 
35
if len(sys.argv) <= 1:
 
36
    print "Usage: python makeuser.py <username>"
47
37
    sys.exit()
48
38
 
49
39
username = sys.argv[1]
50
 
password = sys.argv[2]
51
 
nick = sys.argv[3]
52
 
fullname = sys.argv[4]
53
 
rolenm = sys.argv[5]
54
 
if len(sys.argv) > 6:
55
 
    studentid = sys.argv[6]
56
 
else:
57
 
    studentid = None
58
40
 
59
41
try:
60
 
    # Resolve the user's username into a UID
61
 
    # Create the user if it does not exist
62
 
    try:
63
 
        (_,_,uid,_,_,_,_) = pwd.getpwnam(username)
64
 
    except KeyError:
65
 
        if os.system("useradd '%s'" % username) != 0:
66
 
            raise Exception("Failed to add Unix user account")
67
 
        try:
68
 
            (_,_,uid,_,_,_,_) = pwd.getpwnam(username)
69
 
        except KeyError:
70
 
            raise Exception("Failed to add Unix user account")
71
 
    # Make the user's jail
72
 
    common.makeuser.make_jail(username, uid)
73
 
    # Make the user's database entry
74
 
    common.makeuser.make_user_db(username, uid, password, nick, fullname,
75
 
        rolenm, studentid)
 
42
    common.makeuser.makeuser(username)
76
43
except Exception, message:
77
44
    print "Error: " + str(message)
78
45
    sys.exit(1)
79
46
 
80
 
print "Successfully created user %s (%s)." % (username, fullname)
 
47
print "Successfully created user " + username + "."