account-financial-reporting/mis_builder/models/mis_safe_eval.py

# -*- 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, NameDataError


__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)  # pylint: disable=eval-used
    except NameError:
        val = NameDataError('#NAME', traceback.format_exc())
    except ZeroDivisionError:
        val = DataError('#DIV/0', traceback.format_exc())
    except:
        val = DataError('#ERR', traceback.format_exc())
    return val