middleware.py 1.84 KB
Newer Older
1 2 3 4 5 6 7 8 9
#   Copyright (c) 2008 Mikeal Rogers
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
10
#   distribuetd under the License is distributed on an "AS IS" BASIS,
11 12 13 14 15 16
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

from mako.lookup import TemplateLookup
import tempfile
Piotr Mitros committed
17
from django.template import RequestContext
18
from django.conf import settings
Piotr Mitros committed
19 20

requestcontext = None
21
lookup = {}
22 23 24 25 26 27

class MakoMiddleware(object):
    def __init__(self):
        """Setup mako variables and lookup object"""
        from django.conf import settings
        # Set all mako variables based on django settings
28
        template_locations = settings.MAKO_TEMPLATES
29
        module_directory = getattr(settings, 'MAKO_MODULE_DIR', None)
30

31 32 33
        if module_directory is None:
            module_directory = tempfile.mkdtemp()

34 35
        for location in template_locations:
            lookup[location] = TemplateLookup(directories=template_locations[location], 
36
                                module_directory=module_directory,
37 38 39
                                output_encoding='utf-8', 
                                input_encoding='utf-8', 
                                encoding_errors='replace',
40
                                )
41

42 43
        import mitxmako
        mitxmako.lookup = lookup
Piotr Mitros committed
44 45 46 47

    def process_request (self, request):
        global requestcontext
        requestcontext = RequestContext(request)
48 49
        requestcontext['is_secure'] = request.is_secure()
        requestcontext['site'] = settings.SITE_NAME