← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-enrol

ivle.studpath.url_to_jailpaths: Fix the doctest to use new paths.

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