Add temporary directory context manager

Author: morganjwilliams

pyrolite/util/general.py

@@ -1,3 +1,4 @@
+import contextlib
 import datetime
 import operator
 import os
@@ -52,6 +53,14 @@ def temp_path(suffix=""):
     return Path(directory)
 
 
+@contextlib.contextmanager
+def tempdir(**kwargs):
+    tmp_dir = Path(mkdtemp(**kwargs))
+    try:
+        yield tmp_dir
+    finally:
+        remove_tempdir(tmp_dir)
+
 def flatten_dict(d, climb=False, safemode=False):
     """
     Flattens a nested dictionary containing only string keys.