openpyxl 生成损坏的工作簿

Question

我有以下代码从 MongoDB 获取数据并将其写入 excel，但是当我打开生成的工作簿时它给出了一个错误（“我们在 ~workbook_name~... 中发现了一个问题”）。

代码：

ws = wb.active
client = pymongo.MongoClient("...")
db = client['...']
collection = db['...']
r = 2
for doc in collection.find({'is_timeline_valid': True}):
    for dic in doc['timeline']['datas']:
        if 'pedido' in dic['tramite'].lower():
           ws.cell(row = r, column = 1).value =doc['id_process_unformatted']
        ws.cell(row = r, column = 2).value = dic['data']
        ws.cell(row = r, column = 3).value = dic['tramite']
        wb.save('pedido.xlsx')
        print('*****************************')
        print(dic['tramite'])
        # print('check!')
        r += 1

Answer 1

设置单元格值完成后保存工作簿。

另外，将单元格中设置的值转换为适当的格式。我选择在这里使用str。您可以根据值的内容使用json.dumps。

from itertools import count
from openpyxl import Workbook

wb = Workbook()
ws = wb.active
client = pymongo.MongoClient("...")
db = client['...']
collection = db['...']
valid_timelines = collection.find({'is_timeline_valid': True})
counter = count(start=2)

for doc in valid_timelines:
    for row, dic in zip(counter, doc['timeline']['datas']):
        if 'pedido' in dic['tramite'].lower():
           ws.cell(row=row, column=1, value=str(doc['id_process_unformatted']))

        ws.cell(row=row, column=2, value=str(dic['data']))
        ws.cell(row=row, column=3, value=str(dic['tramite']))

wb.save('pedido.xlsx')