From 6c55681833bd8c45435586ea767ac2324eeac6dc Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 23 Sep 2015 16:07:22 +0200 Subject: Imports: manage m2m in m2m --- ishtar_common/data_importer.py | 29 +++++++++++++++++++++-------- 1 file changed, 21 insertions(+), 8 deletions(-) (limited to 'ishtar_common/data_importer.py') diff --git a/ishtar_common/data_importer.py b/ishtar_common/data_importer.py index a64b908ea..42dab251c 100644 --- a/ishtar_common/data_importer.py +++ b/ishtar_common/data_importer.py @@ -918,18 +918,31 @@ class Importer(object): if not notempty: continue + field_names = model._meta.get_all_field_names() for v in vals: v['defaults'] = v.get('defaults', {}) - if 'history_modifier' in \ - model._meta.get_all_field_names(): + if 'history_modifier' in field_names: v['defaults']['history_modifier'] = \ self.history_modifier - v, created = model.objects.get_or_create( - **v) - if self.import_instance \ - and hasattr(v, 'imports') and created: - v.imports.add(self.import_instance) - m2ms.append((attribute, v)) + m2m_m2ms = [] + for k in v.keys(): + if k not in field_names: + continue + field_object, m, direct, m2m = \ + model._meta.get_field_by_name(k) + if m2m: + m2m_m2ms.append((k, v.pop(k))) + v, created = model.objects.get_or_create( + **v) + for att, objs in m2m_m2ms: + if type(objs) not in (list, tuple): + objs = [objs] + for obj in objs: + getattr(v, att).add(obj) + if self.import_instance \ + and hasattr(v, 'imports') and created: + v.imports.add(self.import_instance) + m2ms.append((attribute, v)) elif hasattr(field_object, 'rel') and field_object.rel and \ type(data[attribute]) == dict: c_path.append(attribute) -- cgit v1.2.3