Avinashnalla7 commited on
Commit
667075f
·
1 Parent(s): f57da06

fix: add SFTP timeouts + prevent /api/pdf hang

Browse files
Files changed (1) hide show
  1. backend/sftp_store.py +4 -1
backend/sftp_store.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import posixpath
3
  import paramiko
4
 
@@ -23,7 +24,9 @@ def download_bytes(remote_path: str) -> bytes:
23
 
24
  rp = remote_path.lstrip("/")
25
 
26
- transport = paramiko.Transport((host, port))
 
 
27
  transport.connect(username=user, password=pw)
28
  sftp = paramiko.SFTPClient.from_transport(transport)
29
 
 
1
  import os
2
+ import socket
3
  import posixpath
4
  import paramiko
5
 
 
24
 
25
  rp = remote_path.lstrip("/")
26
 
27
+ transport = paramiko.Transport(socket.create_connection((host, port), timeout=10))
28
+ transport.banner_timeout = 10
29
+ transport.auth_timeout = 10
30
  transport.connect(username=user, password=pw)
31
  sftp = paramiko.SFTPClient.from_transport(transport)
32