diff options
Diffstat (limited to 'ishtar_common/data_importer.py')
| -rw-r--r-- | ishtar_common/data_importer.py | 31 |
1 files changed, 28 insertions, 3 deletions
diff --git a/ishtar_common/data_importer.py b/ishtar_common/data_importer.py index 98569ed84..e19c2dcc6 100644 --- a/ishtar_common/data_importer.py +++ b/ishtar_common/data_importer.py @@ -24,6 +24,7 @@ import io import logging import re import sys +import zipfile from django.contrib.auth.models import User from django.db import IntegrityError, transaction @@ -401,6 +402,22 @@ class DateFormater(Formater): 'value': value}) +class FileFormater(Formater): + need_archive = True + + def format(self, value, archive): + value = value.strip() + if not value: + return + zp = zipfile.ZipFile(archive) + try: + with zp.open(value.strip().replace(u'\\', u'/')) as z: + return z.read() + except KeyError: + raise ValueError(_(u"\"%(value)s\" is not a valid path for the " + u"given archive") % {'value': value}) + + class StrToBoolean(Formater, ChoiceChecker): def __init__(self, choices={}, cli=False, strict=False, db_target=None): self.dct = copy.copy(choices) @@ -531,6 +548,9 @@ class Importer(object): self.check_col_num = check_col_num self.line_format = copy.copy(self.LINE_FORMAT) self.import_instance = import_instance + self.archive = None + if import_instance and import_instance.imported_images: + self.archive = import_instance.imported_images self._defaults = self.DEFAULTS.copy() # EXTRA_DEFAULTS are for multiple inheritance if self.EXTRA_DEFAULTS: @@ -825,7 +845,10 @@ class Importer(object): args.append(val_group[idx]) value = func.format(*args) else: - value = func.format(v) + if getattr(func, 'need_archive', False): + value = func.format(v, archive=self.archive) + else: + value = func.format(v) except ValueError, e: if formater.required: self.c_errors = True @@ -924,6 +947,8 @@ class Importer(object): field_names = model._meta.get_all_field_names() for v in vals: if 'history_modifier' in field_names: + if 'defaults' not in v: + v['defaults'] = {} v['defaults']['history_modifier'] = \ self.history_modifier m2m_m2ms = [] @@ -937,7 +962,6 @@ class Importer(object): v = model.objects.create(**v) else: v['defaults'] = v.get('defaults', {}) - print(v) v, created = model.objects.get_or_create( **v) for att, objs in m2m_m2ms: @@ -972,7 +996,8 @@ class Importer(object): continue if not data[attribute]: continue - self.get_field(cls, attribute, data, m2ms, c_path) + if attribute != '__force_new': + self.get_field(cls, attribute, data, m2ms, c_path) # default values path = tuple(path) if path in self._defaults: |