orxporter/svg.py

import re

def translate_color(svg, pfrom, pto):
    """
    Translates colours from a source file's original colours into new colours.
    """
    res, _ = _translate_color(svg, pfrom, pto)
    return res

def translated_colors(svg, pfrom, pto):
    """
    Return a dictionary of (entry: to_colour) for the colours that were
    translated in svg
    """
    _, changed = _translate_color(svg, pfrom, pto)
    return changed

def _translate_color(svg, pfrom, pto):
    """
    Translates colours on the svg file from "pfrom" to "pto", reporting also
    which colours changed.
    """
    # looks for incidences of one colour, compares it and replaces it.
    res = svg
    changed = {}
    for centry, ccol in pfrom.items():
        cr = ccol + ';'
        cro = pto.get(centry)
        if not cro:
            continue
        cro += ';'
        res, count = re.subn(cr, cro, res, flags=re.IGNORECASE)
        if (cr[1], cr[3], cr[5]) == (cr[2], cr[4], cr[6]):
            cr = cr[0] + cr[1] + cr[3] + cr[5] + ';'
            res, count2 = re.sub(cr, cro, res, flags=re.IGNORECASE)
            count += count2

        # Record changes
        if count > 0:
            changed[centry] = cro

    return (res, changed)

def add_license(svg, license_data):
    """
    Adds licensing metadata to an SVG.
    """
    svgidx = svg.index('<svg')
    insidx = svg.index('>', svgidx) + 1
    metastr = '\n<metadata>\n' + license_data + '</metadata>\n'
    return svg[:insidx] + metastr + svg[insidx:]

def get_viewbox_size(svg):
    """
    Returns the size of an SVG's viewbox (as a tuple containing width and height).
    """
    try:
        x1, y1, x2, y2 = list(map(
            int, svg[svg.index('viewBox'):].split('"', 2)[1].split()))
        return x2 - x1, y2 - y1
    except Exception:
        raise ValueError('Could not detect image size')