我正在尝试修改现有的python脚本,以将内存中的CSV转换为(也位于内存中)zip存档,并将其作为电子邮件附件发送。我已经成功附加了一个内存中的CSV文件(通过MIMEText对象),但是由于需要设置文件字节的有效负载,因此ZIP文件(通过MIMEBase对象)遇到了麻烦。
这是我到目前为止的代码:
csv_buffer = cStringIO.StringIO()
buffer = cStringIO.StringIO()
zf = zipfile.ZipFile(buffer,
mode='w',
compression=zipfile.ZIP_DEFLATED,
)
zf.writestr(csvfile + ".csv", csv_buffer.getvalue())
csv_file = MIMEBase('application', 'zip')
csv_file.set_payload(zf.read())
encoders.encode_base64(csv_file)
csv_file.add_header('Content-Disposition', 'attachment',
filename=csvfile + ".zip")
msg.attach(csv_file)
在此This user's most upvoted answer上,我可以通过对zip文件执行常规的read() takes at least 2 arguments
操作来解决收到的open()
错误,但是由于该文件是缓冲流,因此无法工作。
我不确定如何将zipfile
对象加载到set_payload
中,但是我认为应该也可以。
答案 0 :(得分:2)
read
对象的ZipFile
方法用于读取zip文件中命名文件的内容。
通过buffer
实例的writestr
方法写入缓冲区后,应该使用ZipFile
的内容:
csv_file.set_payload(buffer.getvalue())