From 92b829869786b3d4b075bd6f868a7b164b5f2b0a Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Tue, 11 Jun 2019 14:39:18 +0300
Subject: [PATCH] bugfix, remote task stop from web-app

---
 trains/task.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/trains/task.py b/trains/task.py
index 5d4094ec..5c121b38 100644
--- a/trains/task.py
+++ b/trains/task.py
@@ -814,6 +814,7 @@ class Task(_Task):
             self._dev_worker.unregister()
 
         # NOTICE! This will end the entire execution tree!
+        self.__exit_hook.remote_user_aborted = True
         self._kill_all_child_processes(send_kill=False)
         time.sleep(2.0)
         self._kill_all_child_processes(send_kill=True)
@@ -908,12 +909,13 @@ class Task(_Task):
                     self._dev_worker.unregister()
                 # check if we crashed, ot the signal is not interrupt (manual break)
                 if self.__exit_hook:
-                    if self.__exit_hook.exception is not None or self.__exit_hook.signal not in (None, 2):
+                    if self.__exit_hook.exception is not None or \
+                            (not self.__exit_hook.remote_user_aborted and self.__exit_hook.signal not in (None, 2)):
                         self.mark_failed(status_reason='Exception')
                         wait_for_uploads = False
                     else:
                         self.stopped()
-                        wait_for_uploads = (self.__exit_hook.signal is None)
+                        wait_for_uploads = (self.__exit_hook.remote_user_aborted or self.__exit_hook.signal is None)
                 else:
                     self.stopped()
 
@@ -941,6 +943,7 @@ class Task(_Task):
         class ExitHooks(object):
             _orig_exit = None
             _orig_exc_handler = None
+            remote_user_aborted = False
 
             def __init__(self, callback):
                 self.exit_code = None