🐛 imports - manage post-import errors

1 files changed, 4 insertions, 4 deletions

diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index 8ffc2ff2f..95dd5cac2 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -80,7 +80,7 @@ from django.template.defaultfilters import slugify
 from django.urls import reverse
 from django.utils.functional import lazy
 from django.utils.safestring import mark_safe
-from ishtar_common.data_importer import post_importer_action
+from ishtar_common.data_importer import ImporterError, post_importer_action
 from ishtar_common.utils import (
     ugettext_lazy as _,
     ugettext,
@@ -5908,11 +5908,11 @@ class Document(
         Container = apps.get_model("archaeological_warehouse", "Container")
         try:
             container = Container.objects.get(code=code)
-        except DoesNotExist:
-            raise ImportError(
+        except ObjectDoesNotExist:
+            raise ImporterError(
                 str(_("Container with code {} not found.")).format(code))
         except MultipleObjectsReturned:
-            raise ImportError(
+            raise ImporterError(
                 str(_("Multiple containers with code {} found.")).format(code))
         self.__class__.objects.filter(pk=self.id).update(
             **{target: container.pk})