v1 milestone

venv/lib/python3.12/site-packages/yt_dlp/postprocessor/movefilesafterdownload.py

@@ -0,0 +1,50 @@
+import os
+
+from .common import PostProcessor
+from ..compat import shutil
+from ..utils import (
+    PostProcessingError,
+    make_dir,
+)
+
+
+class MoveFilesAfterDownloadPP(PostProcessor):
+
+    def __init__(self, downloader=None, downloaded=True):
+        PostProcessor.__init__(self, downloader)
+        self._downloaded = downloaded
+
+    @classmethod
+    def pp_key(cls):
+        return 'MoveFiles'
+
+    def run(self, info):
+        dl_path, dl_name = os.path.split(info['filepath'])
+        finaldir = info.get('__finaldir', dl_path)
+        finalpath = os.path.join(finaldir, dl_name)
+        if self._downloaded:
+            info['__files_to_move'][info['filepath']] = finalpath
+
+        make_newfilename = lambda old: os.path.join(finaldir, os.path.basename(old))
+        for oldfile, newfile in info['__files_to_move'].items():
+            if not newfile:
+                newfile = make_newfilename(oldfile)
+            if os.path.abspath(oldfile) == os.path.abspath(newfile):
+                continue
+            if not os.path.exists(oldfile):
+                self.report_warning(f'File "{oldfile}" cannot be found')
+                continue
+            if os.path.exists(newfile):
+                if self.get_param('overwrites', True):
+                    self.report_warning(f'Replacing existing file "{newfile}"')
+                    os.remove(newfile)
+                else:
+                    self.report_warning(
+                        f'Cannot move file "{oldfile}" out of temporary directory since "{newfile}" already exists. ')
+                    continue
+            make_dir(newfile, PostProcessingError)
+            self.to_screen(f'Moving file "{oldfile}" to "{newfile}"')
+            shutil.move(oldfile, newfile)  # os.rename cannot move between volumes
+
+        info['filepath'] = finalpath
+        return [], info