如何在 Python 中使用信号模块或其他模块使 Paramiko sftp.put() 超时？

Question

我想要超时功能sftp.put()，我已经尝试使用信号模块，但如果上传时间超过 10 秒，脚本不会死掉。我用它通过 ssh (paramiko) 传输文件。

[...]
def handler(signum, frame):
        print 'Signal handler called with signal', signum
        raise IOError("Couldn't upload the fileeeeeeeeeeee!!!!")

[...]
raspi = paramiko.SSHClient()
raspi.set_missing_host_key_policy(paramiko.AutoAddPolicy())
raspi.connect(ip , username= "", password= "" , timeout=10)
sftp = raspi.open_sftp()        

[...]   
signal.signal(signal.SIGALRM, handler)
signal.alarm(10)

sftp.put(source, destination , callback=None, confirm=True)  
    
signal.alarm(0)
raspi.close()

[...]

更新 1：

如果服务器停止响应一段时间，我想中止传输。实际上，我的 python 脚本检查（循环）文件夹中的任何文件，并将其发送到此远程服务器。但在这里的问题是，我想在传输过程中服务器突然无法访问的情况下保留此功能（IP 服务器更改，不再有互联网，...）。但是当我模拟断开连接时，脚本仍然停留在这个函数 sftp.put 无论如何...）

更新二：

当服务器在传输过程中脱机时，put() 似乎永远被阻止。此行也会发生这种情况：

sftp.get_channel().settimeout(xx)

失去频道怎么办？

更新 3 & 脚本目标

Ubuntu 18.04 和 paramiko 版本 2.6.0

您好，为了遵循您的评论和问题，我必须提供有关我非常丑陋的脚本的更多详细信息，对此深表歉意:)

实际上，我不想手动杀死一个线程并打开一个新线程。对于我的应用程序，我希望脚本运行完全自主，如果在此过程中出现问题，它仍然可以继续。为此，我使用 Python 异常处理。一切都按照我的意愿进行，除非远程服务器在传输过程中关闭：脚本在 put() 函数中保持阻塞状态，我认为在循环中。下面，由于您的帮助，脚本总共包含 3 个超时函数，但显然没有什么可以离开这个该死的 sftp.put()！你有什么新想法吗？

Import […]
[...]

def handler(signum, frame):
        print 'Signal handler called with signal', signum
        raise IOError("Couldn't upload the fileeeeeeeeeeee!!!!")

def check_time(size, file_size):
    global start_time
        if (time.time() - start_time) > 10:
            raise Exception

i = 0
while i == 0:

    try:

        
        time.sleep(1) # CPU break
        print ("go!")

        #collect ip server
        fichierIplist = open("/home/robert/Documents/iplist.txt", "r")
        file_lines = fichierIplist.readlines()
        fichierIplist.close()
        last_line = file_lines [len (file_lines)-1]
        lastKnowip = last_line

        data = glob.glob("/home/robert/Documents/data/*")   
        items = len(data) 

        if items != 0: 
        time.sleep(60) #anyway
            print("some Files!:)")
            raspi = paramiko.SSHClient()
            raspi.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            raspi.connect(lastKnowip, username= "", password= "" , timeout=10)

            for source in data: #Upload file by file    
                filename = os.path.basename(source) #
                destination = '/home/pi/Documents/pest/'+ filename #p

                sftp = raspi.open_sftp()

                signal.signal(signal.SIGALRM, handler)
                signal.alarm(10)

                sftp.get_channel().settimeout(10)
                start_time = time.time()

                sftp.put(source, destination, callback=check_time)

                sftp.close()    
                signal.alarm(0) 
            raspi.close()

        else:

            print("noFile!")

    except:
        pass

Answer 1

请改用 sftp.get_channel().settimeout(s)。

Answer 2

如果要超时，当服务器停止响应时：

设置 SSHClient.connect 的 timeout 参数（您已经这样做了），
并按照@EOhm

sftp.get_channel().settimeout

如果即使服务器正在响应也想超时，但速度较慢，请实施 callback 参数以在特定时间后中止传输：

start_time = time.time()

def check_time(size, file_size):
    global start_time
    if (time.time() - start_time) > ...:
        raise Exception

sftp.put(source, destination, callback=check_time)

这不会立即取消转移。为了优化传输性能，Paramiko 将写入请求排队到服务器。一旦您尝试取消传输，Paramiko 必须等待 SFTPFile.close() 中对这些请求的响应以清除队列。您可以通过使用 SFTPClient.putfo() 并避免在取消传输时调用 SFTPFile.close() 来解决该问题。但是之后您将无法使用该连接。当然，你也可以不使用优化，那么你可以毫不拖延地取消传输。但这违背了所有这一切的要点，不是吗？

或者，您可以运行在单独的线程中传输，如果花费的时间太长则终止线程。丑陋但可靠的解决方案。

Answer 3

在尝试了很多事情之后，在您的帮助和建议下，我找到了一个可靠的解决方案来满足我的需求。我在单独的线程中执行 sftp.put 并且我的脚本执行我想要的操作。非常感谢您的帮助

现在，如果服务器在传输过程中关闭，60 秒后，我的脚本将继续使用：

[...]
import threading
[...]
th = threading.Thread(target=sftp.put, args=(source,destination))
th.start()                  
h.join(60)
[...]

如何在 Python 中使用信号模块或其他模块使 Paramiko sftp.put() 超时？

How to timeout Paramiko sftp.put() with signal module or other in Python?

python

ssh

sftp

paramiko