refactor svgtemplate into seperate file
This commit is contained in:
parent
375c73d383
commit
7a915605d3
2 changed files with 52 additions and 50 deletions
43
svgtemplate.py
Normal file
43
svgtemplate.py
Normal file
|
@ -0,0 +1,43 @@
|
|||
# vim: tabstop=4 shiftwidth=4 expandtab
|
||||
import builtins
|
||||
import cssutils
|
||||
import os
|
||||
from lxml import etree
|
||||
from xml.sax.saxutils import escape as xmlescape
|
||||
|
||||
def open(task):
|
||||
with builtins.open(os.path.join(task.workdir, task.infile), 'r') as fp:
|
||||
return fp.read()
|
||||
|
||||
def replacetext(svgstr, task):
|
||||
for key in task.parameters.keys():
|
||||
svgstr = svgstr.replace(key, xmlescape(str(task.parameters[key])))
|
||||
return svgstr
|
||||
|
||||
def transform(svgstr, frame, task):
|
||||
parser = etree.XMLParser(huge_tree=True)
|
||||
svg = etree.fromstring(svgstr.encode('utf-8'), parser)
|
||||
# apply the replace-pairs to the input text, by finding the specified xml-elements by their id and modify their css-parameter the correct value
|
||||
for replaceinfo in frame:
|
||||
(id, type, key, value) = replaceinfo
|
||||
for el in svg.findall(".//*[@id='" + id.replace("'", "\\'") + "']"):
|
||||
if type == 'style':
|
||||
style = cssutils.parseStyle(el.attrib['style'] if 'style' in el.attrib else '')
|
||||
style[key] = str(value)
|
||||
el.attrib['style'] = style.cssText
|
||||
elif type == 'attr':
|
||||
el.attrib[key] = str(value)
|
||||
elif type == 'text':
|
||||
el.text = str(value)
|
||||
# if '$subtitle' in task.parameters and task.parameters['$subtitle'] == '':
|
||||
# child = svg.findall(".//*[@id='subtitle']")[0]
|
||||
# child.getparent().remove(child)
|
||||
return etree.tostring(svg, encoding='unicode')
|
||||
|
||||
def write(svgstr, task):
|
||||
# open the output-file (named ".gen.svg" in the workdir)
|
||||
outfile = os.path.join(task.workdir, '.gen.svg')
|
||||
with builtins.open(outfile, 'w') as fp:
|
||||
# write the generated svg-text into the output-file
|
||||
fp.write(svgstr)
|
||||
return outfile
|
Loading…
Add table
Add a link
Reference in a new issue