Add ipython mimetypes extension

README.md

@@ -121,6 +121,17 @@ Type "help", "copyright", "credits" or "license" for more information.
 
 You should see a plot!
 
+### image mimetype support
+
+If you want to display a wider variety of objects using ipython's mimetype capturing, load our bundled ipython extension:
+
+```ipython
+$ %load_ext itermplot.ipython
+$ from PIL import Image
+$ Image.open("my_cool_image.png")
+# your image should be displayed
+```
+
 ## Uninstall
 
 You can disable this backend by unsetting the `MPLBACKEND` environment variable.

itermplot/ipython/__init__.py

@@ -0,0 +1,30 @@
+"""
+ipython extensions for image output handling.
+e.g.
+
+```ipython
+%load_ext itermplot.ipython
+```
+
+implemented based on https://ipython.readthedocs.io/en/stable/config/shell_mimerenderer.html
+"""
+
+from .. import imgcat
+
+
+def register_mimerenderer(ipython, mime, handler):
+    ipython.display_formatter.active_types.append(mime)
+    ipython.display_formatter.formatters[mime].enabled = True
+    ipython.mime_renderers[mime] = handler
+
+
+def imgcat_factory(fn):
+    def _wrapper(img, _):
+        imgcat(img, fn=fn)
+
+    return _wrapper
+
+
+def load_ipython_extension(ipython):
+    register_mimerenderer(ipython, "image/png", imgcat_factory(fn="img.png"))
+    register_mimerenderer(ipython, "image/jpeg", imgcat_factory(fn="img.jpg"))