Code to rename a group of files while keeping them locked

src/calibre/utils/copy_files.py

@@ -33,6 +33,10 @@ class UnixFileCopier:
     def __exit__(self, *a) -> None:
         pass
 
+    def rename_all(self) -> None:
+        for src_path, dest_path in self.copy_map.items():
+            os.replace(src_path, dest_path)
+
     def copy_all(self) -> None:
         for src_path, dest_path in self.copy_map.items():
             with suppress(OSError):
@@ -114,6 +118,10 @@ class WindowsFileCopier:
                     f.write(raw)
             shutil.copystat(src_path, dest_path, follow_symlinks=False)
 
+    def rename_all(self) -> None:
+        for src_path, dest_path in self.copy_map.items():
+            winutil.move_file(src_path, dest_path)
+
     def delete_all_source_files(self) -> None:
         for src_path in self.copy_map:
             winutil.delete_file(make_long_path_useable(src_path))
@@ -123,6 +131,15 @@ def get_copier() -> Union[UnixFileCopier | WindowsFileCopier]:
     return WindowsFileCopier() if iswindows else UnixFileCopier()
 
 
+def rename_files(src_to_dest_map: Dict[str, str]) -> None:
+    ' Rename a bunch of files. On Windows all files are locked before renaming so no other process can interfere. '
+    copier = get_copier()
+    for s, d in src_to_dest_map.items():
+        copier.register(s, d)
+    with copier:
+        copier.rename_all()
+
+
 def copy_files(src_to_dest_map: Dict[str, str], delete_source: bool = False) -> None:
     copier = get_copier()
     for s, d in src_to_dest_map.items():

src/calibre/utils/copy_files_test.py

@@ -10,7 +10,7 @@ import unittest
 from calibre import walk
 from calibre.constants import iswindows
 
-from .copy_files import copy_tree
+from .copy_files import copy_tree, rename_files
 from .filenames import nlinks_file
 
 
@@ -52,6 +52,15 @@ class TestCopyFiles(unittest.TestCase):
         self.tearDown()
         self.setUp()
 
+    def test_renaming_of_files(self):
+        for name in 'one two'.split():
+            with open(os.path.join(self.tdir, name), 'w') as f:
+                f.write(name)
+        renames = {os.path.join(self.tdir, k): os.path.join(self.tdir, v) for k, v in {'one': 'One', 'two': 'three'}.items()}
+        rename_files(renames)
+        contents = set(os.listdir(self.tdir)) - {'base', 'src'}
+        self.ae(contents, {'One', 'three'})
+
     def test_copying_of_trees(self):
         src, dest = self.s(), self.d()
         copy_tree(src, dest)