OWPythonScript: dropping and pasting of python scripts

Author: markotoplak

Orange/widgets/data/owpythonscript.py

@@ -9,14 +9,15 @@
 from AnyQt.QtWidgets import (
     QPlainTextEdit, QListView, QSizePolicy, QMenu, QSplitter, QLineEdit,
     QAction, QToolButton, QFileDialog, QStyledItemDelegate,
-    QStyleOptionViewItem, QPlainTextDocumentLayout
-)
+    QStyleOptionViewItem, QPlainTextDocumentLayout,
+    QMessageBox)
 from AnyQt.QtGui import (
     QColor, QBrush, QPalette, QFont, QTextDocument,
     QSyntaxHighlighter, QTextCharFormat, QTextCursor, QKeySequence,
 )
 from AnyQt.QtCore import Qt, QRegExp, QByteArray, QItemSelectionModel
 
+from Orange.canvas.gui.utils import OSX_NSURL_toLocalFile
 from Orange.data import Table
 from Orange.base import Learner, Model
 from Orange.widgets import widget, gui
@@ -35,6 +36,41 @@ def text_format(foreground=Qt.black, weight=QFont.Normal):
     return fmt
 
 
+def to_local_file(url):
+    return OSX_NSURL_toLocalFile(url) or url.toLocalFile()
+
+
+def read_file_content(url, limit=None):
+    filename = to_local_file(url)
+    try:
+        with open(filename, encoding="utf-8", errors='strict') as f:
+            text = f.read(limit)
+            return text
+    except (OSError, UnicodeDecodeError):
+        return None
+
+
+def dialog_replace_append(filename):
+    dlg = QMessageBox(
+        QMessageBox.Information,
+        "Add script",
+        "Adding script from {}\n\n"
+        "Replace the current script or append to it?".format(filename))
+    role = QMessageBox.AcceptRole
+    replace = dlg.addButton("Replace", role)
+    append = dlg.addButton("Append", role)
+    cancel = dlg.addButton("Cancel", role)
+    dlg.setEscapeButton(cancel)
+    dlg.setDefaultButton(replace)
+    dlg.exec()
+    if dlg.clickedButton() == replace:
+        return True
+    elif dlg.clickedButton() == append:
+        return False
+    else:
+        return None
+
+
 class PythonSyntaxHighlighter(QSyntaxHighlighter):
     def __init__(self, parent=None):
 
@@ -138,6 +174,35 @@ def keyPressEvent(self, event):
         else:
             super().keyPressEvent(event)
 
+    def insertFromMimeData(self, source):
+        """
+        Reimplemented from QPlainTextEdit.insertFromMimeData.
+        """
+        urls = source.urls()
+        if urls:
+            self.pasteFile(urls[0])
+        else:
+            super().insertFromMimeData(source)
+
+    def pasteFile(self, url):
+        new = read_file_content(url)
+        if new:
+            filename = os.path.basename(to_local_file(url))
+            replace = True
+            current_text = self.toPlainText()
+            if current_text.strip():  # only ask if there is current script
+                replace = dialog_replace_append(filename)
+            if replace is None:  # cancel
+                return
+            if replace:  # replace
+                self.document().setPlainText(new)
+            else:  # append
+                self.document().setPlainText(
+                    current_text +
+                    ("\n" if not current_text.endswith("\n") else "") +
+                    new)
+            self.document().setModified(True)
+
 
 class PythonConsole(QPlainTextEdit, code.InteractiveConsole):
     def __init__(self, locals=None, parent=None):
@@ -533,6 +598,8 @@ def __init__(self):
         if self.splitterState is not None:
             self.splitCanvas.restoreState(QByteArray(self.splitterState))
 
+        self.setAcceptDrops(True)
+
         self.splitCanvas.splitterMoved[int, int].connect(self.onSpliterMoved)
         self.controlArea.layout().addStretch(1)
         self.resize(800, 600)
@@ -720,6 +787,20 @@ def commit(self):
                 out_var = None
             getattr(self.Outputs, signal).send(out_var)
 
+    def dragEnterEvent(self, event):
+        urls = event.mimeData().urls()
+        if urls:
+            # try reading the file as text
+            c = read_file_content(urls[0], limit=1000)
+            if c is not None:
+                event.acceptProposedAction()
+
+    def dropEvent(self, event):
+        """Handle file drops"""
+        urls = event.mimeData().urls()
+        if urls:
+            self.text.pasteFile(urls[0])
+
 
 if __name__ == "__main__":  # pragma: no cover
     WidgetPreview(OWPythonScript).run()