python - 从 FTP 检索文件并直接写入 zip 存档

Question

我想从 FTP 服务器下载文件并将其存档在本地 (zip) 存档中。

众所周知如何下载文件并单独保存它们:

import ftplib
remote = ftplib.FTP(ftp_server) 
remote.login(username, password) 
for filename in file_list:
    remote.retrbinary("RETR " + filename, open(filename, 'wb').write)
remote.quit()

如何将文件添加到存档中也是众所周知的:

import zipfile
archive = zipfile.ZipFile(archive_file)
archive.write(filename)
archive.close()

但似乎不可能同时使用两者:

remote.retrbinary("RETR " + filename, archive.write(filename))

这会导致 FileNotFoundError，因为 filename 尚未保存到其间的本地(临时)目录中。

有没有办法直接将文件流从 FTP 发送到 zip 存档中？或者直接下载所有文件，将它们添加到存档中，然后删除文件会更有效吗？我希望硬盘 I/O 尽可能低。

Answer 1

将文件下载到内存并使用ZipFile.writestr :

import ftplib
import zipfile
from io import BytesIO

# ...

archive = zipfile.ZipFile(archive_file, "w")

for filename in file_list:
    flo = BytesIO()
    ftp.retrbinary('RETR ' + filename, flo.write)
    archive.writestr(filename, flo.getvalue())

archive.close()