server-ux/base_tier_validation_formula/models/tier_validation.py
2021-03-23 20:38:33 +01:00

26 lines
796 B
Python

# Copyright 2019 Creu Blanca
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from flectra import api, models, _
from flectra.tools.safe_eval import safe_eval
from flectra.exceptions import UserError
class TierValidation(models.AbstractModel):
_inherit = 'tier.validation'
@api.multi
def evaluate_formula_tier(self, tier):
try:
res = safe_eval(tier.python_code, globals_dict={'rec': self})
except Exception as error:
raise UserError(_(
"Error evaluating tier validation conditions.\n %s") % error)
return res
@api.multi
def evaluate_tier(self, tier):
if tier.definition_type == 'formula':
return self.evaluate_formula_tier(tier)
return super().evaluate_tier(tier)