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

« back to all changes in this revision

Viewing changes to ivle/webapp/media.py

  • Committer: William Grant
  • Date: 2009-05-14 23:02:31 UTC
  • Revision ID: grantw@unimelb.edu.au-20090514230231-r1bhknfxmkf5o44i
Unbreak serve - it was depending on the old names of a module.

This was missed because the .pyc was left around on dev systems...

Show diffs side-by-side

added added

removed removed

Lines of Context:
28
28
from ivle.webapp.base.views import BaseView
29
29
from ivle.webapp.base.plugins import PublicViewPlugin, ViewPlugin, MediaPlugin
30
30
from ivle.webapp.errors import NotFound, Forbidden
31
 
from ivle.webapp.publisher import INF
32
 
from ivle.webapp import ApplicationRoot
33
 
 
34
 
# This maps a media namespace to an external dependency directory (in this
35
 
# case specified by the configuration option media/externals/jquery) and a
36
 
# list of permitted subpaths.
37
 
EXTERNAL_MEDIA_MAP = {'jquery': ('jquery', ['jquery.js'])}
38
31
 
39
32
def media_url(req, plugin, path):
40
33
    '''Generates a URL to a media file.
53
46
 
54
47
    return req.make_path(os.path.join(media_path, plugin, path))
55
48
 
56
 
class MediaFile(object):
57
 
    def __init__(self, root, external, version, ns, path):
58
 
        self.root = root
59
 
        self.external = external
60
 
        self.version = version
 
49
class BaseMediaFileView(BaseView):
 
50
    '''A view for media files.
 
51
 
 
52
    This serves static files from directories registered by plugins.
 
53
 
 
54
    Plugins wishing to export media should declare a 'media' attribute,
 
