delete uploaded file on import job delete

Author: Jaspreet-singh-1032

example/winners/tests/test_models.py

@@ -0,0 +1,18 @@
+import os
+from django.test import TestCase
+from django.core.files.base import ContentFile
+
+from import_export_celery.models.importjob import ImportJob
+
+
+class ImportJobTestCases(TestCase):
+
+ def test_delete_file_on_job_delete(self):
+ job = ImportJob.objects.create(
+ file=ContentFile(b"", "file.csv"),
+ )
+ file_path = job.file.path
+ assert os.path.exists(file_path)
+ job.delete()
+ assert not os.path.exists(file_path)
+ assert not ImportJob.objects.filter(id=job.id).exists()

import_export_celery/models/importjob.py

@@ -8,7 +8,7 @@
 from django.db import models, transaction
 from django.dispatch import receiver
 
-from django.db.models.signals import post_save
+from django.db.models.signals import post_save, post_delete
 from django.utils.translation import gettext_lazy as _
 
 from import_export.formats.base_formats import DEFAULT_FORMATS
@@ -95,3 +95,13 @@ def importjob_post_save(sender, instance, **kwargs):
  dry_run=getattr(settings, "IMPORT_DRY_RUN_FIRST_TIME", True),
  )
  )
+
+
+@receiver(post_delete, sender=ImportJob)
+def auto_delete_file_on_delete(sender, instance, **kwargs):
+ """
+ Deletes file related to the import job
+ """
+ if instance.file:
+ instance.file.delete()
+ instance.delete()