如何通过Django import-export仅导出当前用户的产品？

要实现仅导出当前登录用户名下的产品，只需对现有代码做简单调整，核心思路是过滤出当前用户的产品数据再导出，以下是两种可行方案：

方案一：直接在视图中过滤查询集（推荐） #

这种方式简洁直接，无需修改资源类，仅调整视图函数即可：

1. 确保视图需要登录权限（可选但建议） #

避免匿名用户请求导致错误，给视图添加登录验证装饰器：

from django.contrib.auth.decorators import login_required

2. 修改export_excel视图函数 #

在调用export方法前，先筛选出当前用户的产品，再传递给资源类：

@login_required
def export_excel(request):
    # 过滤当前用户名下的产品
    user_products = Product.objects.filter(author=request.user)
    # 将过滤后的查询集传入export方法
    dataset = ProductResource().export(queryset=user_products)
    
    response = HttpResponse(dataset.xlsx, content_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
    response["Content-Disposition"] = f"attachment; filename=Products-{datetime.datetime.now().date()}.xlsx" 
    
    return response 

方案二：在资源类中封装过滤逻辑 #

如果需要将过滤逻辑统一封装在资源类中，方便复用，可以按以下方式修改：

1. 修改resources.py中的ProductResource类 #

添加初始化参数接收用户对象，并重写get_queryset方法实现过滤：

from import_export import resources
from .models import Product

class ProductResource(resources.ModelResource):
    author = Field() 
    brand_id = Field() 

    def __init__(self, user=None):
        self.user = user
        super().__init__()

    def get_queryset(self):
        queryset = super().get_queryset()
        if self.user:
            # 仅返回当前用户的产品
            queryset = queryset.filter(author=self.user)
        return queryset

    class Meta:
        model = Product
        fields = ["id", "author", "brand_id", "name", "barcode", "unit"]
        export_order = ["id", "author", "brand_id", "name", "barcode", "unit"]

    def dehydrate_author(self, product: Product) -> str:
        return f"{product.author.username}" 

    def dehydrate_brand_id(self, product: Product) -> str:
        return f"{product.brand_id.name}" 

2. 对应修改视图函数 #

实例化资源类时传入当前登录用户：

@login_required
def export_excel(request):
    dataset = ProductResource(user=request.user).export()
    
    response = HttpResponse(dataset.xlsx, content_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
    response["Content-Disposition"] = f"attachment; filename=Products-{datetime.datetime.now().date()}.xlsx" 
    
    return response 

两种方案都能实现需求，方案一适合简单场景，方案二更适合需要复用资源类的复杂场景。

内容的提问来源于stack exchange，提问作者Eric Maniel