55
    pointing to the directory to serve (relative to the module's directory).
 
56
    The contents of that directory will then be available under
 
57
    /+media/python.path.to.module.
 
58
    '''
 
59
    def __init__(self, req, ns, path):
61
60
        self.ns = ns
62
61
        self.path = path
63
62
 
64
 
    @property
65
 
    def filename(self):
66
 
        if self.external:
67
 
            try:
68
 
                extern = EXTERNAL_MEDIA_MAP[self.ns]
69
 
            except KeyError:
70
 
                return None
71
 
 
72
 
            # Unless it's a whitelisted path, we don't want to hear about it.
73
 
            # If the whitelist is None (not []), we allow all.
74
 
            if extern[1] is not None and self.path not in extern[1]:
75
 
                return None
76
 
 
77
 
            # Grab the admin-configured path for this particular external dep.
78
 
            externdir = self.root.config['media']['externals'][extern[0]]
79
 
 
80
 
            assert isinstance(externdir, basestring)
81
 
 
82
 
            return os.path.join(externdir, self.path)
83
 
        else:
84
 
            try:
85
 
                plugin = self.root.config.plugins[self.ns]
86
 
            except KeyError:
87
 
                return None
88
 
 
89
 
            if not issubclass(plugin, MediaPlugin):
90
 
                return None
91
 
 
92
 
            mediadir = plugin.media
93
 
            plugindir = os.path.dirname(inspect.getmodule(plugin).__file__)
94
 
 
95
 
            return os.path.join(plugindir, mediadir, self.path)
96
 
 
97
 
class MediaFileView(BaseView):
98
 
    def authorize(self, req):
99
 
        return True
 
63
    def _make_filename(self, req):
 
64
        raise NotImplementedError()
100
65
 
101
66
    def render(self, req):
102
67
        # If it begins with ".." or separator, it's illegal. Die.
103
 
        if self.context.path.startswith("..") or \
104
 
           self.context.path.startswith('/'):
 
68
        if self.path.startswith("..") or self.path.startswith('/'):
105
69
            raise Forbidden()
106
70
 
107
 
        filename = self.get_filename(req)
108
 
        if filename is None:
109
 
            raise NotFound()
 
71
        filename = self._make_filename(req)
110
72
 
111
73
        # Find an appropriate MIME type.
112
74
        (type, _) = mimetypes.guess_type(filename)
119
81
        if not os.access(filename, os.R_OK) or os.path.isdir(filename):
120
82
            raise Forbidden()
121
83
 
122
 
        if self.context.version is not None:
123
 
            req.headers_out['Expires'] = email.utils.formatdate(
124
 
                                timeval=time.time() + (60*60*24*365),
125
 
                                localtime=False,
126
 
                                usegmt=True)
127
 
 
128
 
 
129
84
        req.content_type = type
130
85
        req.sendfile(filename)
131
86
 
132
 
    def get_filename(self, req):
133
 
        return self.context.filename
134
 
 
135
 
def root_to_media(root, *segments):
136
 
    if segments[0].startswith('+'):
137
 
        if segments[0] == '+external':
138
 
            external = True
139
 
            version = None
140
 
            path = segments[1:]
141
 
        else:
142
 
            version = segments[0][1:]
143
 
            if segments[1] == '+external':
144
 
                external = True
145
 
                path = segments[2:]
146
 
            else:
147
 
                external = False
148
 
                path = segments[1:]
149
 
    else:
150
 
        external = False
151
 
        version = None
152
 
        path = segments
153
 
 
154
 
    if version is not None and version != root.config['media']['version']:
155
 
        return None
156
 
 
157
 
    ns = path[0]
158
 
    path = os.path.normpath(os.path.join(*path[1:]))
159
 
 
160
 
    return MediaFile(root, external, version, ns, path)
 
87
 
 
88
class MediaFileView(BaseMediaFileView):
 
89
    '''A view for media files.
 
90
 
 
91
    This serves static files from directories registered by plugins.
 
92
 
 
93
    Plugins wishing to export media should declare a 'media' attribute,
 
94
    pointing to the directory to serve (relative to the module's directory).
 
95
    The contents of that directory will then be available under
 
96
    /+media/python.path.to.module.
 
97
    '''
 
98
    permission = None
 
99
 
 
100
    def _make_filename(self, req):
 
101
        try:
 
102
            plugin = req.config.plugins[self.ns]
 
103
        except KeyError:
 
104
            raise NotFound()
 
105
 
 
106
        if not issubclass(plugin, MediaPlugin):
 
107
            raise NotFound()
 
108
 
 
109
        mediadir = plugin.media
 
110
        plugindir = os.path.dirname(inspect.getmodule(plugin).__file__)
 
111
 
 
112
        return os.path.join(plugindir, mediadir, self.path)
 
113
 
 
114
    def get_permissions(self, user):
 
115
        return set()
 
116
 
 
117
class VersionedMediaFileView(MediaFileView):
 
118
    '''A view for versioned media files, with aggressive caching.
 
119
 
 
120
    This serves static media files with a version string, and requests that
 
121
    browsers cache them for a long time.
 
122
    '''
 
123
 
 
124
    def __init__(self, req, ns, path, version):
 
125
        super(VersionedMediaFileView, self).__init__(req, ns, path)
 
126
        self.version = version
 
127
 
 
128
    def _make_filename(self, req):
 
129
        if self.version != req.config['media']['version']:
 
130
            raise NotFound()
 
131
 
 
132
        # Don't expire for a year.
 
133
        req.headers_out['Expires'] = email.utils.formatdate(
 
134
                                    timeval=time.time() + (60*60*24*365),
 
135
                                    localtime=False,
 
136
                                    usegmt=True)
 
137
        return super(VersionedMediaFileView, self)._make_filename(req)
 
138
 
 
139
 
 
140
# This maps a media namespace to an external dependency directory (in this
 
141
# case specified by the configuration option media/externals/jquery) and a
 
142
# list of permitted subpaths.
 
143
EXTERNAL_MEDIA_MAP = {'jquery': ('jquery', ['jquery.js'])}
 
144
 
 
145
class ExternalMediaFileView(BaseMediaFileView):
 
146
    '''A view for media files from external dependencies.
 
147
 
 
148
    This serves specific static files from external dependencies as defined in
 
149
    the IVLE configuration.
 
150
    '''
 
151
    permission = None
 
152
 
 
153
    def _make_filename(self, req):
 
154
        try:
 
155
            extern = EXTERNAL_MEDIA_MAP[self.ns]
 
156
        except KeyError:
 
157
            raise NotFound()
 
158
 
 
159
        # Unless it's a whitelisted path, we don't want to hear about it.
 
160
        if self.path not in extern[1]:
 
161
            raise NotFound()
 
162
 
 
163
        # Grab the admin-configured path for this particular external dep.
 
164
        externdir = req.config['media']['externals'][extern[0]]
 
165
 
 
166
        assert isinstance(externdir, basestring)
 
167
 
 
168
        return os.path.join(externdir, self.path)
 
169
 
 
170
    def get_permissions(self, user):
 
171
        return set()
 
172
 
 
173
class ExternalVersionedMediaFileView(ExternalMediaFileView):
 
174
    '''A view for versioned media files from external dependencies, with
 
175
    aggressive caching.
 
176
 
 
177
    This serves specific static media files from external dependencies with a
 
178
    version string, and requests that browsers cache them for a long time.
 
179
    '''
 
180
 
 
181
    def __init__(self, req, ns, path, version):
 
182
        super(ExternalVersionedMediaFileView, self).__init__(req, ns, path)
 
183
        self.version = version
 
184
 
 
185
    def _make_filename(self, req):
 
186
        if self.version != req.config['media']['version']:
 
187
            raise NotFound()
 
188
 
 
189
        # Don't expire for a year.
 
190
        req.headers_out['Expires'] = email.utils.formatdate(
 
191
                                    timeval=time.time() + (60*60*24*365),
 
192
                                    localtime=False,
 
193
                                    usegmt=True)
 
194
        return super(ExternalVersionedMediaFileView, self)._make_filename(req)
 
195
 
161
196
 
162
197
class Plugin(ViewPlugin, PublicViewPlugin):
163
 
    forward_routes = [(ApplicationRoot, '+media', root_to_media, INF)]
164
 
    views = [(MediaFile, '+index', MediaFileView)]
165
 
    public_forward_routes = forward_routes
166
 
    public_views = views
 
198
    urls = [
 
199
        ('+media/+:version/+external/:ns/*path', ExternalVersionedMediaFileView),
 
200
        ('+media/+external/:ns/*path', ExternalMediaFileView),
 
201
        ('+media/+:version/:ns/*path', VersionedMediaFileView),
 
202
        ('+media/:ns/*path', MediaFileView),
 
203
    ]
 
204
 
 
205
    public_urls = urls