Merge pull request matplotlib#3981 from blink1073/webagg-save-fixes

Fixes for File Saving in Webagg

Author: pelson

lib/matplotlib/backends/backend_webagg.py

@@ -17,7 +17,6 @@
 
 import datetime
 import errno
-import io
 import json
 import os
 import random
@@ -197,7 +196,7 @@ def get(self, fignum, fmt):
 
             self.set_header('Content-Type', mimetypes.get(fmt, 'binary'))
 
-            buff = io.BytesIO()
+            buff = six.BytesIO()
             manager.canvas.print_figure(buff, format=fmt)
             self.write(buff.getvalue())
 

lib/matplotlib/backends/backend_webagg_core.py

@@ -404,7 +404,8 @@ def get_javascript(cls, stream=None):
         for filetype, ext in sorted(FigureCanvasWebAggCore.
                                     get_supported_filetypes_grouped().
                                     items()):
-            extensions.append(ext[0])
+            if not ext[0] == 'pgf':  # pgf does not support BytesIO
+                extensions.append(ext[0])
         output.write("mpl.extensions = {0};\n\n".format(
             json.dumps(extensions)))