Merge pull request #69 from xalien10/ft/customise-export-queryset

Ft/customise export queryset

Author: timthelion

.gitignore

@@ -10,3 +10,4 @@ example/django-import-export-celery-export-jobs/
 pipenv/
 django_import_export_celery.egg-info/
 db/
+.idea/

example/winners/models.py

@@ -27,6 +27,10 @@ class WinnersResource(ModelResource):
     class Meta:
         model = Winner
 
+    def get_export_queryset(self):
+        """To customise the queryset of the model resource with annotation override"""
+        return self.Meta.model.objects.all()
+
 
 class WinnersWithAllCapsResource(WinnersResource):
     name_all_caps = Field()

import_export_celery/models/exportjob.py

@@ -100,6 +100,12 @@ def get_content_type(self):
 
     def get_queryset(self):
         pks = json.loads(self.queryset)
+        # If customised queryset for the model exists
+        # then it'll apply filter on that otherwise it'll
+        # apply filter directly on the model.
+        resource_class = self.get_resource_class()
+        if hasattr(resource_class, "get_export_queryset"):
+            return resource_class().get_export_queryset().filter(pk__in=pks)
         return self.get_content_type().model_class().objects.filter(pk__in=pks)
 
     def get_resource_choices(self):