Merge pull request #9869 from anntzer/clean-tmpdir

dstansby · web-flow · commit 045c951cfc41 · 2017-11-28T11:40:00.000Z
Clean tmpdir at exit.
diff --git a/lib/matplotlib/__init__.py b/lib/matplotlib/__init__.py
@@ -104,6 +104,7 @@
 
 import six
 
+import atexit
 from collections import MutableMapping
 import contextlib
 import distutils.version
@@ -116,6 +117,7 @@
 import logging
 import os
 import re
+import shutil
 import sys
 import tempfile
 import warnings
@@ -605,6 +607,7 @@ def _create_tmp_config_dir():
     """
     configdir = os.environ['MPLCONFIGDIR'] = (
         tempfile.mkdtemp(prefix='matplotlib-'))
+    atexit.register(shutil.rmtree, configdir)
     return configdir
 
 
