← 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 05:06:00 UTC
  • Revision ID: grantw@unimelb.edu.au-20090428050600-hogd9d6wo7ksyqy8
ivle.database.get_store() now takes a configuration object.

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