Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support custom python object serialization #108

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Support custom python object serialization #108

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions chevron/renderer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ def _get_partial(name, partials_dict, partials_path, partials_ext):

def render(template='', data={}, partials_path='.', partials_ext='mustache',
partials_dict={}, padding='', def_ldel='{{', def_rdel='}}',
scopes=None, warn=False, keep=False):
scopes=None, warn=False, keep=False, serializer=str):
"""Render a mustache template.

Renders a mustache template with a data scope and partial capability.
Expand Down Expand Up @@ -179,6 +179,8 @@ def render(template='', data={}, partials_path='.', partials_ext='mustache',

keep -- Keep unreplaced tags when a template substitution isn't found in the data

serializer -- Python data serializer (str by default)


Returns:

Expand Down Expand Up @@ -237,15 +239,15 @@ def render(template='', data={}, partials_path='.', partials_ext='mustache',
# then get the un-coerced object (next in the stack)
thing = scopes[1]
if not isinstance(thing, unicode_type):
thing = unicode(str(thing), 'utf-8')
thing = unicode(serializer(thing), 'utf-8')
output += _html_escape(thing)

# If we're a no html escape tag
elif tag == 'no escape':
# Just lookup the key and add it
thing = _get_key(key, scopes, warn=warn, keep=keep, def_ldel=def_ldel, def_rdel=def_rdel)
if not isinstance(thing, unicode_type):
thing = unicode(str(thing), 'utf-8')
thing = unicode(serializer(thing), 'utf-8')
output += thing

# If we're a section tag
Expand Down
15 changes: 15 additions & 0 deletions test_spec.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -552,6 +552,21 @@ def test_keep_from_partials(self):
expected = '1st {{ missing_key }} 3rd'
self.assertEqual(result, expected)

def test_custom_serializer(self):
args = {
'template': '{{ value }}',
'data': {
'value': {
'key': None,
},
},
'serializer': json.dumps,
}

result = chevron.render(**args)
expected = '{"key": null}'
self.assertEqual(result, expected)


# Run unit tests from command line
if __name__ == "__main__":
Expand Down