diff options
author | Étienne Loks <etienne.loks@proxience.com> | 2015-06-16 12:50:53 +0200 |
---|---|---|
committer | Étienne Loks <etienne.loks@proxience.com> | 2015-06-16 12:50:53 +0200 |
commit | 43adb2fcc5c46783066423561b36448f2f45855c (patch) | |
tree | 237f8f2c826cf957bc052861decb58f62f9d07d9 /ishtar_common/data_importer.py | |
parent | 01367019109e1f344eb68c6a5edf1ad481a643bc (diff) | |
parent | fc8f8313dff6511f1ee588ff413027f1145aa1d9 (diff) | |
download | Ishtar-43adb2fcc5c46783066423561b36448f2f45855c.tar.bz2 Ishtar-43adb2fcc5c46783066423561b36448f2f45855c.zip |
Merge branch 'stable'
Diffstat (limited to 'ishtar_common/data_importer.py')
-rw-r--r-- | ishtar_common/data_importer.py | 84 |
1 files changed, 44 insertions, 40 deletions
diff --git a/ishtar_common/data_importer.py b/ishtar_common/data_importer.py index 0ec579b4f..19382c4aa 100644 --- a/ishtar_common/data_importer.py +++ b/ishtar_common/data_importer.py @@ -837,51 +837,55 @@ class Importer(object): field_object, model, direct, m2m = \ cls._meta.get_field_by_name(attribute) if m2m: - val = data.pop(attribute) + many_values = data.pop(attribute) 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: - vals = [] - - # contruct many dict for each values - default_dict = {} - ## init with simple values that will be duplicated - for key in val.keys(): - if type(val[key]) not in (list, tuple): - default_dict[key] = val[key] - ## manage multiple values - for key in val.keys(): - if type(val[key]) in (list, tuple): - for idx, v in enumerate(val[key]): - if len(vals) <= idx: - vals.append(default_dict.copy()) - vals[idx][key] = v - - # check that m2m are not empty - notempty = False - for dct in vals: - for k in dct: - if dct[k] not in ("", None): - notempty = True - break - if not notempty: - continue - - for v in vals: - v['defaults'] = v.get('defaults', {}) - if 'history_modifier' in \ - model._meta.get_all_field_names(): - v['defaults']['history_modifier'] = \ + if type(many_values) not in (list, tuple): + many_values = [many_values] + for val in many_values: + if val.__class__ == model: + m2ms.append((attribute, val)) + elif val.__class__ != model and type(val) == dict: + vals = [] + + # contruct many dict for each values + default_dict = {} + ## init with simple values that will be duplicated + for key in val.keys(): + if type(val[key]) not in (list, tuple): + default_dict[key] = val[key] + ## manage multiple values + for key in val.keys(): + if type(val[key]) in (list, tuple): + for idx, v in enumerate(val[key]): + if len(vals) <= idx: + vals.append(default_dict.copy()) + vals[idx][key] = v + + # check that m2m are not empty + notempty = False + for dct in vals: + for k in dct: + if dct[k] not in ("", None): + notempty = True + break + if not notempty: + continue + + for v in vals: + v['defaults'] = v.get('defaults', {}) + if 'history_modifier' in \ + model._meta.get_all_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'): - v.imports.add(self.import_instance) - m2ms.append((attribute, v)) + v, created = model.objects.get_or_create( + **v) + if self.import_instance \ + and hasattr(v, 'imports'): + 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) |