Imports: fix for finds import

1 files changed, 6 insertions, 3 deletions

diff --git a/ishtar_common/data_importer.py b/ishtar_common/data_importer.py
index 30dd7d76c..333a81877 100644
--- a/ishtar_common/data_importer.py
+++ b/ishtar_common/data_importer.py
@@ -408,7 +408,7 @@ class StrToBoolean(Formater):
             for missing in self.missings:
                 try:
                     q = {'target':self.db_target, 'key':missing}
-                    models.TargetKey.objects.create(**q)
+                    TargetKey.objects.create(**q)
                 except IntegrityError:
                     pass
 
@@ -782,7 +782,10 @@ class Importer(object):
                                     cls._meta.get_field_by_name(attribute)
                 if m2m:
                     val = data.pop(attribute)
-                    model = field_object.rel.to
+                    if hasattr(field_object, 'rel'):
+                        model = field_object.rel.to
+                    elif hasattr(field_object, 'to'):
+                        model = field_object.to
                     if val.__class__ == model:
                         m2ms.append((attribute, val))
                     elif val.__class__ != model and type(val) == dict:
@@ -852,7 +855,7 @@ class Importer(object):
                 try:
                     dct = create_dict.copy()
                     dct['defaults'] = defaults
-                    obj, created = cls.objects.get_or_create(**create_dict)
+                    obj, created = cls.objects.get_or_create(**dct)
                     if self.import_instance and hasattr(obj, 'imports'):
                         obj.imports.add(self.import_instance)
                 except IntegrityError as e: