dbt-selly/dbt-env/lib/python3.8/site-packages/leather/svg.py

44 lines
947 B
Python

#!/usr/bin/env python
"""
Helpers for working with SVG.
"""
import xml.etree.ElementTree as ET
import six
HEADER = '<?xml version="1.0" standalone="no"?>\n' + \
'<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN"\n' + \
'"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">\n'
def stringify(root):
"""
Convert an SVG XML tree to a unicode string.
"""
if six.PY3:
return ET.tostring(root, encoding='unicode')
else:
return ET.tostring(root, encoding='utf-8')
def save(f, root):
"""
Save an SVG XML tree to a file.
"""
f.write(HEADER)
f.write(stringify(root))
def translate(x, y):
"""
Generate an SVG transform statement representing a simple translation.
"""
return 'translate(%i %i)' % (x, y)
def rotate(deg, x, y):
"""
Generate an SVG transform statement representing rotation around a given
point.
"""
return 'rotate(%i %i %i)' % (deg, x, y)