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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
# -*- coding: utf-8 -*-
from os.path import join
from django import template as django_template_module
from django.conf import settings
from django.template import Context as DjangoContext
from django.template import loader as django_loader
from django.template import Node
from django.template import TokenParser
from django.template import Variable
from rwbench import dateformat
from rwbench import ROOT
settings.configure(
TEMPLATE_DIRS=(join(ROOT, "django"),),
TEMPLATE_LOADERS=(
(
"django.template.loaders.cached.Loader",
("django.template.loaders.filesystem.Loader",),
),
),
)
# for django extensions. We monkey patch our extensions in so that
# we don't have to initialize a more complex django setup.
django_extensions = django_template_module.Library()
django_template_module.builtins.append(django_extensions)
django_extensions.filter(dateformat)
def var_or_none(x):
if x is not None:
return Variable(x)
# and more django extensions
@django_extensions.tag
def input_field(parser, token):
p = TokenParser(token.contents)
args = [p.value()]
while p.more():
args.append(p.value())
return InputFieldNode(*args)
@django_extensions.tag
def textarea(parser, token):
p = TokenParser(token.contents)
args = [p.value()]
while p.more():
args.append(p.value())
return TextareaNode(*args)
@django_extensions.tag
def form(parser, token):
p = TokenParser(token.contents)
args = []
while p.more():
args.append(p.value())
body = parser.parse(("endform",))
parser.delete_first_token()
return FormNode(body, *args)
class InputFieldNode(Node):
def __init__(self, name, type=None, value=None):
self.name = var_or_none(name)
self.type = var_or_none(type)
self.value = var_or_none(value)
def render(self, context):
name = self.name.resolve(context)
type = "text"
value = ""
if self.type is not None:
type = self.type.resolve(context)
if self.value is not None:
value = self.value.resolve(context)
tmpl = django_loader.get_template("_input_field.html")
return tmpl.render(DjangoContext({"name": name, "type": type, "value": value}))
class TextareaNode(Node):
def __init__(self, name, rows=None, cols=None, value=None):
self.name = var_or_none(name)
self.rows = var_or_none(rows)
self.cols = var_or_none(cols)
self.value = var_or_none(value)
def render(self, context):
name = self.name.resolve(context)
rows = 10
cols = 40
value = ""
if self.rows is not None:
rows = int(self.rows.resolve(context))
if self.cols is not None:
cols = int(self.cols.resolve(context))
if self.value is not None:
value = self.value.resolve(context)
tmpl = django_loader.get_template("_textarea.html")
return tmpl.render(
DjangoContext({"name": name, "rows": rows, "cols": cols, "value": value})
)
class FormNode(Node):
def __init__(self, body, action=None, method=None):
self.body = body
self.action = action
self.method = method
def render(self, context):
body = self.body.render(context)
action = ""
method = "post"
if self.action is not None:
action = self.action.resolve(context)
if self.method is not None:
method = self.method.resolve(context)
tmpl = django_loader.get_template("_form.html")
return tmpl.render(
DjangoContext({"body": body, "action": action, "method": method})
)
|