如何在Django-import-export管理面板获取导入文件名并拦截重复上传

解决方案：拦截导入流程获取文件名并阻止重复上传 #

要实现你的需求，核心是在Django-import-export的导入流程中捕获上传文件名，验证其唯一性，并在导入成功后持久化记录该文件名。下面提供两种可行的实现方式，你可以根据自身代码结构选择：

方法一：重写Admin导入视图（直接拦截上传请求）

这种方式最直接，在Admin层提前拦截上传请求，先完成文件名唯一性校验，再执行导入逻辑，同时用事务保证只有导入成功才留存文件名记录。

1. 第一步：创建记录已上传文件名的模型
   先定义一个模型来存储所有成功导入的文件名，通过unique=True强制文件名唯一：

from django.db import models

class ImportedFileRecord(models.Model):
    filename = models.CharField(max_length=255, unique=True)
    uploaded_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.filename

    # 可选：统一文件名大小写，避免大小写差异被误判为不同文件
    def save(self, *args, **kwargs):
        self.filename = self.filename.lower()
        super().save(*args, **kwargs)

2. 第二步：在Admin类中重写import_view方法
   在目标模型的Admin类中，继承ImportExportMixin后重写导入视图，加入文件名检查和事务处理：

from django.contrib import admin
from import_export.admin import ImportExportMixin
from django.core.exceptions import ValidationError
from django.db import transaction
from .models import YourTargetModel, ImportedFileRecord

class YourTargetModelAdmin(ImportExportMixin, admin.ModelAdmin):
    # 配置你的Admin选项，比如list_display、resource_class等
    resource_class = YourTargetResource  # 若有自定义资源类则指定

    def import_view(self, request, *args, **kwargs):
        if request.method == 'POST':
            import_file = request.FILES.get('import_file')
            if not import_file:
                return super().import_view(request, *args, **kwargs)
            
            # 同步模型的文件名处理逻辑
            filename = import_file.name.lower()
            
            # 检查文件名是否已存在
            if ImportedFileRecord.objects.filter(filename=filename).exists():
                raise ValidationError(f"错误：文件「{import_file.name}」已经上传过，无法重复导入！")
            
            # 事务包裹导入逻辑，确保只有导入成功才记录文件名
            try:
                with transaction.atomic():
                    # 先执行父类的导入逻辑
                    response = super().import_view(request, *args, **kwargs)
                    # 导入成功后持久化文件名记录
                    ImportedFileRecord.objects.create(filename=filename)
                    return response
            except Exception as e:
                # 导入失败时抛出错误，事务自动回滚，不会留存无效记录
                raise ValidationError(f"导入失败：{str(e)}") from e
        # GET请求（如导入页面）直接调用父类方法
        return super().import_view(request, *args, **kwargs)

方法二：利用资源类的钩子函数（更贴合import-export扩展设计）

如果你希望把文件名验证逻辑和资源类绑定，可以借助import-export提供的before_import和after_import钩子，在导入前后处理文件名：

1. 第一步：创建ImportedFileRecord模型
   和方法一的模型完全一致，此处不再重复。

2. 第二步：自定义资源类并添加钩子
   在资源类中通过钩子获取文件名、验证唯一性，并在导入成功后记录：

from import_export import resources
from django.core.exceptions import ValidationError
from .models import YourTargetModel, ImportedFileRecord

class YourTargetResource(resources.ModelResource):
    class Meta:
        model = YourTargetModel

    def before_import(self, dataset, using_transactions, dry_run, **kwargs):
        # 获取上传文件名（dataset自动携带上传文件的名称）
        filename = dataset.filename
        if not filename:
            return
        
        # 同步模型的文件名处理逻辑
        normalized_filename = filename.lower()
        
        # 预览模式下仅验证不保存记录
        if dry_run:
            if ImportedFileRecord.objects.filter(filename=normalized_filename).exists():
                raise ValidationError(f"预览错误：文件「{filename}」已经上传过！")
            return
        
        # 实际导入前校验唯一性
        if ImportedFileRecord.objects.filter(filename=normalized_filename).exists():
            raise ValidationError(f"错误：文件「{filename}」已经上传过，无法重复导入！")

    def after_import(self, dataset, result, using_transactions, dry_run, **kwargs):
        # 仅在非预览模式且导入无错误时记录文件名
        if not dry_run and not result.has_errors():
            filename = dataset.filename
            if filename:
                normalized_filename = filename.lower()
                ImportedFileRecord.objects.create(filename=normalized_filename)

最后在Admin类中指定resource_class = YourTargetResource即可生效。

注意事项

• 事务保障：两种方法都通过事务或错误判断机制，确保只有导入成功时才保存文件名记录，避免导入失败后用户无法再次上传同名文件。
• 大小写处理：如果需要忽略文件名大小写差异，务必在模型和验证逻辑中统一处理（如转小写），防止"File.csv"和"file.csv"被误判为不同文件。
• 错误提示：抛出的ValidationError会直接在Admin页面展示给用户，清晰告知重复上传问题。

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