Use tempfile.TemporaryDirectory()

openmc/data/neutron.py

@@ -840,10 +840,7 @@ class IncidentNeutron(EqualityMixin):
             Incident neutron continuous-energy data
 
         """
-        # Create temporary directory -- it would be preferable to use
-        # TemporaryDirectory(), but it is only available in Python 3.2
-        tmpdir = tempfile.mkdtemp()
-        try:
+        with tempfile.TemporaryDirectory() as tmpdir:
             # Run NJOY to create an ACE library
             ace_file = os.path.join(tmpdir, 'ace')
             xsdir_file = os.path.join(tmpdir, 'xsdir')
@@ -871,8 +868,4 @@ class IncidentNeutron(EqualityMixin):
                 data.energy['0K'] = xs.x
                 data[2].xs['0K'] = xs
 
-        finally:
-            # Get rid of temporary files
-            shutil.rmtree(tmpdir)
-
         return data

openmc/data/njoy.py

@@ -149,10 +149,7 @@ def run(commands, tapein, tapeout, input_filename=None, stdout=False,
         with open(input_filename, 'w') as f:
             f.write(commands)
 
-    # Create temporary directory -- it would be preferable to use
-    # TemporaryDirectory(), but it is only available in Python 3.2
-    tmpdir = tempfile.mkdtemp()
-    try:
+    with tempfile.TemporaryDirectory() as tmpdir:
         # Copy evaluations to appropriates 'tapes'
         for tape_num, filename in tapein.items():
             tmpfilename = os.path.join(tmpdir, 'tape{}'.format(tape_num))
@@ -186,8 +183,6 @@ def run(commands, tapein, tapeout, input_filename=None, stdout=False,
             tmpfilename = os.path.join(tmpdir, 'tape{}'.format(tape_num))
             if os.path.isfile(tmpfilename):
                 shutil.move(tmpfilename, filename)
-    finally:
-        shutil.rmtree(tmpdir)
 
 
 def make_pendf(filename, pendf='pendf', error=0.001, stdout=False):

openmc/data/thermal.py

@@ -623,10 +623,7 @@ class ThermalScattering(EqualityMixin):
             Thermal scattering data
 
         """
-        # Create temporary directory -- it would be preferable to use
-        # TemporaryDirectory(), but it is only available in Python 3.2
-        tmpdir = tempfile.mkdtemp()
-        try:
+        with tempfile.TemporaryDirectory() as tmpdir:
             # Run NJOY to create an ACE library
             ace_file = os.path.join(tmpdir, 'ace')
             xsdir_file = os.path.join(tmpdir, 'xsdir')
@@ -638,8 +635,5 @@ class ThermalScattering(EqualityMixin):
             data = cls.from_ace(lib.tables[0])
             for table in lib.tables[1:]:
                 data.add_temperature_from_ace(table)
-        finally:
-            # Get rid of temporary files
-            shutil.rmtree(tmpdir)
 
         return data