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

« back to all changes in this revision

Viewing changes to ivle/webapp/base/forms.py

  • Committer: William Grant
  • Date: 2010-02-25 07:34:50 UTC
  • Revision ID: grantw@unimelb.edu.au-20100225073450-zcl8ev5hlyhbszeu
Activate the Storm C extensions if possible. Moar speed.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# IVLE
 
2
# Copyright (C) 2010 The University of Melbourne
 
3
#
 
4
# This program is free software; you can redistribute it and/or modify
 
5
# it under the terms of the GNU General Public License as published by
 
6
# the Free Software Foundation; either version 2 of the License, or
 
7
# (at your option) any later version.
 
8
#
 
9
# This program is distributed in the hope that it will be useful,
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
# GNU General Public License for more details.
 
13
#
 
14
# You should have received a copy of the GNU General Public License
 
15
# along with this program; if not, write to the Free Software
 
16
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
17
 
 
18
import re
 
19
import datetime
 
20
 
 
21
import formencode
 
22
import formencode.validators
 
23
from genshi.filters import HTMLFormFiller
 
24
 
 
25
from ivle.webapp.base.xhtml import XHTMLView
 
26
 
 
27
 
 
28
class BaseFormView(XHTMLView):
 
29
    """A base form view."""
 
30
 
 
31
    @property
 
32
    def validator(self):
 
33
        """The FormEncode validator to use.
 
34
 
 
35
        The request will be passed in as state, after potentially being
 
36
        modified by populate_state().
 
37
        """
 
38
        raise NotImplementedError()
 
39
 
 
40
    def populate_state(self, state):
 
41
        """Populate the state given to the FormEncode validator.
 
42
 
 
43
        Subclasses can override this and set additional attributes.
 
44
        """
 
45
        pass
 
46
 
 
47
    def get_return_url(self, obj):
 
48
        """Return the URL to which the completed form should redirect.
 
49
 
 
50
        By default this will redirect to the saved object.
 
51
        """
 
52
        return self.req.publisher.generate(obj)
 
53
 
 
54
    def get_default_data(self, req):
 
55
        """Return a dict mapping field names to default form values.
 
56
 
 
57
        For an edit form, this should return the object's existing data.
 
58
        For a creation form, this should probably return an empty dict.
 
59
 
 
60
        This must be overridden by subclasses.
 
61
        """
 
62
        raise NotImplementedError()
 
63
 
 
64
    def save_object(self, req, data):
 
65
        """Take the validated form data and turn it into an object.
 
66
 
 
67
        The object must then be returned.
 
68
 
 
69
        For an edit form, this should just overwrite data on an existing
 
70
        object.
 
71
        For a creation form, this should create a new object with the given
 
72
        data and add it to the request's store.
 
73
        """
 
74
        raise NotImplementedError()
 
75
 
 
76
    def filter(self, stream, ctx):
 
77
        return stream | HTMLFormFiller(data=ctx['data'])
 
78
 
 
79
    def populate(self, req, ctx):
 
80
        if req.method == 'POST':
 
81
            data = dict(req.get_fieldstorage())
 
82
            try:
 
83
                self.populate_state(req)
 
84
                data = self.validator.to_python(data, state=req)
 
85
 
 
86
                obj = self.save_object(req, data)
 
87
 
 
88
                req.store.commit()
 
89
                req.throw_redirect(self.get_return_url(obj))
 
90
            except formencode.Invalid, e:
 
91
                error_value = e.msg
 
92
                errors = e.unpack_errors()
 
93
        else:
 
94
            data = self.get_default_data(req)
 
95
            error_value = None
 
96
            errors = {}
 
97
 
 
98
        if errors:
 
99
            req.store.rollback()
 
100
 
 
101
        ctx['req'] = req
 
102
        ctx['context'] = self.context
 
103
        ctx['data'] = data or {}
 
104
        ctx['errors'] = errors
 
105
        # If all of the fields validated, set the global form error.
 
106
        if isinstance(errors, basestring):
 
107
            ctx['error_value'] = errors
 
108
 
 
109
 
 
110
VALID_URL_NAME = re.compile(r'^[a-z0-9][a-z0-9_\+\.\-]*$')
 
111
 
 
112
 
 
113
class URLNameValidator(formencode.validators.UnicodeString):
 
114
    def validate_python(self, value, state):
 
115
        super(URLNameValidator, self).validate_python(value, state)
 
116
        if not VALID_URL_NAME.match(value):
 
117
            raise formencode.Invalid(
 
118
                'Must consist of a lowercase alphanumeric character followed '
 
119
                'by any number of lowercase alphanumerics, ., +, - or _.',
 
120
                value, state)
 
121
 
 
122
class DateTimeValidator(formencode.validators.FancyValidator):
 
123
    """Accepts a date/time in YYYY-MM-DD HH:MM:SS format. Converts to a
 
124
    datetime.datetime object."""
 
125
    def _to_python(self, value, state):
 
126
        """Validate and convert."""
 
127
        try:
 
128
            return datetime.datetime.strptime(value, "%Y-%m-%d %H:%M:%S")
 
129
        except ValueError, e:
 
130
            raise formencode.Invalid(str(e) + " -> " + repr(value), value, state)
 
131
            raise formencode.Invalid("Must be a timestamp in "
 
132
                "YYYY-MM-DD HH:MM:SS format", value, state)
 
133
    def _from_python(self, value, state):
 
134
        try:
 
135
            return value.strftime("%Y-%m-%d %H:%M:%S")
 
136
        except AttributeError:
 
137
            raise formencode.Invalid("Must be a datetime.datetime object")