Somehow this fix didn't get merged from trunk... (Saving gzipped Svg files)

mdboom · mdboom · commit f01765fea8a5 · 2007-12-12T20:13:52.000Z
svn path=/branches/transforms/; revision=4717
diff --git a/lib/matplotlib/backends/backend_svg.py b/lib/matplotlib/backends/backend_svg.py
@@ -517,9 +517,15 @@ def print_svg(self, filename, *args, **kwargs):
         return self._print_svg(filename, svgwriter, fh_to_close)
 
     def print_svgz(self, filename, *args, **kwargs):
-        gzipwriter = gzip.GzipFile(filename, 'w')
-        svgwriter = codecs.EncodedFile(gzipwriter, 'utf-8')
-        return self._print_svg(filename, svgwriter)
+        if is_string_like(filename):
+            gzipwriter = gzip.GzipFile(filename, 'w')
+            fh_to_close = svgwriter = codecs.EncodedFile(gzipwriter, 'utf-8')
+        elif is_writable_file_like(filename):
+            fh_to_close = gzipwriter = gzip.GzipFile(fileobj=filename, mode='w')
+            svgwriter = codecs.EncodedFile(gzipwriter, 'utf-8')
+        else:
+            raise ValueError("filename must be a path or a file-like object")
+        return self._print_svg(filename, svgwriter, fh_to_close)
 
     def _print_svg(self, filename, svgwriter, fh_to_close=None):
         self.figure.set_dpi(72.0)
