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

Autodetect header/footer encoding

Browse Source
This commit is contained in:
Eric Lapouyade 2016-07-13 16:38:40 +02:00
parent 4088bc4ab0
commit 12316e94d7
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
docxtpl/__init__.py
View File

@ -88,16 +88,23 @@ class DocxTemplate(object):
def get_headers_footers_xml(self, uri):
for relKey, val in self.docx._part._rels.items():
if val.reltype == uri:
yield relKey, val._target._blob.decode()
yield relKey, val._target._blob
def get_headers_footers_encoding(self,xml):
m = re.match(r'<\?xml[^\?]+\bencoding="([^"]+)"',xml,re.I)
if m:
return m.group(1)
return 'utf-8'
def build_headers_footers_xml(self,context, uri,jinja_env=None):
for relKey, xml in self.get_headers_footers_xml(uri):
xml = self.patch_xml(xml)
encoding = self.get_headers_footers_encoding(xml)
xml = self.patch_xml(xml).decode(encoding)
xml = self.render_xml(xml, context, jinja_env)
yield relKey, xml
yield relKey, xml.encode(encoding)
def map_headers_footers_xml(self, relKey, xml):
self.docx._part._rels[relKey]._target._blob = xml.encode()
self.docx._part._rels[relKey]._target._blob = xml
def render(self,context,jinja_env=None):
# Body