Adding extra check to make sure running process stops

Author: mklomp

mirte_robot/linetrace.py

@@ -16,6 +16,7 @@
 # Global shared memory objects (TODO: check if we need shared memory, why is server working?)
 stepper = multiprocessing.Value('b', True)
 do_step = multiprocessing.Value('b', False)
+is_running = False
 
 def stop_mirte():
      process.terminate()
@@ -65,25 +66,32 @@ def traceit(frame, event, arg):
 
 def start_mirte():
     global process
+    # process should already have been killed after stop, or disconnect
+    # but, just in case make sure to stop this
+    if process.is_alive():
+      process.terminate()
     process = multiprocessing.Process(target = load_mirte_module, args=(stepper, do_step))
     process.start()
 
 def client_left(client, server):
     stop_mirte()
 
 def message_received(client, server, message):
-   global stepper, do_step
+   global stepper, do_step, is_running
+
    if message == "b": #break (pause)
       stepper.value = True
    if message == "c": #continue (play)
       stepper.value = False
-      if not process.is_alive():
+      if not is_running:
          start_mirte()
+         is_running = True
    if message == "s": #step (step)
       do_step.value = True
    if message == "e": #exit (stop)
       stepper.value = True
       do_step.value = False
+      is_running = False
       stop_mirte()
 
 server = WebsocketServer(host="0.0.0.0", port=8001, loglevel=logging.CRITICAL)