dgsoft/python-docx-template
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #134 from jhpyle/error-context

Browse Source 
add docx file context to TemplateError exceptions
This commit is contained in:
Eric Lapouyade 2018-08-03 11:26:21 +02:00 committed by GitHub
commit c9afd31320
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
docxtpl/__init__.py
View File

@ -13,6 +13,7 @@ from docx.opc.oxml import serialize_part_xml, parse_xml
import docx.oxml.ns import docx.oxml.ns
from docx.opc.constants import RELATIONSHIP_TYPE as REL_TYPE from docx.opc.constants import RELATIONSHIP_TYPE as REL_TYPE
from jinja2 import Template from jinja2 import Template
from jinja2.exceptions import TemplateError
from cgi import escape from cgi import escape
import re import re
import six import six
@ -105,11 +106,19 @@ class DocxTemplate(object):
return src_xml return src_xml
def render_xml(self,src_xml,context,jinja_env=None): def render_xml(self,src_xml,context,jinja_env=None):
if jinja_env: src_xml = src_xml.replace(r'<w:p>', '\n<w:p>')
template = jinja_env.from_string(src_xml) try:
else: if jinja_env:
template = Template(src_xml) template = jinja_env.from_string(src_xml)
dst_xml = template.render(context) else:
template = Template(src_xml)
dst_xml = template.render(context)
except TemplateError as exc:
if hasattr(exc, 'lineno') and exc.lineno is not None:
line_number = max(exc.lineno - 4, 0)
exc.docx_context = map(lambda x: re.sub(r'<[^>]+>', '', x), src_xml.splitlines()[line_number:(line_number + 7)])
raise exc
dst_xml = dst_xml.replace('\n<w:p>', '<w:p>')
dst_xml = dst_xml.replace('{_{','{{').replace('}_}','}}').replace('{_%','{%').replace('%_}','%}') dst_xml = dst_xml.replace('{_{','{{').replace('}_}','}}').replace('{_%','{%').replace('%_}','%}')
return dst_xml return dst_xml