← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-enrol

Merged from trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
bin/ivle-enrol
25
25
import os
26
26
import sys
27
27
 
28
 
from ivle.database import get_store, Subject, Semester, Offering, User, \
29
 
                          AlreadyEnrolledError
 
28
from ivle.database import get_store, Subject, Semester, Offering, User
30
29
 
31
30
def die(error):
32
31
    print >> sys.stderr, '%s: %s' % (os.path.basename(sys.argv[0]), error)
33
32
    sys.exit(1)
34
33
 
35
 
if len(sys.argv) != 5:
36
 
    print >> sys.stderr, "usage: %s <login> <subjectcode> <year> <semester>" \
 
34
if len(sys.argv) not in (5, 6):
 
35
    print >> sys.stderr, '''usage: %s <login> <subjectcode> <year> <semester> [role]
 
36
 
 
37
'role' should be one of student (the default), tutor or lecturer.
 
38
If an enrolment already exists, it will be reactivated and its role set.''' \
37
39
                         % os.path.basename(sys.argv[0])
38
40
    sys.exit()
39
41
 
52
54
if not offering:
53
55
    die("subject offering not found")
54
56
 
55
 
if not offering.semester.active:
56
 
    die("cannot enrol - semester inactive")
57
 
    
 
57
role = unicode(sys.argv[5]) if len(sys.argv) == 6 else u'student'
 
58
 
58
59
try:
59
 
    offering.enrol(user)
 
60
    offering.enrol(user, role)
60
61
except AlreadyEnrolledError:
61
62
    die("already enrolled")
62
63