← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-makeuser

Added module ivle.config, which takes care of some work interfacing with
    configobj, including searching for the file and opening the object.
ivle.conf.conf now uses this instead of having its own search.

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
 
 
34
from ivle.database import get_store, User
 
35
from ivle.pulldown_subj import enrol_user
 
36
 
 
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)
33
57
 
34
58
if os.getuid() != 0:
35
59
    print "Must run %s as root." % os.path.basename(sys.argv[0])
36
60
    sys.exit(1)
37
61
 
38
 
from ivle.config import Config
39
 
from ivle.database import get_store, User
40
 
from ivle.pulldown_subj import enrol_user
41
 
 
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
 
 
 
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'])