Python 的 ftplib 与 tqdm

Question

我有一个使用 ftplib 的控制台脚本作为从 ftp 服务器获取大量文件的后端。我想使用 tqdm给用户一些反馈，前提是他们打开了“详细”开关。这必须是可选的，因为某些用户可能会在没有tty 访问权限的情况下使用该脚本。

ftplib的 retrbinary 方法接受回调，因此应该可以 Hook tqdm不知何故在那里。但是，我不知道这个回调会是什么样子。

Answer 1

来自FTP.retrbinary:

The callback function is called for each block of data received, with a single string argument giving the data block.

所以回调可能是这样的:

with open(filename, 'wb') as fd:
    total = ftpclient.size(filename)

    with tqdm(total=total) as pbar:
        def callback_(data):
            l = len(data)
            pbar.update(l)
            fd.write(data)

        ftpclient.retrbinary('RETR {}'.format(filename), callback_)

注意:此代码未经测试，可能需要修改。