【django】 django后台管理导出excel表

wanghong1994 2020-04-07 原文

from django.contrib import admin

# Register your models here.
import xlwt
from django.http import HttpResponse

from dal import models
# Register your models here.

from openpyxl import Workbook


class FundAdmin(admin.ModelAdmin):
    actions = ["export_as_excel"]
    list_display = ('title', 'name','type','unit','status','create_time')
    list_filter = ['title', 'name','type','unit','status','create_time',]


    def export_as_excel(self, request, queryset):
        print("queryset",queryset)
        meta = self.model._meta  # 用于定义文件名, 格式为: app名.模型类名
        print("meta",meta)
        field_names = [field.name for field in meta.fields]  # 模型所有字段名

        response = HttpResponse(content_type='application/msexcel')  # 定义响应内容类型
        response['Content-Disposition'] = f'attachment; filename={meta}.xlsx'  # 定义响应数据格式
        wb = Workbook()  # 新建Workbook
        ws = wb.active  # 使用当前活动的Sheet表
        ws.append(field_names)  # 将模型字段名作为标题写入第一行
        for obj in queryset:  # 遍历选择的对象列表
            print(obj)
            for field in field_names:

                data = [f'{getattr(obj, field)}' for field in field_names]  # 将模型属性值的文本格式组成列表

            ws.append(data)  # 写入模型属性值
        wb.save(response)  # 将数据存入响应内容
        return response


    export_as_excel.short_description = '导出Excel'  # 该动作在admin中的显示文字



admin.site.register(models.Fund, FundAdmin)