Fix missing svg attribute bug #76

mateosss · Jul 25, 2022 · 2fe52ef · 2fe52ef
1 parent 1a05515
commit 2fe52ef
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/svg2png.py b/svg2png.py
@@ -50,10 +50,7 @@ def prettify(xml_string):
         ET.register_namespace(key, value)
 
     root = dom.getroot()
-    width, height = (
-        int_ignore_units(root.attrib["width"]),
-        int_ignore_units(root.attrib["height"]),
-    )
+    _, _, width, height = map(int, filter(bool, re.split(r'[ ,]', root.attrib["viewBox"])))
     width_gap, height_gap = (1 - FRAC) * width / 2, (1 - FRAC) * height / 2
 
     # Group all elements that are children of <svg> while changing their 'style' attributes