Merge pull request #1 from darkvisiontech/fix-svg-suffix

DarrenPadraigDV · web-flow · commit caad0fdae355 · 2025-05-26T15:07:59.000-07:00
Fix svg suffix
diff --git a/src/docx/image/svg.py b/src/docx/image/svg.py
@@ -33,13 +33,28 @@ def default_ext(self):
         """
         return "svg"
 
+    @classmethod
+    def _parse_svg_dims(cls, value):
+        svg_dims_dict = {
+            'px': 1,
+            'cm': 37.7952755906,
+            'pt': 1 / 0.75,
+            'in': 96,
+            'mm': 3.77952755906,
+        }
+        try:
+            return int(value)
+        except ValueError:
+            for unit, factor in svg_dims_dict.items():
+                if value.endswith(unit):
+                    return int(float(value[:-len(unit)]) * factor)
+            
     @classmethod
     def _dimensions_from_stream(cls, stream):
         stream.seek(0)
         data = stream.read()
         root = ET.fromstring(data)
-        # FIXME: The width could be expressed as '4cm'
         # See https://www.w3.org/TR/SVG11/struct.html#NewDocument
-        width = int(root.attrib["width"])
-        height = int(root.attrib["height"])
+        width = cls._parse_svg_dims(root.attrib["width"])
+        height = cls._parse_svg_dims(root.attrib["height"])
         return width, height
