← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-adduser

  • Committer: Matt Giuca
  • Date: 2010-07-28 06:09:00 UTC
  • Revision ID: matt.giuca@gmail.com-20100728060900-6a0lcuexcv1juh5r
https://launchpad.net/bugs/526853
ivle/webapp/submit/submit.html: Rewrote error message when an offering could not be found to submit to. This can have one of several causes, and the old error assumed it was because you weren't in a subject dir. Now enumerates the possible reasons. (LP: #526853)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bin/ivle-adduser
29
29
 
30
30
import sys
31
31
import os
32
 
import getopt
33
 
 
 
32
import optparse
 
33
 
 
34
if os.getuid() != 0:
 
35
    print "Must run %s as root." % os.path.basename(sys.argv[0])
 
36
    sys.exit(1)
 
37
 
 
38
from ivle.config import Config
34
39
from ivle.database import get_store, User
35
40
from ivle.pulldown_subj import enrol_user
36
41
 
37
 
# Requireds and optionals will be used to display the usage message
38
 
# AND do argument processing
39
 
# The names here must correspond to the fields in the database.
40
 
requireds = ["login", "fullname", "rolenm"]
41
 
optionals = [
42
 
    ('p', 'password', "Cleartext password for this user"),
43
 
    ('n', 'nick', "Display name (defaults to <fullname>)"),
44
 
    ('e', 'email', "Email address"),
45
 
    ('s', 'studentid', "Student ID")
46
 
]
47
 
 
48
 
if len(sys.argv) <= 3:
49
 
    # Nicely format the usage message using the optionals
50
 
    print ("Usage: %s [OPTIONS] %s\n    OPTIONS"
51
 
        % (os.path.basename(sys.argv[0]),
52
 
           ' '.join(['<%s>' % x for x in requireds])))
53
 
    for short, long, desc in optionals:
54
 
        t = "        -%s | --%s" % (short, long)
55
 
        print t + (' ' * max(28 - len(t), 2)) + desc
56
 
    sys.exit(1)
57
 
 
58
 
if os.getuid() != 0:
59
 
    print "Must run %s as root." % os.path.basename(sys.argv[0])
60
 
    sys.exit(1)
61
 
 
62
 
shorts = ''.join([o[0] + ":" for o in optionals])
63
 
longs = [o[1] + "=" for o in optionals]
64
 
opts, args = getopt.gnu_getopt(sys.argv[1:], shorts, longs)
65
 
opts = dict(opts)
66
 
 
67
 
# Get the dictionary of fields from opts and args
68
 
user = {}
69
 
for i in range(0, len(requireds)):
70
 
    user[requireds[i]] = unicode(args[i])
71
 
for short, long, _ in optionals:
72
 
    try:
73
 
        user[long] = unicode(opts['-' + short])
74
 
    except KeyError:
75
 
        try:
76
 
            user[long] = unicode(opts['--' + long])
77
 
        except KeyError:
78
 
            pass
79
 
login = user['login']
 
42
usage = """usage: %prog [OPTIONS] <LOGIN> <FULLNAME>
 
43
Creates an IVLE user."""
 
44
 
 
45
parser = optparse.OptionParser(usage)
 
46
parser.add_option("-p", "--password",
 
47
    action="store", dest="password",
 
48
    help="Cleartext password"
 
49
    )
 
50
parser.add_option("-n", "--nick",
 
51
    action="store", dest="nick",
 
52
    help="Display name (defaults to <FULLNAME>)"
 
53
    )
 
54
parser.add_option("-e", "--email",
 
55
    action="store", dest="email",
 
56
    help="Email address"
 
57
    )
 
58
parser.add_option("-s", "--studentid",
 
59
    action="store", dest="studentid",
 
60
    help="Student ID"
 
61
    )
 
62
parser.add_option("--admin",
 
63
    action="store_true", dest="admin",
 
64
    help="Give the user full administrative privileges",
 
65
    default=False
 
66
    )
 
67
 
 
68
(options, args) = parser.parse_args()
 
69
 
 
70
if len(args) != 2:
 
71
    parser.error("incorrect number of arguments")
 
72
 
 
73
user = {'login': unicode(args[0]), 'fullname': unicode(args[1])}
 
74
 
 
75
for attr in ('password', 'nick', 'email', 'studentid', 'admin'):
 
76
    val = getattr(options, attr)
 
77
    if val is not None:
 
78
        if isinstance(val, str):
 
79
            val = unicode(val)
 
80
        user[attr] = val
 
81
 
80
82
if 'nick' not in user:
81
83
    user['nick'] = user['fullname']
82
84
 
83
 
store = get_store()
 
85
config = Config()
 
86
store = get_store(config)
84
87
 
85
 
try:
86
 
    # Make the user's database entry
87
 
    userobj = User(**user)
88
 
    store.add(userobj)
89
 
    enrol_user(store, userobj)
90
 
    store.commit()
91
 
except Exception, message:
92
 
    print "Error: " + str(message)
 
88
if User.get_by_login(store, user['login']) is not None:
 
89
    print "user '%s' already exists" % user['login']
93
90
    sys.exit(1)
94
91
 
95
 
print "Successfully created user %s (%s)." % (login, user['fullname'])
 
92
# Make the user's database entry
 
93
userobj = User(**user)
 
94
store.add(userobj)
 
95
enrol_user(config, store, userobj)
 
96
store.commit()
 
97
 
 
98
print "successfully created user '%s' (%s)." % (userobj.login, userobj.fullname)