← 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

Fix circular import in ivle.zip.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/webapp/admin/subject.py
26
26
import urllib
27
27
import cgi
28
28
 
29
 
from storm.locals import Desc
30
 
from genshi.filters import HTMLFormFiller
31
 
import formencode
32
 
 
33
29
from ivle.webapp.base.xhtml import XHTMLView
34
30
from ivle.webapp.base.plugins import ViewPlugin, MediaPlugin
35
31
from ivle.webapp.errors import NotFound
36
 
from ivle.database import Subject, Semester, Offering, Enrolment, User
 
32
from ivle.database import Subject
 
33
from ivle import util
37
34
 
38
35
 
39
36
class SubjectsView(XHTMLView):
40
37
    '''The view of the list of subjects.'''
41
38
    template = 'subjects.html'
42
 
    tab = 'subjects'
 
39
    appname = 'subjects' # XXX
43
40
 
44
41
    def authorize(self, req):
45
42
        return req.user is not None
46
43
 
47
44
    def populate(self, req, ctx):
48
 
        ctx['user'] = req.user
49
 
        ctx['semesters'] = []
50
 
        for semester in req.store.find(Semester).order_by(Desc(Semester.year),
51
 
                                                     Desc(Semester.semester)):
52
 
            enrolments = semester.enrolments.find(user=req.user)
53
 
            if enrolments.count():
54
 
                ctx['semesters'].append((semester, enrolments))
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
 
45
        enrolled_subjects = req.user.subjects
 
46
        unenrolled_subjects = [subject for subject in
 
47
                               req.store.find(Subject)
 
48
                               if subject not in enrolled_subjects]
 
49
 
 
50
        ctx['enrolled_subjects'] = []
 
51
        ctx['other_subjects'] = []
 
52
 
 
53
        req.content_type = "text/html"
 
54
        req.write_html_head_foot = True
 
55
 
 
56
        for subject in enrolled_subjects:
 
57
            new_subj = {}
 
58
            new_subj['name'] = subject.name
 
59
            new_subj['url'] = subject.url
 
60
            ctx['enrolled_subjects'].append(new_subj)
 
61
 
 
62
        if len(unenrolled_subjects) > 0:
 
63
            for subject in unenrolled_subjects:
 
64
                new_subj = {}
 
65
                new_subj['name'] = subject.name
 
66
                new_subj['url'] = subject.url
 
67
                ctx['other_subjects'].append(new_subj)
125
68
 
126
69
 
127
70
class Plugin(ViewPlugin, MediaPlugin):
128
71
    urls = [
129
72
        ('subjects/', SubjectsView),
130
 
        ('subjects/:subject/:year/:semester/+enrolments/+new', EnrolView),
131
73
    ]
132
74
 
133
75
    tabs = [
134
 
        ('subjects', 'Subjects',
135
 
         'View subject content and complete worksheets',
136
 
         'subjects.png', 'subjects', 5)
 
76
        ('subjects', 'Subjects', 'Announcements and information about the '
 
77
         'subjects you are enrolled in.', 'subjects.png', 'subjects', 5)
137
78
    ]
138
79
 
139
80
    media = 'subject-media'