launch multiprocessing programs without in python mode

Author: fabianp

examples/multiprocessing_example.py

@@ -2,7 +2,7 @@
 An undecorated example of a script that allocates memory in multiprocessing
 workers to demonstrate the use of memory_profiler with multiple processes.
 
-Run this script with mprof run -M python multiprocessing_example.py
+Run this script with mprof run -M multiprocessing_example.py
 You can then visualize the usage with mprof plot.
 """
 

mprof

@@ -219,7 +219,15 @@ def run_action():
 
     # .. TODO: more than one script as argument ? ..
     if args[0].endswith('.py') and not options.nopython:
-        options.python = True
+        if options.multiprocess and not args[0].startswith("python"):
+            # in multiprocessing mode you want to spawn a separate
+            # python process
+            args.insert(0, "python")
+        else:
+            options.python = True
+    if options.multiprocess:
+        # multiprocess can't work in python mode
+        options.python = False
     if options.python:
         print("running as a Python program...")
         if not args[0].startswith("python"):