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
|
import os
from django.db import models
from django.template.loader import (Context, Template, TemplateDoesNotExist,
find_template_loader)
from django.utils.translation import ugettext_lazy as _
DEFAULT_TEMPLATE_LOADERS = (
'django.template.loaders.filesystem.Loader',
'django.template.loaders.app_directories.Loader',
)
class TemplateChoices(object):
def __init__(self, template_loaders):
self.template_loaders = template_loaders
def __iter__(self):
seen = set()
for loader in self.template_loaders:
for basepath in loader.get_template_sources('.'):
path = os.path.join(basepath, 'content', 'template')
try:
templates = os.listdir(path)
except (OSError, IOError):
continue
for template in templates:
if template in seen:
continue
if template.endswith(('~', '.tmp')):
continue
seen.add(template)
return ((t, t) for t in sorted(seen))
class TemplateContent(models.Model):
"""
This content type scans all template folders for files in the
``content/template/`` folder and lets the website administrator select
any template from a set of provided choices.
The templates aren't restricted in any way.
"""
class Meta:
abstract = True
verbose_name = _('template content')
verbose_name_plural = _('template contents')
@classmethod
def initialize_type(cls, TEMPLATE_LOADERS=DEFAULT_TEMPLATE_LOADERS):
cls.template_loaders = [find_template_loader(loader)
for loader in TEMPLATE_LOADERS if loader]
cls.add_to_class('filename', models.CharField(_('template'), max_length=100,
choices=TemplateChoices(cls.template_loaders)))
def render(self, **kwargs):
context = kwargs.pop('context', None)
name = 'content/template/%s' % self.filename
for loader in self.template_loaders:
try:
template, display_name = loader.load_template(name)
except TemplateDoesNotExist:
continue
if not hasattr(template, 'render'):
template = Template(template, name=name)
if context:
ctx = context
ctx.update(dict(content=self, **kwargs))
else:
ctx = Context(dict(content=self, **kwargs))
result = template.render(ctx)
if context:
context.pop()
return result
return u'' # Fail?
|
