#135 added ability to run a python string

Author: Almenon

index.ts

@@ -3,7 +3,7 @@ import { ChildProcess,spawn, SpawnOptions, exec } from 'child_process';
 import {EOL as newline, tmpdir} from 'os';
 import {join, sep} from 'path'
 import {Readable,Writable} from 'stream'
-import { writeFile } from 'fs';
+import { writeFile, writeFileSync } from 'fs';
 
 function toArray<T>(source?:T|T[]):T[] {
     if (typeof source === 'undefined' || source === null) {
@@ -198,7 +198,7 @@ export class PythonShell extends EventEmitter{
 	 * @returns {Promise} rejects w/ stderr if syntax failure
 	 */
 	static async checkSyntax(code:string){
-        let randomInt = Math.floor(Math.random()*10000000000);
+        let randomInt = PythonShell.getRandomInt();
         let filePath = tmpdir + sep + `pythonShellSyntaxCheck${randomInt}.py`
 
         // todo: replace this with util.promisify (once we no longer support node v7)
@@ -246,6 +246,23 @@ export class PythonShell extends EventEmitter{
         });
     };
 
+    /**
+     * Runs the inputted string of python code and returns collected messages. DO NOT ALLOW UNTRUSTED USER INPUT HERE!
+     * @param  {string}   code   The python code to execute
+     * @param  {Options}   options  The execution options
+     * @param  {Function} callback The callback function to invoke with the script results
+     * @return {PythonShell}       The PythonShell instance
+     */
+    static runString(code:string, options?:Options, callback?:(err:PythonShellError, output?:any[])=>any) {
+
+        // put code in temp file
+        let randomInt = PythonShell.getRandomInt();
+        let filePath = tmpdir + sep + `pythonShellFile${randomInt}.py`
+        writeFileSync(filePath, code);
+
+        return PythonShell.run(filePath, options, callback);
+    };
+
     /**
      * Parses an error thrown from the Python process through stderr
      * @param  {string|Buffer} data The stderr contents to parse
@@ -272,6 +289,13 @@ export class PythonShell extends EventEmitter{
         return error;
     };
 
+    /**
+     * gets a random int from 0-10000000000
+     */
+    private static getRandomInt(){
+        return Math.floor(Math.random()*10000000000);
+    }
+
     /**
      * Sends a message to the Python shell through stdin
      * Override this method to format data to be sent to the Python process

test/test-python-shell.ts

@@ -54,6 +54,26 @@ describe('PythonShell', function () {
         })
     })
 
+    describe('#runString(script, options)', function () {
+        before(()=>{
+            PythonShell.defaultOptions = {};
+        })
+        it('should be able to execute a string of python code', function (done) {
+            PythonShell.runString('print("hello");print("world")', null, function (err, results) {
+                if (err) return done(err);
+                results.should.be.an.Array().and.have.lengthOf(2);
+                results.should.eql(['hello', 'world']);
+                done();
+            });
+        });
+        after(()=>{
+            PythonShell.defaultOptions = {
+                // reset to match initial value
+                scriptPath: './test/python'
+            };
+        })
+    });
+
     describe('#run(script, options)', function () {
         it('should run the script and return output data', function (done) {
             PythonShell.run('echo_args.py', {