diff options
Diffstat (limited to 'ishtar_common/data_importer.py')
| -rw-r--r-- | ishtar_common/data_importer.py | 29 |
1 files changed, 24 insertions, 5 deletions
diff --git a/ishtar_common/data_importer.py b/ishtar_common/data_importer.py index 4285a658b..7717c0e03 100644 --- a/ishtar_common/data_importer.py +++ b/ishtar_common/data_importer.py @@ -32,7 +32,7 @@ from django.contrib.auth.models import User from django.contrib.contenttypes.models import ContentType from django.contrib.gis.geos.error import GEOSException from django.db.models.fields import FieldDoesNotExist -from django.core.exceptions import FieldError +from django.core.exceptions import FieldError, MultipleObjectsReturned from django.core.files import File from django.db import IntegrityError, DatabaseError, transaction from django.db.models import Q @@ -881,10 +881,29 @@ class Importer(object): def post_processing(self, idx_line, item): # force django based post-processing for the item item = item.__class__.objects.get(pk=item.pk) - for func, context, value in self._item_post_processing: + for cls, func, context, value in self._item_post_processing: context["import_object"] = self.import_instance + if cls != item.__class__: + # try to get associated item + current_item = None + for id_key in ("id", "pk", "external_id"): + if context.get(id_key, None): + try: + current_item = cls.objects.get(**{id_key: context[id_key]}) + except (cls.DoesNotExist, MultipleObjectsReturned) as __: + break + if not current_item: + self.errors.append( + (idx_line, None, + str(_("Item {} with values: {} not identified for post-treatment - need a non ambiguous key")).format( + cls.__name__, str(context) + )) + ) + continue + else: + current_item = item try: - f = getattr(item, func) + f = getattr(current_item, func) returned = f(context, value) if returned and not getattr(f, "_no_repost_save", False): if not isinstance(returned, Iterable): @@ -1215,7 +1234,7 @@ class Importer(object): self._throughs = [] # list of (formater, value) self._post_processing = [] # list of (formater, value) - self._item_post_processing = [] + self._item_post_processing = [] # cls, attribute, data, value data = {} self.current_csv_line = None @@ -1811,7 +1830,7 @@ class Importer(object): if func.importer_trigger == "pre": func(data, data[attribute]) elif func.importer_trigger == "post": - self._item_post_processing.append([attribute, data, data[attribute]]) + self._item_post_processing.append([cls, attribute, data, data[attribute]]) else: logger.warning( "Unknow importer_trigger '{}' for '{}'".format( |