django-import-export :让post_export更新模型

Question

请告诉我如何使用post_export信号更新下面导出的公司模型。我不想导出以前未导出的项目。所以最好的办法就是使用信号。

但是，如果我在下面代码的末尾打印信号中的模型，我会得到一个类型类，例如<class 'app.models.Company'>

我知道我做错了什么，拯救了一个模型，但找不到我周围的路。

也许我不理解here中的django-import-export库的文档。

models.py

class Company(models.Model):
    class Meta:
        verbose_name_plural = "Companies"

    company_name = models.CharField(max_length=254, blank=True)
    website = models.URLField(max_length=254, unique=True)
    address = models.CharField(max_length=254, blank=True, null=True)
    imported = models.BooleanField(default=False)
    exported = models.BooleanField(default=False)
    user = models.ForeignKey(User)

    def __str__(self):
        if self.company_name:
            return self.company_name
        return self.domain

admin.py

from django.dispatch import receiver 
from import_export.signals import post_import, post_export

class CompanyResource(resources.ModelResource):
    class Meta:
        model = Company
        fields = ('website', 'user', 'country', 'source', 'industry')

@admin.register(Company)
class CompanyAmin(ImportExportModelAdmin):
    resource_class = CompanyResource
    list_display = ('domain', 'website', 'exported', 'added_on')
    list_filter = ('user', 'country', 'imported', 'exported', 'added_on')


@receiver(post_export, dispatch_uid='ss1')
def _post_export(model, **kwargs):
    print(model) 
    model.exported = True
    model.save() # This line when included raises Error: "save() missing 1 required positional argument: 'self'" so code breaks here

Answer 1

https://docs.djangoproject.com/en/1.11/topics/signals/#connecting-receiver-functions

注意，该函数接受一个发送方参数以及通配符关键字参数(**kwargs)；所有信号处理程序都必须接受这些参数。

@receiver(post_export, dispatch_uid='ss1')
def _post_export(sender, model, **kwargs):
    model.exported = True
    model.save()

更新: django-import-export不在信号中发送导出的queryset

因此您可以在ModelAdmin中重写export_action

@admin.register(Company)
class CompanyAmin(ImportExportModelAdmin):
    resource_class = CompanyResource
    list_display = ('domain', 'website', 'exported', 'added_on')
    list_filter = ('user', 'country', 'imported', 'exported', 'added_on')

    def export_action(self, request, *args, **kwargs):
        response = super().export_action(request, *args, **kwargs)
        qs = self.get_export_queryset(request)
        qs.update(exported=True)
        return response

Answer 2

АрсенийКраснов解决方案有问题，请在导出之前更新模型，而不是在导出之后。

要在之后导出，必须在resources.ModelResource中添加以下函数

class CompanyResource(resources.ModelResource):
    class Meta:
        model = Company
        fields = ('website', 'user', 'country', 'source', 'industry')

    def after_export(self, queryset, data, *args, **kwargs):
        queryset.update(exported=True)