43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
# 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
|