← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-makeuser

  • Committer: Matt Giuca
  • Date: 2009-02-24 02:02:03 UTC
  • mto: This revision was merged to the branch mainline in revision 1119.
  • Revision ID: matt.giuca@gmail.com-20090224020203-aqdcjnsj6y7wl32o
Added a new look to the IVLE header bar. Mmmm... Web 2.0.
Added top-level directory image-source, containing SVG and script files for
    generating the images.
ivle/webapp/coremedia/images: Added 'chrome' directory containing the rendered
    images.
Modified ivle/webapp/base/ivle-headings.html and
    ivle/webapp/coremedia/ivle.css to support the new images.
    Note that the H1 and H2 at the top of the page are no longer displayed
    (and their custom styles have been removed). There is a heading image
    instead.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bin/ivle-makeuser
29
29
 
30
30
import sys
31
31
import os
32
 
import optparse
 
32
import getopt
33
33
 
34
34
if os.getuid() != 0:
35
35
    print "Must run %s as root." % os.path.basename(sys.argv[0])
36
36
    sys.exit(1)
37
37
 
38
 
from ivle.config import Config
39
38
from ivle.database import get_store, User
40
39
from ivle.pulldown_subj import enrol_user
41
40
 
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
 
 
 
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: %s [OPTIONS] %s\n    OPTIONS"
 
55
        % (os.path.basename(sys.argv[0]),
 
56
           ' '.join(['<%s>' % x for x in requireds])))
 
57
    for short, long, desc in optionals:
 
58
        t = "        -%s | --%s" % (short, long)
 
59
        print t + (' ' * max(28 - len(t), 2)) + desc
 
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']
82
80
if 'nick' not in user:
83
81
    user['nick'] = user['fullname']
84
82
 
85
 
config = Config()
86
 
store = get_store(config)
 
83
store = get_store()
87
84
 
88
 
if User.get_by_login(store, user['login']) is not None:
89
 
    print "user '%s' already exists" % user['login']
 
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)
90
93
    sys.exit(1)
91
94
 
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)
 
95
print "Successfully created user %s (%s)." % (login, user['fullname'])