Use stdlib mimetypes instead of hardcoding them.

anntzer · anntzer · commit 2e9397cbabde · 2019-05-14T14:36:07.000+02:00
The stdlib provides all necessary mimetypes (see _default_mime_types),
except for application/emf -- but we don't have an emf backend anymore.

This also makes the dict (relatively) easily extensible by the end user,
who could call `mimetypes.init(...)` themselves to add entries to the
mapping.
diff --git a/examples/user_interfaces/embedding_webagg_sgskip.py b/examples/user_interfaces/embedding_webagg_sgskip.py
@@ -3,16 +3,16 @@
 Embedding WebAgg
 ================
 
-This example demonstrates how to embed matplotlib WebAgg interactive
-plotting in your own web application and framework.  It is not
-necessary to do all this if you merely want to display a plot in a
-browser or use matplotlib's built-in Tornado-based server "on the
-side".
+This example demonstrates how to embed Matplotlib WebAgg interactive plotting
+in your own web application and framework.  It is not necessary to do all this
+if you merely want to display a plot in a browser or use Matplotlib's built-in
+Tornado-based server "on the side".
 
 The framework being used must support web sockets.
 """
 
 import io
+import mimetypes
 
 try:
     import tornado
@@ -138,20 +138,8 @@ class Download(tornado.web.RequestHandler):
 
         def get(self, fmt):
             manager = self.application.manager
-
-            mimetypes = {
-                'ps': 'application/postscript',
-                'eps': 'application/postscript',
-                'pdf': 'application/pdf',
-                'svg': 'image/svg+xml',
-                'png': 'image/png',
-                'jpeg': 'image/jpeg',
-                'tif': 'image/tiff',
-                'emf': 'application/emf'
-            }
-
-            self.set_header('Content-Type', mimetypes.get(fmt, 'binary'))
-
+            self.set_header(
+                'Content-Type', mimetypes.types_map.get(fmt, 'binary'))
             buff = io.BytesIO()
             manager.canvas.figure.savefig(buff, format=fmt)
             self.write(buff.getvalue())
diff --git a/lib/matplotlib/backends/backend_webagg.py b/lib/matplotlib/backends/backend_webagg.py
@@ -15,6 +15,7 @@
 import errno
 from io import BytesIO
 import json
+import mimetypes
 from pathlib import Path
 import random
 import sys
@@ -113,21 +114,8 @@ class Download(tornado.web.RequestHandler):
         def get(self, fignum, fmt):
             fignum = int(fignum)
             manager = Gcf.get_fig_manager(fignum)
-
-            # TODO: Move this to a central location
-            mimetypes = {
-                'ps': 'application/postscript',
-                'eps': 'application/postscript',
-                'pdf': 'application/pdf',
-                'svg': 'image/svg+xml',
-                'png': 'image/png',
-                'jpeg': 'image/jpeg',
-                'tif': 'image/tiff',
-                'emf': 'application/emf'
-            }
-
-            self.set_header('Content-Type', mimetypes.get(fmt, 'binary'))
-
+            self.set_header(
+                'Content-Type', mimetypes.types_map.get(fmt, 'binary'))
             buff = BytesIO()
             manager.canvas.figure.savefig(buff, format=fmt)
             self.write(buff.getvalue())
