[IMP] make SimpleArray produce DataError in memberwise operations

Put DataError and mis_safe_eval in their own files

mis_builder/models/data_error.py

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+# © 2016 ACSONE SA/NV (<http://acsone.eu>)
+# © 2016 Akretion (<http://akretion.com>)
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
+
+
+class DataError(Exception):
+
+    def __init__(self, name, msg):
+        self.name = name
+        self.msg = msg

mis_builder/models/mis_safe_eval.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# © 2016 ACSONE SA/NV (<http://acsone.eu>)
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
+
+import traceback
+
+from openerp.tools.safe_eval import test_expr, _SAFE_OPCODES, _BUILTINS
+
+from .data_error import DataError
+
+
+__all__ = ['mis_safe_eval']
+
+
+def mis_safe_eval(expr, locals_dict):
+    """ Evaluate an expression using safe_eval
+
+    Returns the evaluated value or DataError.
+
+    Raises NameError if the evaluation depends on a variable that is not
+    present in local_dict.
+    """
+    try:
+        c = test_expr(expr, _SAFE_OPCODES, mode='eval')
+        globals_dict = {'__builtins__': _BUILTINS}
+        val = eval(c, globals_dict, locals_dict)
+    except NameError:
+        raise
+    except ZeroDivisionError:
+        val = DataError('#DIV/0', traceback.format_exc())
+    except:
+        val = DataError('#ERR', traceback.format_exc())
+    return val
+
+
+if __name__ == '__main__':
+    import doctest
+    doctest.testmod()

mis_builder/models/simple_array.py

@@ -46,9 +46,16 @@ SimpleArray((2.0, 2.5, 3.0))
 SimpleArray((8.0, 10.0, 12.0))
 >>> b += 2 ; b
 SimpleArray((6.0, 7.0, 8.0))
+>>> a / ((1.0, 0.0, 1.0))
+SimpleArray((1.0, DataError(), 3.0))
+>>> a / 0.0
+SimpleArray((DataError(), DataError(), DataError()))
 """
 
 import operator
+import traceback
+
+from .data_error import DataError
 
 
 __all__ = ['SimpleArray']
@@ -57,12 +64,20 @@ __all__ = ['SimpleArray']
 class SimpleArray(tuple):
 
     def _op(self, op, other):
+        def _o2(x, y):
+            try:
+                return op(x, y)
+            except ZeroDivisionError:
+                return DataError('#DIV/0', traceback.format_exc())
+            except:
+                return DataError('#ERR', traceback.format_exc())
+
         if isinstance(other, tuple):
             if len(other) != len(self):
                 raise TypeError("tuples must have same length for %s" % op)
-            return SimpleArray(map(op, self, other))
+            return SimpleArray(map(_o2, self, other))
         else:
-            return SimpleArray(map(lambda x: op(x, other), self))
+            return SimpleArray(map(lambda z: _o2(z, other), self))
 
     def __add__(self, other):
         return self._op(operator.add, other)

mis_builder/tests/__init__.py

@@ -6,4 +6,5 @@ from . import test_accounting_none
 from . import test_aep
 from . import test_aggregate
 from . import test_mis_builder
+from . import test_mis_safe_eval
 from . import test_simple_array