fix: stop process bug

Author: cnstark

task/models.py

@@ -99,7 +99,7 @@ def __str__(self):
         return self.task.name + '-' + str(self.index)
 
     def kill(self):
-        os.kill(self.pid, signal.SIGKILL)
+        os.kill(self.pid, signal.SIGINT)
 
     def delete_log_file(self):
         if os.path.isfile(self.log_file_path):

task/utils.py

@@ -15,9 +15,9 @@
 def generate_ssh_cmd(host, user, exec_cmd, private_key_path=None):
     exec_cmd = exec_cmd.replace('$', '\\$')
     if private_key_path is None:
-        cmd = "ssh -o StrictHostKeyChecking=no {}@{} \"{}\"".format(user, host, exec_cmd)
+        cmd = "ssh -tt -o StrictHostKeyChecking=no {}@{} \"{}\"".format(user, host, exec_cmd)
     else:
-        cmd = "ssh -o StrictHostKeyChecking=no -i {} {}@{} \"{}\"".format(private_key_path, user, host, exec_cmd)
+        cmd = "ssh -tt -o StrictHostKeyChecking=no -i {} {}@{} \"{}\"".format(private_key_path, user, host, exec_cmd)
     return cmd
 
 
@@ -28,7 +28,7 @@ def __init__(self, user, host, cmd, workspace="~", private_key_path=None, output
         if output_file is not None:
             self.output_file = output_file
             with open(self.output_file, "wb") as out:
-                self.proc = subprocess.Popen(self.cmd, shell=True, stdout=out, stderr=out, bufsize=1)
+                self.proc = subprocess.Popen(self.cmd, shell=True, stdin=subprocess.PIPE, stdout=out, stderr=out, bufsize=1)
         else:
             self.proc = subprocess.Popen(self.cmd, shell=True)
 
@@ -37,7 +37,8 @@ def pid(self):
 
     def kill(self):
         # os.killpg(os.getpgid(self.proc.pid), signal.SIGKILL)
-        os.kill(self.proc.pid, signal.SIGKILL)
+        # os.kill(self.proc.pid, signal.SIGKILL)
+        self.proc.send_signal(signal.SIGINT)
 
     def get_return_code(self):
         self.proc.wait()