← Back to branch summary

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

« back to all changes in this revision

Viewing changes to ivle/webapp/media.py

Reimplement setup/buildjail.sh in Python. This means that sites can configure
additional repositories and packages to include in the jail, without modifying
the source tree.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/webapp/media.py
20
20
'''Media file support for the framework.'''
21
21
 
22
22
import os
23
 
import time
24
23
import inspect
25
24
import mimetypes
26
 
import email.utils
27
25
 
 
26
import ivle.conf
28
27
from ivle.webapp.base.views import BaseView
29
 
from ivle.webapp.base.plugins import PublicViewPlugin, ViewPlugin, MediaPlugin
 
28
from ivle.webapp.base.plugins import ViewPlugin, MediaPlugin
30
29
from ivle.webapp.errors import NotFound, Forbidden
31
 
from ivle.webapp.routing 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
30
 
39
31
def media_url(req, plugin, path):
40
32
    '''Generates a URL to a media file.
41
 
 
 
33
    
42
34
    Plugin can be a string, in which case it is put into the path literally,
43
 
    or a plugin object, in which case its name is looked up.
44
 
 
45
 
    If a version is specified in the IVLE configuration, a versioned URL will
46
 
    be generated.
 
35
    or a plugin object, in which case its name is looked up.'''
 
36
    if not isinstance(plugin, basestring):
 
37
        plugin = req.reverse_plugins[plugin]
 
38
 
 
39
    return os.path.join(ivle.conf.root_dir, '+media', plugin, path)
 
40
 
 
41
class MediaFileView(BaseView):
 
42
    '''A view for media files.
 
43
 
 
44
    This serves static files from directories registered by plugins.
 
45
 
 
46
    Plugins wishing to export media should declare a 'media' attribute,
 
47
    pointing to the directory to serve (relative to the module's directory).
 
48
    The contents of that directory will then be available under
 
49
    /+media/python.path.to.module.
47
50
    '''
48
 
    if not isinstance(plugin, basestring):
49
 
        plugin = req.config.reverse_plugins[plugin]
50
 
 
51
 
    media_path = os.path.join('+media', '+' + req.config['media']['version']) \
52
 
                    if req.config['media']['version'] else '+media'
53
 
 
54
 
    return req.make_path(os.path.join(media_path, plugin, path))
55
 
 
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
 
51
    def __init__(self, req, ns, path):
61
52
        self.ns = ns
62
53
        self.path = path
63
54
 
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 self.path not in extern[1]:
74
 
                return None
75
 
 
76
 
            # Grab the admin-configured path for this particular external dep.
77
 
            externdir = self.root.config['media']['externals'][extern[0]]
78
 
 
79
 
            assert isinstance(externdir, basestring)
80
 
 
81
 
            return os.path.join(externdir, self.path)
82
 
        else:
83
 
            try:
84
 
                plugin = self.root.config.plugins[self.ns]
85
 
            except KeyError:
86
 
                return None
87
 
 
88
 
            if not issubclass(plugin, MediaPlugin):
89
 
                return None
90
 
 
91
 
            mediadir = plugin.media
92
 
            plugindir = os.path.dirname(inspect.getmodule(plugin).__file__)
93
 
 
94
 
            return os.path.join(plugindir, mediadir, self.path)
95
 
 
96
 
class MediaFileView(BaseView):
97
 
    permission = None
 
55
    def _make_filename(self, req):
 
56
        try:
 
57
            plugin = req.plugins[self.ns]
 
58
        except KeyError:
 
59
            raise NotFound()
 
60
 
 
61
        if not issubclass(plugin, MediaPlugin):
 
62
            raise NotFound()
 
63
 
 
64
        mediadir = plugin.media
 
65
        plugindir = os.path.dirname(inspect.getmodule(plugin).__file__)
 
66
 
 
67
        return os.path.join(plugindir, mediadir, self.path)
98
68
 
99
69
    def render(self, req):
100
70
        # If it begins with ".." or separator, it's illegal. Die.
101
 
        if self.context.path.startswith("..") or \
102
 
           self.context.path.startswith('/'):
 
71
        if self.path.startswith("..") or self.path.startswith('/'):
103
72
            raise Forbidden()
104
73
 
105
 
        filename = self.context.filename
106
 
        if filename is None:
107
 
            raise NotFound()
 
74
        filename = self._make_filename(req)
108
75
 
109
76
        # Find an appropriate MIME type.
110
77
        (type, _) = mimetypes.guess_type(filename)
117
84
        if not os.access(filename, os.R_OK) or os.path.isdir(filename):
118
85
            raise Forbidden()
119
86
 
120
 
        if self.context.version is not None:
121
 
            req.headers_out['Expires'] = email.utils.formatdate(
122
 
                                timeval=time.time() + (60*60*24*365),
123
 
                                localtime=False,
124
 
                                usegmt=True)
125
 
 
126
 
 
127
87
        req.content_type = type
128
88
        req.sendfile(filename)
129
89
 
130
 
    def get_permissions(self, user):
131
 
        return set()
132
 
 
133
 
def root_to_media(root, *segments):
134
 
    if segments[0].startswith('+'):
135
 
        if segments[0] == '+external':
136
 
            external = True
137
 
            version = None
138
 
            path = segments[1:]
139
 
        else:
140
 
            version = segments[0][1:]
141
 
            if segments[1] == '+external':
142
 
                external = True
143
 
                path = segments[2:]
144
 
            else:
145
 
                external = False
146
 
                path = segments[1:]
147
 
    else:
148
 
        external = False
149
 
        version = None
150
 
        path = segments
151
 
 
152
 
    if version is not None and version != root.config['media']['version']:
153
 
        return None
154
 
 
155
 
    ns = path[0]
156
 
    path = os.path.normpath(os.path.join(*path[1:]))
157
 
 
158
 
    return MediaFile(root, external, version, ns, path)
159
 
 
160
 
class Plugin(ViewPlugin, PublicViewPlugin):
161
 
    forward_routes = [(ApplicationRoot, '+media', root_to_media, INF)]
162
 
    views = [(MediaFile, '+index', MediaFileView)]
 
90
class Plugin(ViewPlugin):
 
91
    urls = [
 
92
        ('+media/:ns/*path', MediaFileView),
 
93
    ]