BF: Before Experiment was not being written for JS (fixes #3115)

psychopy · Sep 4, 2020 · bbfd310 · peircej · Sep 6, 2020 · kjj11033
1 parent 0650307
commit bbfd310
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/psychopy/experiment/_experiment.py b/psychopy/experiment/_experiment.py
@@ -223,6 +223,8 @@ def writeScript(self, expPath=None, target="PsychoPy", modular=True):
                 self_copy._currentRoutine = entry
                 if hasattr(entry, 'writeInitCodeJS'):
                     entry.writeInitCodeJS(script)
+                if hasattr(entry, 'writePreCodeJS'):
+                    entry.writePreCode(script)
 
             # create globalClock etc
             code = ("// Create some handy timers\n"

diff --git a/psychopy/experiment/components/code/__init__.py b/psychopy/experiment/components/code/__init__.py
@@ -184,6 +184,11 @@ def writePreCode(self, buff):
             alerttools.checkPythonSyntax(self, 'Before Experiment')
             buff.writeIndentedLines(str(self.params['Before Experiment']) + '\n')
 
+    def writePreCodeJS(self, buff):
+        if len(str(self.params['Before JS Experiment'])):
+            alerttools.checkPythonSyntax(self, 'Before Experiment')
+            buff.writeIndentedLines(str(self.params['Before JS Experiment']) + '\n')
+
     def writeInitCode(self, buff):
         if len(str(self.params['Begin Experiment'])):
             alerttools.checkPythonSyntax(self, 'Begin Experiment')