Spaces:
Sleeping
Sleeping
Commit ·
667075f
1
Parent(s): f57da06
fix: add SFTP timeouts + prevent /api/pdf hang
Browse files- 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 |
|