← 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: Nick Chadwick
  • Date: 2009-03-09 00:45:58 UTC
  • mfrom: (1156 trunk)
  • mto: This revision was merged to the branch mainline in revision 1162.
  • Revision ID: chadnickbok@gmail.com-20090309004558-xvvj990b9h9y23bu
Merged from trunk.

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