VultureChecker/VultureCheckerService.py

changeset 47
4e19b213fc3f
parent 45
232ed15782bb
child 49
7b55dc5d09d2
diff -r 9f7a962f3655 -r 4e19b213fc3f VultureChecker/VultureCheckerService.py
--- a/VultureChecker/VultureCheckerService.py	Mon Mar 27 19:51:21 2017 +0200
+++ b/VultureChecker/VultureCheckerService.py	Fri Mar 31 17:59:28 2017 +0200
@@ -14,8 +14,9 @@
 except NameError:
     pass
 
+import sys
 import multiprocessing
-import sys
+import queue
 
 from vulture import Vulture
 
@@ -89,11 +90,25 @@
     # Get and send results
     endIndex = len(argumentsList) - initialTasks
     for i in range(len(argumentsList)):
-        filename, result = doneQueue.get()
-        send(fx, filename, result)
-        if cancelled():
+        resultSent = False
+        wasCancelled = False
+        
+        while not resultSent:
+            try:
+                # get result (waiting max. 3 seconds and send it to frontend
+                filename, result = doneQueue.get()
+                send(fx, filename, result)
+                resultSent = True
+            except queue.Empty:
+                # ignore empty queue, just carry on
+                if cancelled():
+                    wasCancelled = True
+                    break
+        
+        if wasCancelled or cancelled():
             # just exit the loop ignoring the results of queued tasks
             break
+        
         if i < endIndex:
             taskQueue.put(argumentsList[i + initialTasks])
 

eric ide

mercurial