通过文本和文件流保存 openpyxl 文件

Question

我正在将 OpenPyXL 构建到一个应用程序中，该应用程序需要一个包含 excel 文件内容的字符串，以便通过文件流写入。

根据我对 OpenPyXL 源代码的调查，它看起来并不支持这种输出。有谁有修改 openpyxl 以支持这一点的经验吗？

或者有什么一般建议/解决方法吗？

谢谢。

Answer 1

在 openpyxl 2.6 中，调用 save_virtual_workbook 方法会发出以下警告：

DeprecationWarning: Call to deprecated function save_virtual_workbook (Use a NamedTemporaryFile).

在某些时候 save_virtual_workbook 将从 openpyxl 中删除。

在 Python 3 中，将 openpyxl 工作簿保存到文件流的典型用法变为：

from io import BytesIO
from tempfile import NamedTemporaryFile
from openpyxl import Workbook

wb = Workbook()
with NamedTemporaryFile() as tmp:
    wb.save(tmp.name)
    output = BytesIO(tmp.read())

在查看 WorkBook save 方法的实现之后，“文件名”被直接发送到 ZipFile，该 ZipFile 接受路径或类似文件对象，因此不需要 NamedTemporaryFile，只需使用内存中的 BytesIO：

from io import BytesIO
from openpyxl import Workbook

wb = Workbook()
virtual_workbook = BytesIO()
wb.save(virtual_workbook)

# now use virtual_workbook to send to a stream; email attachment, etc

Answer 2

jcollado的答案实际上是有效的，但在 openpyxl.writer.excel 中还有一个名为“save_virtual_workbook”的函数（遗憾的是尚未记录），它将获取您的工作簿并将工作簿作为字符串返回：

from openpyxl.workbook import Workbook
from openpyxl.writer.excel import save_virtual_workbook

wb = Workbook()
print save_virtual_workbook(wb)

您正在寻找的是字符串由 save_virtual_workbook() 返回

Answer 3

如何使用 StringIO 对象来保存文件的内容：

from openpyxl.workbook import Workbook
from StringIO import StringIO

output = StringIO()
wb = Workbook()
wb.save(output)
print output.getvalue()

您要查找的字符串就是本示例最后一行中打印的内容。

Answer 4

自版本 2.6 起，已弃用与 save_virtual_workbook 兼容的实现：

from io import BytesIO
from tempfile import NamedTemporaryFile


def save_virtual_workbook(workbook):
    with NamedTemporaryFile() as tf:
        workbook.save(tf.name)
        in_memory = BytesIO(tf.read())
        return in_memory.getvalue()

Answer 5

from openpyxl import Workbook
from io import BytesIO

rows = [[1,2], [3,4]]

book = Workbook()
sheet = book.active

for row in rows:
    sheet.append(row)

io = BytesIO
book.save(io)

content = io.getValue()

return Response(
    content,
    mimetype=magic.from_buffer(content, mime=True),
    headers={
    'Content-Disposition': 'attachment;filename=' + 'test.xlsx'}
)