← Back to branch summary

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

« back to all changes in this revision

Viewing changes to ivle/webapp/admin/subject.py

  • Committer: William Grant
  • Date: 2009-04-28 08:17:26 UTC
  • Revision ID: grantw@unimelb.edu.au-20090428081726-7h8fyjclxtydjma0
No more ivle.conf in ivle.webapp.fileservice...

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/webapp/admin/subject.py
27
27
import cgi
28
28
 
29
29
from storm.locals import Desc
 
30
from genshi.filters import HTMLFormFiller
 
31
import formencode
30
32
 
31
33
from ivle.webapp.base.xhtml import XHTMLView
32
34
from ivle.webapp.base.plugins import ViewPlugin, MediaPlugin
33
35
from ivle.webapp.errors import NotFound
34
 
from ivle.database import Subject, Semester
35
 
from ivle import util
 
36
from ivle.database import Subject, Semester, Offering, Enrolment, User
36
37
 
37
38
 
38
39
class SubjectsView(XHTMLView):
44
45
        return req.user is not None
45
46
 
46
47
    def populate(self, req, ctx):
 
48
        ctx['user'] = req.user
47
49
        ctx['semesters'] = []
48
50
        for semester in req.store.find(Semester).order_by(Desc(Semester.year),
49
51
                                                     Desc(Semester.semester)):
51
53
            if enrolments.count():
52
54
                ctx['semesters'].append((semester, enrolments))
53
55
 
 
56
 
 
57
class UserValidator(formencode.FancyValidator):
 
58
    """A FormEncode validator that turns a username into a user.
 
59
 
 
60
    The state must have a 'store' attribute, which is the Storm store
 
61
    to use."""
 
62
    def _to_python(self, value, state):
 
63
        user = User.get_by_login(state.store, value)
 
64
        if user:
 
65
            return user
 
66
        else:
 
67
            raise formencode.Invalid('User does not exist', value, state)
 
68
 
 
69
 
 
70
class NoEnrolmentValidator(formencode.FancyValidator):
 
71
    """A FormEncode validator that ensures absence of an enrolment.
 
72
 
 
73
    The state must have an 'offering' attribute.
 
74
    """
 
75
    def _to_python(self, value, state):
 
76
        if state.offering.get_enrolment(value):
 
77
            raise formencode.Invalid('User already enrolled', value, state)
 
78
        return value
 
79
 
 
80
 
 
81
class EnrolSchema(formencode.Schema):
 
82
    user = formencode.All(NoEnrolmentValidator(), UserValidator())
 
83
 
 
84
 
 
85
class EnrolView(XHTMLView):
 
86
    """A form to enrol a user in an offering."""
 
87
    template = 'enrol.html'
 
88
    tab = 'subjects'
 
89
    permission = 'edit'
 
90
 
 
91
    def __init__(self, req, subject, year, semester):
 
92
        """Find the given offering by subject, year and semester."""
 
93
        self.context = req.store.find(Offering,
 
94
            Offering.subject_id == Subject.id,
 
95
            Subject.short_name == subject,
 
96
            Offering.semester_id == Semester.id,
 
97
            Semester.year == year,
 
98
            Semester.semester == semester).one()
 
99
 
 
100
        if not self.context:
 
101
            raise NotFound()
 
102
 
 
103
    def filter(self, stream, ctx):
 
104
        return stream | HTMLFormFiller(data=ctx['data'])
 
105
 
 
106
    def populate(self, req, ctx):
 
107
        if req.method == 'POST':
 
108
            data = dict(req.get_fieldstorage())
 
109
            try:
 
110
                validator = EnrolSchema()
 
111
                req.offering = self.context # XXX: Getting into state.
 
112
                data = validator.to_python(data, state=req)
 
113
                self.context.enrol(data['user'])
 
114
                req.store.commit()
 
115
                req.throw_redirect(req.uri)
 
116
            except formencode.Invalid, e:
 
117
                errors = e.unpack_errors()
 
118
        else:
 
119
            data = {}
 
120
            errors = {}
 
121
 
 
122
        ctx['data'] = data or {}
 
123
        ctx['offering'] = self.context
 
124
        ctx['errors'] = errors
 
125
 
 
126
 
54
127
class Plugin(ViewPlugin, MediaPlugin):
55
128
    urls = [
56
129
        ('subjects/', SubjectsView),
 
130
        ('subjects/:subject/:year/:semester/+enrolments/+new', EnrolView),
57
131
    ]
58
132
 
59
133
    tabs = [