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
|
from jinja2 import nodes
from jinja2.ext import Extension
import cacheops
from cacheops.utils import carefully_strip_whitespace
__all__ = ['cache']
class CacheopsExtension(Extension):
tags = ['cached_as', 'cached']
def parse(self, parser):
lineno = parser.stream.current.lineno
tag_name = parser.stream.current.value
tag_location = '%s:%s' % (parser.name, lineno)
next(parser.stream)
args, kwargs = self.parse_args(parser)
args = [nodes.Const(tag_name), nodes.Const(tag_location)] + args
block_call = self.call_method('handle_tag', args, kwargs)
body = parser.parse_statements(['name:end%s' % tag_name], drop_needle=True)
return nodes.CallBlock(block_call, [], [], body).set_lineno(lineno)
def handle_tag(self, tag_name, tag_location, *args, **kwargs):
caller = kwargs.pop('caller')
cacheops_decorator = getattr(cacheops, tag_name)
kwargs.setdefault('extra', '')
if isinstance(kwargs['extra'], tuple):
kwargs['extra'] += (tag_location,)
else:
kwargs['extra'] = str(kwargs['extra']) + tag_location
@cacheops_decorator(*args, **kwargs)
def _handle_tag():
content = caller()
# TODO: make this cache preparation configurable
return carefully_strip_whitespace(content)
return _handle_tag()
def parse_args(self, parser):
args = []
kwargs = []
require_comma = False
while parser.stream.current.type != 'block_end':
if require_comma:
parser.stream.expect('comma')
if parser.stream.current.type == 'name' and parser.stream.look().type == 'assign':
key = parser.stream.current.value
parser.stream.skip(2)
value = parser.parse_expression()
kwargs.append(nodes.Keyword(key, value, lineno=value.lineno))
else:
if kwargs:
parser.fail('Invalid argument syntax for CacheopsExtension tag',
parser.stream.current.lineno)
args.append(parser.parse_expression())
require_comma = True
return args, kwargs
cache = CacheopsExtension
|
