Drag and drop files in and out of TagStudio

tagstudio/src/qt/modals/drop_import.py

@@ -0,0 +1,82 @@
+from pathlib import Path
+import shutil
+import typing
+
+from PySide6.QtGui import QDropEvent, QDragEnterEvent, QDragMoveEvent
+from PySide6.QtWidgets import QMessageBox
+
+if typing.TYPE_CHECKING:
+    from src.qt.ts_qt import QtDriver
+
+
+def dropEvent(driver: "QtDriver", event: QDropEvent):
+    if not event.mimeData().hasUrls():
+        return
+
+    if not event.mimeData().urls()[0].isLocalFile():
+        return
+
+    urls = event.mimeData().urls()
+    duplicate_filesnames = []
+    for url in urls:
+        if Path(driver.lib.library_dir + "/" + url.fileName()).exists():
+            duplicate_filesnames.append(url)
+
+    ret = -1
+
+    if len(duplicate_filesnames) > 0:
+        msgBox = QMessageBox()
+        msgBox.setText(
+            f"The files  {", ".join(map(lambda url:url.fileName(),duplicate_filesnames))}  have filenames that already exist in the library folder."
+        )
+        msgBox.addButton("Skip", QMessageBox.ButtonRole.YesRole)
+        msgBox.addButton("Override", QMessageBox.ButtonRole.DestructiveRole)
+        msgBox.addButton("Rename", QMessageBox.ButtonRole.DestructiveRole)
+        msgBox.addButton("Cancel", QMessageBox.ButtonRole.NoRole)
+        ret = msgBox.exec()
+
+        if ret == 3:  # cancel
+            return
+
+    for url in urls:
+        filename = url.fileName()
+
+        if url in duplicate_filesnames:
+            if ret == 0:  # skip duplicates
+                continue
+
+            if ret == 2:  # rename
+                filename = get_renamed_duplicate_filename(
+                    driver.lib.library_dir, filename
+                )
+                driver.lib.files_not_in_library.append(filename)
+        else:  # override is simply copying but not adding a new entry
+            driver.lib.files_not_in_library.append(filename)
+
+        shutil.copyfile(url.toLocalFile(), driver.lib.library_dir + "/" + filename)
+
+    driver.add_new_files_runnable()
+
+
+def dragEnterEvent(event: QDragEnterEvent):
+    if event.mimeData().hasUrls():
+        event.accept()
+    else:
+        event.ignore()
+
+
+def dragMoveEvent(event: QDragMoveEvent):
+    if event.mimeData().hasUrls():
+        event.accept()
+    else:
+        event.ignore()
+
+
+def get_renamed_duplicate_filename(path, filename) -> str:
+    index = 2
+    o_filename = filename
+    dot_idx = o_filename.index(".")
+    while Path(path + "/" + filename).exists():
+        filename = o_filename[:dot_idx] + f" ({index})" + o_filename[dot_idx:]
+        index += 1
+    return filename

tagstudio/src/qt/ts_qt.py

@@ -89,6 +89,7 @@
     FixUnlinkedEntriesModal,
     FixDupeFilesModal,
     FoldersToTagsModal,
+    drop_import,
 )
 import src.qt.resources_rc
 
@@ -508,6 +509,11 @@ def start(self):
         # self.render_times: list = []
         # self.main_window.setWindowFlag(Qt.FramelessWindowHint)
 
+        self.main_window.setAcceptDrops(True)
+        self.main_window.dragEnterEvent = drop_import.dragEnterEvent
+        self.main_window.dropEvent = lambda event: drop_import.dropEvent(self, event)
+        self.main_window.dragMoveEvent = drop_import.dragMoveEvent
+
         # NOTE: Putting this early will result in a white non-responsive
         # window until everything is loaded. Consider adding a splash screen
         # or implementing some clever loading tricks.