Changeset 12054 for main/waeup.ikoba/trunk/src/waeup/ikoba/customers/batching.py

Timestamp:

25 Nov 2014, 11:29:42 (10 years ago)

Author:

Henrik Bettermann

Message:

Use classname from import file to check if the right processor has been selected.

File:

• main/waeup.ikoba/trunk/src/waeup/ikoba/customers/batching.py (modified) (2 diffs)

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/batching.py

@@ -379 +379 @@
 
     location_fields = []
-    additional_fields = ['document_id']
-    additional_headers = []
+    additional_fields = ['document_id', 'classname']
+    additional_headers = ['classname']
 
     def checkHeaders(self, headerfields, mode='ignore'):
@@ -437 +437 @@
         errs, inv_errs, conv_dict = super(
             CustomerDocumentProcessor, self).checkConversion(row, mode=mode)
-
+        # We need to check if the classname corresponds with the
+        # processor chosen. This is to avoid accidentally wrong imports.
+        if mode != 'remove':
+            classname = row.get('classname', None)
+            if classname != self.factory_name.strip('waeup.'):
+                errs.append(('classname','wrong processor'))
         # We have to check document_id.
         document_id = row.get('document_id', None)
-        if not document_id:
-            document_id = generate_document_id()
-            conv_dict['document_id'] = document_id
-            return errs, inv_errs, conv_dict
-        # document_id must not exist.
         if mode == 'create':
+            if not document_id:
+                document_id = generate_document_id()
+                conv_dict['document_id'] = document_id
+                return errs, inv_errs, conv_dict
             cat = queryUtility(ICatalog, name='documents_catalog')
             results = list(
                 cat.searchResults(document_id=(document_id, document_id)))
             if results:
+                # document_id must not exist.
                 errs.append(('document_id','id exists'))
-        if not document_id.startswith('d'):
-            errs.append(('document_id','invalid format'))
+        else:
+            if not document_id.startswith('d'):
+                errs.append(('document_id','invalid format'))
         return errs, inv_errs, conv_dict