创建CSV,压缩和附加电子邮件的最佳方式

时间:2014-07-11 09:43:06

标签: python django email csv

我有一个问题,我正在寻找执行以下任务的最佳方式。

  • 创建内存中的csv文件
  • 压缩它(Gizip或Zip)
  • 使用电子邮件附上

我很难找到我正在寻找的确切解决方案。

这是我的代码。

prospects = Customer.objects.filter(state_id=2)
csvfile = StringIO.StringIO()
writer = UnicodeWriter(csvfile, encoding='utf-8')
writer.writerow(["Name", "Email"])
for prospect in prospects:
    writer.writerow(
        [prospect.name, prospect.email]
    )
csvobj = csvfile.getvalue() # not sure about it. 

代码第2部分

g = gzip.GzipFile(fileobj=csvobj)
g.close()

代码第3部分

  msg = EmailMultiAlternatives(
            subject="Innovation",
            body="text_context",
            from_email=settings.DEFAULT_FROM_EMAIL,
            to=["bac@gmail.com"]
        )
  msg.attach("z.gzip",g, 'application/zip')
  msg.send(True)

错误

TypeError: 'GzipFile' object does not support indexing

1 个答案:

答案 0 :(得分:2)

以下是我的表现。

import StringIO
import tablib

headers = ["name", "Email"]
values = Customer.objects.filter(state_id=2).values_list(‘name’, ‘email’)
data = tablib.Dataset(*values, headers=headers)

csvfile = StringIO.StringIO()
csvfile.write(data.csv)

gzipped = gzip.GzipFile(mode='wb', fileobj=csvfile.getvalue())
gzipped.close()

# msg stuff...
msg.attach("z.gzip", gzipped, 'application/zip')

对我来说很好,如果你尝试的话,你仍然会收到错误吗?