3、DjangoAdmin导出excel和csv文件

一、导出Excel

1、安装openpyxl库

2、admin文件

# 导入openpyxl库中的Workbook类，用于创建Excel文件  
from openpyxl import Workbook  
# 导入Django的admin模块，用于在Django admin后台注册和管理模型  
from django.contrib import admin  
# 导入HttpResponse类，用于生成HTTP响应  
from django.http import HttpResponse  
# 假设这是从某个应用的models.py文件中导入的PrettyNum模型  
from .models import Category  

# 定义一个名为ExportExcelMixin的混合类（Mixin），用于添加导出Excel的功能  
class ExportExcelMixin(object):  
    # 定义一个方法export_as_excel，用于将查询集（queryset）导出为Excel文件  
    def export_as_excel(self, request, queryset):  
        # 获取当前模型的元数据对象  
        meta = self.model._meta  
        # 从元数据对象中提取所有字段的名称  
        field_names = [field.name for field in meta.fields]  
      
        # 创建一个HttpResponse对象，设置内容类型为Excel文件  
        response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')  
        # 设置响应的Content-Disposition头部，指定文件名为模型名加.xlsx后缀  
        response['Content-Disposition'] = f'attachment; filename={meta.model_name}.xlsx'  
      
        # 创建一个Workbook对象，即一个Excel工作簿  
        wb = Workbook()  
        # 激活工作簿中的第一个工作表  
        ws = wb.active  
        # 将字段名称作为第一行写入工作表  
        ws.append(field_names)  
      
        # 遍历查询集中的每个对象  
        for obj in queryset:  
            # 对于每个对象，提取其字段值，并将它们作为一行数据写入工作表  
            row_data = [getattr(obj, field) for field in field_names]  
            ws.append(row_data)  
      
        # 将工作簿保存到HttpResponse对象中，以便作为文件下载  
        wb.save(response)  
        # 返回包含Excel文件的HttpResponse对象  
        return response  
  
    # 为export_as_excel方法设置简短描述，以便在Django admin后台显示  
    export_as_excel.short_description = '导出Excel'  


# 使用admin.register装饰器注册PrettyNum模型到Django admin后台  
# 同时，通过继承ExportExcelMixin混合类，为PrettyNumAdmin类添加导出Excel的功能  
@admin.register(Category)  
class PrettyNumAdmin(admin.ModelAdmin, ExportExcelMixin):  
    # 定义在Django admin后台表单中显示的字段  
    fields = ('name', )  
  
    # 定义在Django admin后台列表视图中显示的字段  
    list_display = ('name', )  
  
    # 定义在Django admin后台列表视图中可用的过滤器  
    list_filter = ('name', )   
  
    # 定义在Django admin后台搜索栏中可搜索的字段  
    search_fields = ('name', )   
  
    # 将export_as_excel方法添加到actions列表中，以便在Django admin后台作为操作选项显示  
    actions = ['export_as_excel']

ExportExcelMixin类为excel的导出类可直接复制，带装饰器的类，修改类参数，修改字段。

二、导出csv

导出类似，与导出excel代码合并如下

from django.contrib import admin  
from django.http import HttpResponse  
from django.urls import reverse  
from urllib.parse import quote  
from openpyxl import Workbook  
import csv  
from .models import Category  
  
class ExportExcelMixin:  
    def export_as_excel(self, request, queryset):  
        # 获取当前模型的元数据对象  
        meta = self.model._meta  
        # 从元数据对象中提取所有字段的名称  
        field_names = [field.name for field in meta.fields]  
      
        # 创建一个HttpResponse对象，设置内容类型为Excel文件  
        response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')  
        # 设置响应的Content-Disposition头部，指定文件名为模型名加.xlsx后缀  
        response['Content-Disposition'] = f'attachment; filename={meta.model_name}.xlsx'  
      
        # 创建一个Workbook对象，即一个Excel工作簿  
        wb = Workbook()  
        # 激活工作簿中的第一个工作表  
        ws = wb.active  
        # 将字段名称作为第一行写入工作表  
        ws.append(field_names)  
      
        # 遍历查询集中的每个对象  
        for obj in queryset:  
            # 对于每个对象，提取其字段值，并将它们作为一行数据写入工作表  
            row_data = [getattr(obj, field) for field in field_names]  
            ws.append(row_data)  
      
        # 将工作簿保存到HttpResponse对象中，以便作为文件下载  
        wb.save(response)  
        # 返回包含Excel文件的HttpResponse对象  
        return response  
  
    export_as_excel.short_description = '导出Excel'  
  
class ExportCsvMixin:  
    def export_as_csv(self, request, queryset):  
        meta = self.model._meta  
        field_names = [field.name for field in meta.fields]  
  
        filename = f"{meta.verbose_name_plural}.csv"  
        filename_quoted = quote(filename)  
  
        response = HttpResponse(content_type='text/csv; charset=utf-8')  
        response['Content-Disposition'] = f'attachment; filename="{filename_quoted}"'  
  
        writer = csv.writer(response)  
        writer.writerow(field_names)  
        for obj in queryset:  
            row_data = [getattr(obj, field) for field in field_names]  
            writer.writerow(row_data)  
  
        return response  
  
    export_as_csv.short_description = '导出CSV'  
  
@admin.register(Category)  
class CategoryAdmin(admin.ModelAdmin, ExportExcelMixin, ExportCsvMixin):  
    fields = ('name',)  
    list_display = ('name',)  
    list_filter = ('name',)  
    search_fields = ('name',)  
    actions = ['export_as_excel', 'export_as_csv']  

写导出excel和导出csv文件的两个类，写一个导出字段的类引入以上两个类。