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

Rename pic_to_replace to pics_to_replace

Browse Source 
Latter name is more intuitive because this dictionary may hold the metadata
of several pictures.
This commit is contained in:
Alan Evangelista 2021-04-20 11:59:45 -03:00
parent 04d1bfc5fe
commit d90087d6f7
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
docxtpl/__init__.py
View File

@ -39,7 +39,7 @@ class DocxTemplate(object):
self.crc_to_new_media = {} self.crc_to_new_media = {}
self.crc_to_new_embedded = {} self.crc_to_new_embedded = {}
self.zipname_to_replace = {} self.zipname_to_replace = {}
self.pic_to_replace = {} self.pics_to_replace = {}
self.pic_map = {} self.pic_map = {}
self.current_rendering_part = None self.current_rendering_part = None
@ -477,10 +477,10 @@ class DocxTemplate(object):
if hasattr(dst_file, 'read'): if hasattr(dst_file, 'read'):
# NOTE: file extension not checked # NOTE: file extension not checked
self.pic_to_replace[embedded_file] = dst_file.read() self.pics_to_replace[embedded_file] = dst_file.read()
else: else:
with open(dst_file, 'rb') as fh: with open(dst_file, 'rb') as fh:
self.pic_to_replace[embedded_file] = fh.read() self.pics_to_replace[embedded_file] = fh.read()
def replace_embedded(self, src_file, dst_file): def replace_embedded(self, src_file, dst_file):
"""Replace one embedded object by another one into a docx """Replace one embedded object by another one into a docx
@ -568,11 +568,11 @@ class DocxTemplate(object):
def pre_processing(self): def pre_processing(self):
if self.pic_to_replace: if self.pics_to_replace:
self.build_pic_map() self.build_pic_map()
# Do the actual replacement # Do the actual replacement
for embedded_file, stream in six.iteritems(self.pic_to_replace): for embedded_file, stream in six.iteritems(self.pics_to_replace):
if embedded_file not in self.pic_map: if embedded_file not in self.pic_map:
raise ValueError('Picture "%s" not found in the docx template' raise ValueError('Picture "%s" not found in the docx template'
% embedded_file) % embedded_file)
@ -581,7 +581,7 @@ class DocxTemplate(object):
def build_pic_map(self): def build_pic_map(self):
"""Searches in docx template all the xml pictures tag and store them """Searches in docx template all the xml pictures tag and store them
in pic_map dict""" in pic_map dict"""
if self.pic_to_replace: if self.pics_to_replace:
# Main document # Main document
part = self.docx.part part = self.docx.part
self.pic_map.update(self._img_filename_to_part(part)) self.pic_map.update(self._img_filename_to_part(part))