Renamed auto escaping environment.
Marc Brinkmann
8 years ago
| 14 | 14 |
|
| 15 | 15 |
from datetime import datetime
|
| 16 | 16 |
|
| 17 | |
from jinja_vanish import DynEscapeAutoenvironment, markup_escape_func
|
|
17 |
from jinja_vanish import DynAutoEscapeEnvironment, markup_escape_func
|
| 18 | 18 |
from psycopg2.extensions import adapt
|
| 19 | 19 |
|
| 20 | 20 |
@markup_escape_func
|
|
| 24 | 24 |
return adapt(v)
|
| 25 | 25 |
|
| 26 | 26 |
|
| 27 | |
env = DynEscapeAutoenvironment(autoescape=True, escape_func=sql_escape)
|
|
27 |
env = DynAutoEscapeEnvironment(autoescape=True, escape_func=sql_escape)
|
| 28 | 28 |
tpl = env.from_string('SELECT * FROM foo where post_date <= {{now}}')
|
| 29 | 29 |
|
| 30 | 30 |
print(tpl.render(now=datetime.now()))
|
| 17 | 17 |
self.writeline('{} = {}'.format(name, override))
|
| 18 | 18 |
|
| 19 | 19 |
|
| 20 | |
class DynEscapeAutoenvironment(Environment):
|
|
20 |
class DynAutoEscapeEnvironment(Environment):
|
| 21 | 21 |
code_generator_class = LocalOverridingCodeGenerator
|
| 22 | 22 |
|
| 23 | 23 |
def __init__(self, *args, **kwargs):
|
| 24 | 24 |
escape_func = kwargs.pop('escape_func', None)
|
| 25 | 25 |
markup_class = kwargs.pop('markup_class', None)
|
| 26 | 26 |
|
| 27 | |
super(DynEscapeAutoenvironment, self).__init__(*args, **kwargs)
|
|
27 |
super(DynAutoEscapeEnvironment, self).__init__(*args, **kwargs)
|
| 28 | 28 |
|
| 29 | 29 |
# we need to disable constant-evaluation at compile time, because it
|
| 30 | 30 |
# calls jinja's own escape function.
|
| 0 | 0 |
# coding: utf8
|
| 1 | 1 |
|
| 2 | 2 |
from markupsafe import Markup
|
| 3 | |
from jinja_vanish import DynEscapeAutoenvironment
|
|
3 |
from jinja_vanish import DynAutoEscapeEnvironment
|
| 4 | 4 |
from six import text_type
|
| 5 | 5 |
import pytest
|
| 6 | 6 |
|
|
| 40 | 40 |
|
| 41 | 41 |
@pytest.fixture()
|
| 42 | 42 |
def env(escape_func):
|
| 43 | |
return DynEscapeAutoenvironment(autoescape=True, escape_func=escape_func)
|
|
43 |
return DynAutoEscapeEnvironment(autoescape=True, escape_func=escape_func)
|
| 44 | 44 |
|
| 45 | 45 |
|
| 46 | 46 |
@pytest.fixture()
|