IMovementGenerator's getAggregatedMovementList() should return a list of...

IMovementGenerator's getAggregatedMovementList() should return a list of temporary Simulation Movements.


git-svn-id: https://svn.erp5.org/repos/public/erp5/trunk@31198 20353a03-c40f-0410-a6d1-a30d3c3de9de

product/ERP5/Document/NewOrderRule.py

@@ -34,6 +34,7 @@ from AccessControl import ClassSecurityInfo
 from Products.ERP5Type import Permissions, PropertySheet, interfaces
 from Products.ERP5.Document.Predicate import Predicate
 from Products.ERP5.mixin.rule import RuleMixin
+from Products.ERP5.MovementCollectionDiff import _getPropertyAndCategoryList
 
 class NewOrderRule(RuleMixin, Predicate):
   """
@@ -106,8 +107,14 @@ class OrderRuleMovementGenerator(object):
                                 rounding=False):
     """Input movement list comes from order"""
     order = context.getDefaultCausalityValue()
-    if order is not None:
-      return [x.asContext(order=x.getRelativeUrl()) for x in order.getMovementList(
-        portal_type=order.getPortalOrderMovementTypeList())]
-    else:
-      return []
+    ret = []
+    for movement in order.getMovementList(
+      portal_type=order.getPortalOrderMovementTypeList()):
+      kw = _getPropertyAndCategoryList(movement)
+      simulation_movement = context.newContent(
+        portal_type=RuleMixin.movement_type,
+        temp_object=True,
+        order_value=movement,
+        **kw)
+      ret.append(simulation_movement)
+    return ret

product/ERP5/MovementCollectionDiff.py

@@ -89,8 +89,10 @@ class MovementCollectionDiff(object):
     """
     property_dict = self._property_dict_dict.get(movement)
     if property_dict is None:
-      # movement should be 'New Movement'
-      return self._getPropertyAndCategoryList(movement)
+      # movement should be 'Temporary Simulation Movement'
+      return dict([(x,y) for x,y in movement.__dict__.items() \
+                   if not x.startswith('_') and \
+                   not x in ('id', 'portal_type', 'uid')])
     else:
       return property_dict