提取数据库数据写入表格中

import os, django

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "RDPlatform.settings") # project_name 项目名称
django.setup()

from rdkm.models import Knowledge, WorkFlow
from openpyxl import load_workbook

def get_name():
# 查询审核流程到结束的所有知识对象
queryset = Knowledge.objects.filter(workflow__process_status='结束')

alist = []
for i in queryset:
    name = i.knowledge_name
    alist.append(name)

blist = []
for k in range(1, len(alist)):
    blist.append(k)
# print(blist)

adict = dict(zip(blist, alist))

"""
将数据写入表格
"""

# # 设置文件 mingc
addr = "3.xlsx"

# 打开文件
wb = load_workbook(addr)

sheets = wb.sheetnames
# print(sheets)

for sheetname in sheets:
    ws = wb.get_sheet_by_name(sheetname)
    # print(ws)
    ws.append(['id', 'name'])
    ID = 'id'
    NAME = 'name'

    for id, name in adict.items():
        ID = id
        NAME = name
        ws.append([ID, NAME])
    wb.save(addr)

get_name()

posted @ 2019-09-16 10:53  温温温  阅读(102)  评论(0)    收藏  举报