Fix import - workon file import

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: