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

fix #345

Browse Source
This commit is contained in:
wafi543 2021-04-19 12:35:38 +03:00
parent 04d1bfc5fe
commit 2b2628c6df
1 changed files with 49 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
docxtpl/__init__.py
View File

@ -318,6 +318,9 @@ class DocxTemplate(object):
# fix tables if needed # fix tables if needed
tree = self.fix_tables(xml_src) tree = self.fix_tables(xml_src)
# fix shapes if needed
tree = self.fix_shapes(xml_src)
self.map_tree(tree) self.map_tree(tree)
# Headers # Headers
@ -332,6 +335,52 @@ class DocxTemplate(object):
for relKey, xml in footers: for relKey, xml in footers:
self.map_headers_footers_xml(relKey, xml) self.map_headers_footers_xml(relKey, xml)
def fix_shapes(self, xml):
parser = etree.XMLParser(recover=True)
tree = etree.fromstring(xml, parser=parser)
# get namespace
ns = '{' + tree.nsmap['w'] + '}'
nsp = '{' + tree.nsmap['wp'] + '}'
mc = '{' + tree.nsmap['mc'] + '}'
# walk trough xml and fix docPr id duplicates
ids = []
amount = 2000
for p in tree.iter(ns+'p'):
rows = p.findall(ns+'r')
for r in rows:
drawing = r.find(ns+'drawing')
alternative = r.find(mc+'AlternateContent')
if drawing is not None:
anchor = drawing.find(nsp+'anchor')
if anchor is not None:
docPr = anchor.find(nsp+'docPr')
id = docPr.attrib['id']
if not id in ids:
ids.append(id)
else:
docPr.set('id', str(int(id)+amount))
elif alternative is not None:
choice = alternative.find(mc+'Choice')
drawing = choice.find(ns+'drawing')
anchor = drawing.find(nsp+'anchor')
inline = drawing.find(nsp+'inline')
if anchor is not None:
docPr = anchor.find(nsp+'docPr')
id = docPr.attrib['id']
if not id in ids:
ids.append(id)
else:
docPr.set('id', str(int(id)+amount))
elif inline is not None:
docPr = inline.find(nsp+'docPr')
id = docPr.attrib['id']
if not id in ids:
ids.append(id)
else:
docPr.set('id', str(int(id)+amount))
return tree
# using of TC tag in for cycle can cause that count of columns does not # using of TC tag in for cycle can cause that count of columns does not
# correspond to real count of columns in row. This function is able to fix it. # correspond to real count of columns in row. This function is able to fix it.
def fix_tables(self, xml): def fix_tables(self, xml):