TemporaryFileWrapper: cache the value of os.unlink for use by __del__,

to prevent mysterious errors at shutdown due to "os.unlink" turning into
"None.unlink".

1 files changed, 8 insertions, 1 deletions

diff --git a/Lib/tempfile.py b/Lib/tempfile.py
index 96245d8..d488063 100644
--- a/Lib/tempfile.py
+++ b/Lib/tempfile.py
@@ -128,6 +128,13 @@ class TemporaryFileWrapper:
     In particular, it seeks to automatically remove the file when it is
     no longer needed.
     """
+
+    # Cache the unlinker so we don't get spurious errors at shutdown
+    # when the module-level "os" in None'd out.  Note that this must
+    # be referenced as self.unlink, because the name TemporaryFileWrapper
+    # may also get None'd out before __del__ is called.
+    unlink = os.unlink
+
     def __init__(self, file, path):
         self.file = file
         self.path = path
@@ -137,7 +144,7 @@ class TemporaryFileWrapper:
         if not self.close_called:
             self.close_called = 1
             self.file.close()
-            os.unlink(self.path)
+            self.unlink(self.path)
 
     def __del__(self):
         self.close()