← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-enrol

  • Committer: William Grant
  • Date: 2009-04-28 04:50:39 UTC
  • Revision ID: grantw@unimelb.edu.au-20090428045039-ibb7gwtjrhe9osq3
Populate req.config in a cleaner manner.

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
 
import ivle.db
29
 
 
30
 
if len(sys.argv) != 5:
31
 
    print >> sys.stderr, "usage: %s <login> <subjectcode> <semester> <year>" \
 
28
from ivle.database import get_store, Subject, Semester, Offering, User
 
29
 
 
30
def die(error):
 
31
    print >> sys.stderr, '%s: %s' % (os.path.basename(sys.argv[0]), error)
 
32
    sys.exit(1)
 
33
 
 
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.''' \
32
39
                         % os.path.basename(sys.argv[0])
33
 
    sys.exit(1)
34
 
 
35
 
if not ivle.db.DB().add_enrolment(*sys.argv[1:]):
36
 
    print >> sys.stderr, "enrolment failed!"
37
 
    sys.exit(1)
 
40
    sys.exit()
 
41
 
 
42
store = get_store()
 
43
 
 
44
user = User.get_by_login(store, sys.argv[1])
 
45
if user is None:
 
46
    die("user not found")
 
47
 
 
48
offering = store.find(Offering,
 
49
                    Subject.code == unicode(sys.argv[2]),
 
50
                    Semester.year == unicode(sys.argv[3]),
 
51
                    Semester.semester == unicode(sys.argv[4]),
 
52
                    Offering.subject_id == Subject.id,
 
53
                    Offering.semester_id == Semester.id).one()
 
54
if not offering:
 
55
    die("subject offering not found")
 
56
 
 
57
role = unicode(sys.argv[5]) if len(sys.argv) == 6 else u'student'
 
58
 
 
59
try:
 
60
    offering.enrol(user, role)
 
61
except AlreadyEnrolledError:
 
62
    die("already enrolled")
 
63
 
 
64
store.commit()