summaryrefslogtreecommitdiff
path: root/examples/rwbench/djangoext.py
blob: 0052aefc283d28d8c5beca54199e8e7c4d3e418a (plain)
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})
        )