29
29
from storm.locals import Desc
30
from genshi.filters import HTMLFormFiller
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
36
from ivle.database import Subject, Semester, Offering, Enrolment, User
38
39
class SubjectsView(XHTMLView):
51
53
if enrolments.count():
52
54
ctx['semesters'].append((semester, enrolments))
57
class UserValidator(formencode.FancyValidator):
58
"""A FormEncode validator that turns a username into a user.
60
The state must have a 'store' attribute, which is the Storm store
62
def _to_python(self, value, state):
63
user = User.get_by_login(state.store, value)
67
raise formencode.Invalid('User does not exist', value, state)
70
class NoEnrolmentValidator(formencode.FancyValidator):
71
"""A FormEncode validator that ensures absence of an enrolment.
73
The state must have an 'offering' attribute.
75
def _to_python(self, value, state):
76
if state.offering.get_enrolment(value):
77
raise formencode.Invalid('User already enrolled', value, state)
81
class EnrolSchema(formencode.Schema):
82
user = formencode.All(NoEnrolmentValidator(), UserValidator())
85
class EnrolView(XHTMLView):
86
"""A form to enrol a user in an offering."""
87
template = 'enrol.html'
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()
103
def filter(self, stream, ctx):
104
return stream | HTMLFormFiller(data=ctx['data'])
106
def populate(self, req, ctx):
107
if req.method == 'POST':
108
data = dict(req.get_fieldstorage())
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'])
115
req.throw_redirect(req.uri)
116
except formencode.Invalid, e:
117
errors = e.unpack_errors()
122
ctx['data'] = data or {}
123
ctx['offering'] = self.context
124
ctx['errors'] = errors
54
127
class Plugin(ViewPlugin, MediaPlugin):
56
129
('subjects/', SubjectsView),
130
('subjects/:subject/:year/:semester/+enrolments/+new', EnrolView),