-
Notifications
You must be signed in to change notification settings - Fork 5
/
loader.py
89 lines (73 loc) · 3.24 KB
/
loader.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import os
import sys
from importlib import import_module
from django.template import TemplateDoesNotExist
from django.template.loaders.app_directories import Loader as BaseLoader
from django.conf import settings
from django.core.exceptions import ImproperlyConfigured
from django.utils import six
# taken from / inspired by django.template.loaders.app_directories
# At compile time, cache the directories to search.
if not six.PY3:
fs_encoding = sys.getfilesystemencoding() or sys.getdefaultencoding()
app_templates_dirs = []
app_layers_dirs = []
for app in settings.INSTALLED_APPS:
try:
mod = import_module(app)
except ImportError as e:
raise ImproperlyConfigured('ImportError %s: %s' % (app, e.args[0]))
layers_dir = os.path.join(os.path.dirname(mod.__file__), 'layers')
if os.path.isdir(layers_dir):
if not six.PY3:
layers_dir = layers_dir.decode(fs_encoding)
app_layers_dirs.append(layers_dir)
template_dir = os.path.join(os.path.dirname(mod.__file__), 'templates')
if os.path.isdir(template_dir):
if not six.PY3:
template_dir = template_dir.decode(fs_encoding)
app_templates_dirs.append(template_dir)
# It won't change, so convert it to a tuple to save memory.
app_templates_dirs = tuple(app_templates_dirs)
app_layers_dirs = tuple(app_layers_dirs)
"""
Resolve the callable that resolves the layer to be activated, if any
"""
def load_conf(app, module):
try:
mod = import_module(app + "." + module)
if callable(mod.get_layer):
app_layers_funcs.append(mod.get_layer)
except (ImportError, AttributeError):
pass # don't care
app_layers_funcs = []
for app in settings.INSTALLED_APPS:
load_conf(app, "layers")
load_conf(app, "select_layer")
app_layers_funcs = tuple(app_layers_funcs)
class LayerLoader(BaseLoader):
def load_template_source(self, template_name, layers_dirs=None, templates_dirs=None, layers_funcs=None):
from layers.middleware import get_current_request
request = get_current_request()
layers_dirs = layers_dirs or app_layers_dirs
templates_dirs = templates_dirs or app_templates_dirs
layers_funcs = layers_funcs or app_layers_funcs
for f in layers_funcs:
## optimization: check if we didn't already try this prefix in a previous iteration
prefix = f(request)
if prefix:
if layers_dirs:
for filepath in self.get_template_sources(os.path.join(prefix, "templates", template_name), layers_dirs):
try:
with open(filepath, 'rb') as fp:
return (fp.read().decode(settings.FILE_CHARSET), filepath)
except IOError:
pass
if templates_dirs:
for filepath in self.get_template_sources(os.path.join(prefix, template_name), templates_dirs):
try:
with open(filepath, 'rb') as fp:
return (fp.read().decode(settings.FILE_CHARSET), filepath)
except IOError:
pass
raise TemplateDoesNotExist(template_name)