81 lines
2.0 KiB
Python
81 lines
2.0 KiB
Python
import sys
|
|
import threading
|
|
|
|
|
|
# https://stackoverflow.com/a/65447493/6543759
|
|
class ThreadWithResult(threading.Thread):
|
|
def __init__(
|
|
self, group=None, target=None, name=None, args=(), kwargs=None, *, daemon=None
|
|
):
|
|
self.exc = None
|
|
if not kwargs:
|
|
kwargs = {}
|
|
|
|
def function():
|
|
self.exc = None
|
|
try:
|
|
self.result = target(*args, **kwargs)
|
|
except: # noqa
|
|
# Save details of the exception thrown but don't rethrow,
|
|
# just complete the function
|
|
self.exc = sys.exc_info()
|
|
|
|
super().__init__(group=group, target=function, name=name, daemon=daemon)
|
|
|
|
# https://stackoverflow.com/a/12223550/6543759
|
|
def join(self, *args, **kwargs):
|
|
super().join(*args, **kwargs)
|
|
|
|
if self.exc:
|
|
msg = "Thread '%s' threw an exception: %s" % (self.getName(), self.exc[1])
|
|
new_exc = Exception(msg)
|
|
raise new_exc.with_traceback(self.exc[2])
|
|
|
|
|
|
THREADS = {}
|
|
THREAD_ID = 0
|
|
|
|
|
|
def call_slow_function(function, args):
|
|
global THREADS, THREAD_ID
|
|
thread = ThreadWithResult(target=function, args=args, daemon=True)
|
|
THREAD_ID += 1
|
|
THREADS[THREAD_ID] = thread
|
|
thread.start()
|
|
|
|
return THREAD_ID
|
|
|
|
|
|
def has_call_finished(thread_id):
|
|
global THREADS
|
|
|
|
thread = THREADS[thread_id]
|
|
if thread.is_alive():
|
|
# Thread is still working
|
|
return False
|
|
|
|
# Thread has finished, we can return its value using get_call_value()
|
|
return True
|
|
|
|
|
|
def get_call_value(thread_id):
|
|
global THREADS
|
|
|
|
thread = THREADS[thread_id]
|
|
if thread.is_alive():
|
|
# Thread is still working
|
|
raise ValueError("Thread is still running!")
|
|
|
|
# Thread has finished, we can return its value now
|
|
try:
|
|
thread.join()
|
|
finally:
|
|
del THREADS[thread_id]
|
|
|
|
try:
|
|
return thread.result
|
|
except AttributeError:
|
|
raise RuntimeError(
|
|
'The thread does not have the "result" attribute. An unhandled error occurred inside your Thread'
|
|
)
|