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

Timestamp:

16 Dec 2014, 09:02:13 (10 years ago)

Author:

Henrik Bettermann

Message:

Some repairs of document and contract batch processors (work in progress!).

File:

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

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

@@ -381 +381 @@
 
     location_fields = []
-    additional_fields = ['document_id', 'class_name']
+    additional_fields = ['class_name']
     additional_headers = ['class_name']
 
@@ -422 +422 @@
         document_id = row['document_id'].strip('#')
         parent[document_id] = obj
+        # Reset _curr_doc_id if document_id has been imported
+        site = grok.getSite()
+        if row.get('document_id', None) not in (None, IGNORE_MARKER):
+            site._curr_doc_id -= 1
         return
 
@@ -445 +449 @@
             if class_name != self.factory_name.strip('waeup.'):
                 errs.append(('class_name','wrong processor'))
+        try:
+            # Correct doc_id counter. As the IConverter for documents
+            # creates document objects that are not used afterwards, we
+            # have to fix the site-wide doc_id counter.
+            site = grok.getSite()
+            site._curr_doc_id -= 1
+        except (KeyError, TypeError, AttributeError):
+                pass
         # We have to check document_id.
         document_id = row.get('document_id', None)
@@ -478 +490 @@
 
     location_fields = []
-    additional_fields = ['contract_id', 'class_name']
+    additional_fields = ['class_name']
     additional_headers = ['class_name']
 
@@ -519 +531 @@
         contract_id = row['contract_id'].strip('#')
         parent[contract_id] = obj
+        # Reset _curr_con_id if contract_id has been imported
+        site = grok.getSite()
+        if row.get('contract_id', None) not in (None, IGNORE_MARKER):
+            site._curr_con_id -= 1
         return
 
@@ -542 +558 @@
             if class_name != self.factory_name.strip('waeup.'):
                 errs.append(('class_name','wrong processor'))
+        try:
+            # Correct con_id counter. As the IConverter for contracts
+            # creates contract objects that are not used afterwards, we
+            # have to fix the site-wide con_id counter.
+            site = grok.getSite()
+            site._curr_con_id -= 1
+        except (KeyError, TypeError, AttributeError):
+                pass
         # We have to check contract_id.
         contract_id = row.get('contract_id', None)