ksort commited on
Commit
ed38084
·
1 Parent(s): b21b3ae

update upload

Browse files
Files changed (1) hide show
  1. serve/upload.py +11 -6
serve/upload.py CHANGED
@@ -12,6 +12,11 @@ ssh_client = None
12
  sftp_client = None
13
  sftp_client_imgs = None
14
 
 
 
 
 
 
15
  def create_ssh_client(server, port, user, password):
16
  global ssh_client, sftp_client, sftp_client_imgs
17
  ssh_client = paramiko.SSHClient()
@@ -22,12 +27,12 @@ def create_ssh_client(server, port, user, password):
22
  transport = ssh_client.get_transport()
23
  transport.set_keepalive(60)
24
 
25
- sftp_client = ssh_client.open_sftp()
26
- sftp_client_1 = ssh_client.open_sftp()
27
- sftp_client_2 = ssh_client.open_sftp()
28
- sftp_client_3 = ssh_client.open_sftp()
29
- sftp_client_4 = ssh_client.open_sftp()
30
- sftp_client_imgs = [sftp_client_1, sftp_client_2, sftp_client_3, sftp_client_4]
31
 
32
 
33
  def is_connected():
 
12
  sftp_client = None
13
  sftp_client_imgs = None
14
 
15
+ def open_sftp():
16
+ global ssh_client
17
+ sftp_client = ssh_client.open_sftp()
18
+ return sftp_client
19
+
20
  def create_ssh_client(server, port, user, password):
21
  global ssh_client, sftp_client, sftp_client_imgs
22
  ssh_client = paramiko.SSHClient()
 
27
  transport = ssh_client.get_transport()
28
  transport.set_keepalive(60)
29
 
30
+ with concurrent.futures.ThreadPoolExecutor() as executor:
31
+ futures = [executor.submit(ssh_client) for _ in range(5)]
32
+ results = [future.result() for future in futures]
33
+
34
+ sftp_client = results[0]
35
+ sftp_client_imgs = results[1:]
36
 
37
 
38
  def is_connected():