update upload

serve/upload.py

@@ -10,6 +10,7 @@ from .constants import SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD, SSH_LOG, SS
 
 ssh_client = None
 sftp_client = None
+sftp_client_imgs = None
 
 def create_ssh_client(server, port, user, password):
     global ssh_client, sftp_client
@@ -22,6 +23,13 @@ def create_ssh_client(server, port, user, password):
     transport.set_keepalive(60)
 
     sftp_client = ssh_client.open_sftp()
+    sftp_client_1 = ssh_client.open_sftp()
+    sftp_client_2 = ssh_client.open_sftp()
+    sftp_client_3 = ssh_client.open_sftp()
+    sftp_client_4 = ssh_client.open_sftp()
+    sftp_client_imgs = [sftp_client_1, sftp_client_2, sftp_client_3, sftp_client_4]
+
+
 def is_connected():
     global ssh_client, sftp_client
     if ssh_client is None or sftp_client is None:
@@ -95,6 +103,7 @@ def upload_images(i, image_list, output_file_list, sftp_client):
     
 def upload_ssh_all(states, output_dir, data, data_path):
     global sftp_client
+    global sftp_client_imgs
     if not is_connected():
         create_ssh_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
     output_file_list = []
@@ -106,7 +115,7 @@ def upload_ssh_all(states, output_dir, data, data_path):
 
 
     with concurrent.futures.ThreadPoolExecutor() as executor:
-        futures = [executor.submit(upload_images, i, image_list, output_file_list, sftp_client) for i in range(len(output_file_list))]
+        futures = [executor.submit(upload_images, i, image_list, output_file_list, sftp_client_imgs[i]) for i in range(len(output_file_list))]
 
     # for i in range(len(output_file_list)):
     #     if isinstance(image_list[i], str):