owpythonscript: also store the script text

Author: markotoplak

Orange/widgets/data/owpythonscript.py

@@ -392,6 +392,7 @@ class Outputs:
     libraryListSource = \
         Setting([Script("Hello world", "print('Hello world')\n")])
     currentScriptIndex = Setting(0)
+    scriptText = Setting(None, schema_only=True)
     splitterState = Setting(None)
 
     class Error(OWWidget.Error):
@@ -516,6 +517,9 @@ def __init__(self):
 
         select_row(self.libraryView, self.currentScriptIndex)
 
+        self.restoreScriptText()
+        self.text.textChanged.connect(self.saveScriptText)  # after restoring
+
         self.splitCanvas.setSizes([2, 1])
         if self.splitterState is not None:
             self.splitCanvas.restoreState(QByteArray(self.splitterState))
@@ -524,6 +528,16 @@ def __init__(self):
         self.controlArea.layout().addStretch(1)
         self.resize(800, 600)
 
+    def restoreScriptText(self):
+        if self.scriptText is not None:
+            current = self.text.toPlainText()
+            # do not mark scripts as modified
+            if self.scriptText != current:
+                self.text.document().setPlainText(self.scriptText)
+
+    def saveScriptText(self):
+        self.scriptText = self.text.toPlainText()
+
     def handle_input(self, obj, id, signal):
         id = id[0]
         dic = getattr(self, signal)

Orange/widgets/data/tests/test_owpythonscript.py

@@ -124,3 +124,13 @@ def test_restore_from_library(self):
         self.widget.restoreSaved()
         script = self.widget.text.toPlainText()
         self.assertEqual(before, script)
+
+    def test_store_current_script(self):
+        self.widget.text.setPlainText("42")
+        settings = self.widget.settingsHandler.pack_data(self.widget)
+        self.widget = self.create_widget(OWPythonScript)
+        script = self.widget.text.toPlainText()
+        self.assertNotEqual("42", script)
+        self.widget = self.create_widget(OWPythonScript, stored_settings=settings)
+        script = self.widget.text.toPlainText()
+        self.assertEqual("42", script)