← 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

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
32
30
 
33
31
from ivle.webapp.base.xhtml import XHTMLView
34
32
from ivle.webapp.base.plugins import ViewPlugin, MediaPlugin
35
33
from ivle.webapp.errors import NotFound
36
 
from ivle.database import Subject, Semester, Offering, Enrolment, User
 
34
from ivle.database import Subject, Semester
37
35
from ivle import util
38
36
 
39
37
 
46
44
        return req.user is not None
47
45
 
48
46
    def populate(self, req, ctx):
49
 
        ctx['user'] = req.user
50
47
        ctx['semesters'] = []
51
48
        for semester in req.store.find(Semester).order_by(Desc(Semester.year),
52
49
                                                     Desc(Semester.semester)):
54
51
            if enrolments.count():
55
52
                ctx['semesters'].append((semester, enrolments))
56
53
 
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
 
 
128
54
class Plugin(ViewPlugin, MediaPlugin):
129
55
    urls = [
130
56
        ('subjects/', SubjectsView),
131
 
        ('subjects/:subject/:year/:semester/+enrolments/+new', EnrolView),
132
57
    ]
133
58
 
134
59
    tabs = [