filmroellchen/intro-outro-generator
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor svgtemplate into a neat context class SVGTemplate

Browse source
This commit is contained in:
Daniel Molkentin 2019-08-02 00:10:49 +02:00
parent 7a915605d3
commit 1b3892a984
2 changed files with 52 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

77
svgtemplate.py
View file

@ -1,43 +1,54 @@
# vim: tabstop=4 shiftwidth=4 expandtab
import builtins
import cssutils
import logging
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()
cssutils.ser.prefs.lineSeparator = ' '
cssutils.log.setLevel(logging.FATAL)
def replacetext(svgstr, task):
for key in task.parameters.keys():
svgstr = svgstr.replace(key, xmlescape(str(task.parameters[key])))
return svgstr
class SVGTemplate:
def __init__(self, task):
self.task = task
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 __enter__(self):
with builtins.open(os.path.join(self.task.workdir, self.task.infile), 'r') as fp:
self.svgstr = fp.read()
return self
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
def write(self):
# open the output-file (named ".gen.svg" in the workdir)
outfile = os.path.join(self.task.workdir, '.gen.svg')
with builtins.open(outfile, 'w') as fp:
# write the generated svg-text into the output-file
fp.write(self.svgstr)
return outfile
def replacetext(self):
for key in self.task.parameters.keys():
self.svgstr = self.svgstr.replace(key, xmlescape(str(self.task.parameters[key])))
def transform(self, frame):
parser = etree.XMLParser(huge_tree=True)
svg = etree.fromstring(self.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)
self.xmlstr = etree.tostring(svg, encoding='unicode')
def __exit__(self, exception_type, exception_value, traceback):
pass