Merge pull request #5039 from megies/sphinxext_plot_directive_switch_backend

mdboom · tacaswell · commit b34682c35637 · 2015-10-08T12:48:19.000-04:00
sphinxext pot_directive: more robust backend switching
diff --git a/lib/matplotlib/sphinxext/plot_directive.py b/lib/matplotlib/sphinxext/plot_directive.py
@@ -139,6 +139,7 @@
 import sys, os, shutil, io, re, textwrap
 from os.path import relpath
 import traceback
+import warnings
 
 if not six.PY3:
     import cStringIO
@@ -166,8 +167,15 @@ def format_template(template, **kw):
 
 import matplotlib
 import matplotlib.cbook as cbook
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
+try:
+    with warnings.catch_warnings(record=True):
+        warnings.simplefilter("error", UserWarning)
+        matplotlib.use('Agg')
+except UserWarning:
+    import matplotlib.pyplot as plt
+    plt.switch_backend("Agg")
+else:
+    import matplotlib.pyplot as plt
 from matplotlib import _pylab_helpers
 
 __version__ = 2
