← Back to branch summary

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

« back to all changes in this revision

Viewing changes to bin/ivle-enrol

  • Committer: me at id
  • Date: 2009-01-15 01:06:33 UTC
  • mto: This revision was merged to the branch mainline in revision 1090.
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:branches%2Fstorm:1144
ivle.auth.* now uses Storm, and not ivle.(db|user). This changes the interface
    somewhat (lots of things take a 'store' argument), so fix callers.

ivle.db: Remove user_authenticate; it's now a method of ivle.database.User.

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
 
 
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.''' \
 
28
import ivle.db
 
29
 
 
30
if len(sys.argv) != 5:
 
31
    print >> sys.stderr, "usage: %s <login> <subjectcode> <semester> <year>" \
39
32
                         % os.path.basename(sys.argv[0])
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()
 
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)