Try_except with warning

trx/io.py

@@ -4,6 +4,7 @@
 import os
 import logging
 import tempfile
+import sys
 
 try:
     import dipy
@@ -23,7 +24,11 @@ def get_trx_tmpdir():
     else:
         trx_tmp_dir = tempfile.gettempdir()
 
-    return tempfile.TemporaryDirectory(dir=trx_tmp_dir, prefix='trx_')
+    if sys.version_info[1] >= 10:
+        return tempfile.TemporaryDirectory(dir=trx_tmp_dir, prefix='trx_',
+                                           ignore_cleanup_errors=True)
+    else:
+        return tempfile.TemporaryDirectory(dir=trx_tmp_dir, prefix='trx_')
 
 
 def load_sft_with_reference(filepath, reference=None,
@@ -72,7 +77,7 @@ def load(tractogram_filename, reference):
 def save(tractogram_obj, tractogram_filename, bbox_valid_check=False):
     if not dipy_available:
         logging.error('Dipy library is missing, cannot use functions related '
-                    'to the StatefulTractogram.')
+                      'to the StatefulTractogram.')
         return None
     from dipy.io.stateful_tractogram import StatefulTractogram
     from dipy.io.streamline import save_tractogram

trx/trx_file_memmap.py

@@ -1736,24 +1736,10 @@ def to_sft(self, resize=False):
     def close(self) -> None:
         """Cleanup on-disk temporary folder and initialize an empty TrxFile"""
         if self._uncompressed_folder_handle is not None:
-            # Problem on Windows that is solving only in Python above 3.10
-            # Step 1: Traverse the directory tree
-            for dirpath, dirnames, filenames in \
-                os.walk(self._uncompressed_folder_handle.name, topdown=False):
-
-                # Step 2 and 3: Identify and remove symlinks (for files)
-                for filename in filenames:
-                    filepath = os.path.join(dirpath, filename)
-                    if os.path.islink(filepath):
-                        os.unlink(filepath)
-
-                # Step 2 and 3: Identify and remove symlinks (for directories)
-                for dirname in dirnames:
-                    dir_full_path = os.path.join(dirpath, dirname)
-                    if os.path.islink(dir_full_path):
-                        os.unlink(dir_full_path)
-
-            # Step 4: Remove the temporary directory
-            shutil.rmtree(self._uncompressed_folder_handle.name)
+            try:
+                self._uncompressed_folder_handle.cleanup()
+            except PermissionError:
+                logging.error("Windows PermissionError, temporary directory {}" +
+                              "was not deleted!".format(self._uncompressed_folder_handle.name))
         self.__init__()
         logging.debug("Deleted memmaps and intialized empty TrxFile.")