diff --git a/mt.py b/mt.py index 763bf1c..84b9868 100644 --- a/mt.py +++ b/mt.py @@ -26,10 +26,12 @@ class mt(): self.__results = manager.list() self.__progress = mp.Value("i", 0) self.__lock = mp.Lock() + self.__keepalive = manager.list([0]) else: self.__results = [] self.__progress = 0 self.__lock = th.Lock() + self.__keepalive = [0] self.__progressinterval = progressinterval for i in range(self.__threadcount): self.__results.append([]) @@ -70,6 +72,7 @@ class mt(): self.__inc_progress(len(part)) if self.__comment is not None: print("\r" + self.__comment + ": " + str(self.__get_progress()) + "/" + str(len(self.__data)) + " ...", end='', flush=True) + self.__keepalive[0] = 0 with self.__lock: self.__results[i] = results @@ -114,6 +117,7 @@ class mt(): self.__inc_progress(len(part)) if self.__comment is not None: print("\r" + self.__comment + ": " + str(self.__get_progress()) + "/" + str(len(self.__data)) + " ...", end='', flush=True) + self.__keepalive[0] = 0 with self.__lock: self.__results[i] = results @@ -160,6 +164,7 @@ class mt(): self.__inc_progress(len(part)) if self.__comment is not None: print("\r" + self.__comment + ": " + str(self.__get_progress()) + "/" + str(len(self.__data)) + " ...", end='', flush=True) + self.__keepalive[0] = 0 with self.__lock: self.__results[i] = val